Allow passing DB into migration

Rakefile

@@ -12,6 +12,12 @@ task 'migrate' do
   migrate
 end
 
+task 'migrate_test' do
+  require_relative './db/migrate'
+
+  migrate(db: TEST_DB)
+end
+
 task 'scrape' do
   require_relative './db/scrape'
 

db/migrate.rb

@@ -3,8 +3,7 @@ require 'sequel'
 
 require_relative 'connect'
 
-def migrate
-  db = DB
+def migrate(db: DB)
   db.create_table? :threads do
     primary_key :id
     String :title