Extract Migrator#create and #drop sql into separate methods

spec/granite/migrator/migrator_spec.cr

@@ -0,0 +1,135 @@
+require "../../spec_helper"
+
+describe Granite::Migrator::Base do
+  describe "#drop_sql" do
+    it "generates correct SQL with #{{{ env("CURRENT_ADAPTER") }}} adapter" do
+      {% if env("CURRENT_ADAPTER") == "mysql" %}
+        Review.migrator.drop_sql.should eq "DROP TABLE IF EXISTS `reviews`;"
+      {% else %}
+        Review.migrator.drop_sql.should eq "DROP TABLE IF EXISTS \"reviews\";"
+      {% end %}
+    end
+  end
+
+  describe "#create_sql" do
+    it "generates correct SQL with #{{{ env("CURRENT_ADAPTER") }}} adapter" do
+      {% if env("CURRENT_ADAPTER") == "pg" %}
+        Review.migrator.create_sql.should eq <<-SQL
+          CREATE TABLE "reviews"(
+          "id" BIGSERIAL PRIMARY KEY,
+          "name" VARCHAR(255)
+          ,
+          "downvotes" INT
+          ,
+          "upvotes" BIGINT
+          ,
+          "sentiment" FLOAT
+          ,
+          "interest" REAL
+          ,
+          "published" BOOL
+          ,
+          "created_at" TIMESTAMP
+          ) ;\n
+          SQL
+
+        Kvs.migrator.create_sql.should eq <<-SQL
+          CREATE TABLE "kvs"(
+          "k" VARCHAR(255) PRIMARY KEY,
+          "v" VARCHAR(255)
+          ) ;\n
+          SQL
+
+        UUIDModel.migrator.create_sql.should eq <<-SQL
+          CREATE TABLE "uuids"(
+          "uuid" UUID PRIMARY KEY) ;\n
+          SQL
+
+        # Also check Array types for pg
+        ArrayModel.migrator.create_sql.should eq <<-SQL
+          CREATE TABLE "array_model"(
+          "id" SERIAL PRIMARY KEY,
+          "str_array" TEXT[]
+          ,
+          "i16_array" SMALLINT[]
+          ,
+          "i32_array" INT[]
+          ,
+          "i64_array" BIGINT[]
+          ,
+          "f32_array" REAL[]
+          ,
+          "f64_array" DOUBLE PRECISION[]
+          ,
+          "bool_array" BOOLEAN[]
+          ) ;\n
+          SQL
+
+      {% elsif env("CURRENT_ADAPTER") == "mysql" %}
+        Review.migrator.create_sql.should eq <<-SQL
+          CREATE TABLE `reviews`(
+          `id` BIGINT NOT NULL AUTO_INCREMENT PRIMARY KEY,
+          `name` VARCHAR(255)
+          ,
+          `downvotes` INT
+          ,
+          `upvotes` BIGINT
+          ,
+          `sentiment` FLOAT
+          ,
+          `interest` REAL
+          ,
+          `published` BOOL
+          ,
+          `created_at` TIMESTAMP DEFAULT CURRENT_TIMESTAMP
+          ) ;\n
+          SQL
+
+        Kvs.migrator.create_sql.should eq <<-SQL
+          CREATE TABLE `kvs`(
+          `k` VARCHAR(255) PRIMARY KEY,
+          `v` VARCHAR(255)
+          ) ;\n
+          SQL
+
+        UUIDModel.migrator.create_sql.should eq <<-SQL
+          CREATE TABLE `uuids`(
+          `uuid` CHAR(36) PRIMARY KEY) ;\n
+          SQL
+
+      {% elsif env("CURRENT_ADAPTER") == "sqlite" %}
+        Review.migrator.create_sql.should eq <<-SQL
+          CREATE TABLE "reviews"(
+          "id" INTEGER NOT NULL PRIMARY KEY,
+          "name" VARCHAR(255)
+          ,
+          "downvotes" INTEGER
+          ,
+          "upvotes" INTEGER
+          ,
+          "sentiment" FLOAT
+          ,
+          "interest" REAL
+          ,
+          "published" BOOL
+          ,
+          "created_at" VARCHAR
+          ) ;\n
+          SQL
+
+        Kvs.migrator.create_sql.should eq <<-SQL
+          CREATE TABLE "kvs"(
+          "k" VARCHAR(255) PRIMARY KEY,
+          "v" VARCHAR(255)
+          ) ;\n
+          SQL
+
+        UUIDModel.migrator.create_sql.should eq <<-SQL
+          CREATE TABLE "uuids"(
+          "uuid" CHAR(36) PRIMARY KEY) ;\n
+          SQL
+
+      {% end %}
+    end
+  end
+end

src/granite/migrator.cr

@@ -43,16 +43,20 @@ module Granite::Migrator
     {% adapter = "#{klass}.adapter".id %}
 
     disable_granite_docs? class Migrator < Granite::Migrator::Base
+      def drop_sql
+        "DROP TABLE IF EXISTS #{ @quoted_table_name };"
+      end
+
       def drop
-        {{klass}}.exec "DROP TABLE IF EXISTS #{ @quoted_table_name };"
+        {{klass}}.exec drop_sql
       end
 
-      def create
+      def create_sql
         resolve = ->(key : String) {
           {{adapter}}.class.schema_type?(key) || raise "Migrator(#{ {{adapter}}.class.name }) doesn't support '#{key}' yet."
         }
 
-        stmt = String.build do |s|
+        String.build do |s|
           s.puts "CREATE TABLE #{ @quoted_table_name }("
 
           # primary key
@@ -82,8 +86,10 @@ module Granite::Migrator
 
           s.puts ") #{@table_options};"
         end
+      end
 
-        {{klass}}.exec stmt
+      def create
+        {{klass}}.exec create_sql
       end
     end