activerecord-spanner-adapter 0.1.0 → 0.7.0

data/acceptance/cases/migration/create_join_table_test.rb

@@ -0,0 +1,216 @@
+# Copyright 2020 Google LLC
+#
+# Use of this source code is governed by an MIT-style
+# license that can be found in the LICENSE file or at
+# https://opensource.org/licenses/MIT.
+
+# frozen_string_literal: true
+
+require "test_helper"
+
+module ActiveRecord
+  class Migration
+    class CreateJoinTableTest < SpannerAdapter::TestCase
+      include SpannerAdapter::Migration::TestHelper
+
+      def setup
+        skip_test_table_create!
+        super
+      end
+
+      def teardown
+        connection.ddl_batch do
+          %w(artists_musics musics_videos catalog).each do |table_name|
+            connection.drop_table table_name, if_exists: true
+          end
+        end
+      end
+
+      def test_create_join_table
+        connection.ddl_batch do
+          connection.create_join_table :artists, :musics
+        end
+
+        assert_equal %w(artist_id music_id), connection.columns(:artists_musics).map(&:name).sort
+      end
+
+      def test_create_join_table_set_not_null_by_default
+        connection.ddl_batch do
+          connection.create_join_table :artists, :musics
+        end
+
+        assert_equal [false, false], connection.columns(:artists_musics).map(&:null)
+      end
+
+      def test_create_join_table_with_strings
+        connection.ddl_batch do
+          connection.create_join_table "artists", "musics"
+        end
+
+        assert_equal %w(artist_id music_id), connection.columns(:artists_musics).map(&:name).sort
+      end
+
+      def test_create_join_table_with_symbol_and_string
+        connection.ddl_batch do
+          connection.create_join_table :artists, "musics"
+        end
+
+        assert_equal %w(artist_id music_id), connection.columns(:artists_musics).map(&:name).sort
+      end
+
+      def test_create_join_table_with_the_proper_order
+        connection.ddl_batch do
+          connection.create_join_table :videos, :musics
+        end
+
+        assert_equal %w(music_id video_id), connection.columns(:musics_videos).map(&:name).sort
+      end
+
+      def test_create_join_table_with_the_table_name
+        connection.ddl_batch do
+          connection.create_join_table :artists, :musics, table_name: :catalog
+        end
+
+        assert_equal %w(artist_id music_id), connection.columns(:catalog).map(&:name).sort
+      end
+
+      def test_create_join_table_with_the_table_name_as_string
+        connection.ddl_batch do
+          connection.create_join_table :artists, :musics, table_name: "catalog"
+        end
+
+        assert_equal %w(artist_id music_id), connection.columns(:catalog).map(&:name).sort
+      end
+
+      def test_create_join_table_with_column_options
+        connection.ddl_batch do
+          connection.create_join_table :artists, :musics, column_options: { null: true }
+        end
+
+        assert_equal [true, true], connection.columns(:artists_musics).map(&:null)
+      end
+
+      def test_create_join_table_without_indexes
+        connection.ddl_batch do
+          connection.create_join_table :artists, :musics
+        end
+
+        assert_predicate connection.indexes(:artists_musics), :blank?
+      end
+
+      def test_create_join_table_with_index
+        connection.ddl_batch do
+          connection.create_join_table :artists, :musics do |t|
+            t.index [:artist_id, :music_id]
+          end
+        end
+
+        assert_equal [%w(artist_id music_id)], connection.indexes(:artists_musics).map(&:columns)
+      end
+
+      def test_create_join_table_respects_reference_key_type
+        connection.ddl_batch do
+          connection.create_join_table :artists, :musics do |t|
+            t.references :video
+          end
+        end
+
+        artist_id, music_id, video_id = connection.columns(:artists_musics).sort_by(&:name)
+
+        assert_equal video_id.sql_type, artist_id.sql_type
+        assert_equal video_id.sql_type, music_id.sql_type
+      end
+
+      def test_drop_join_table
+        connection.ddl_batch do
+          connection.create_join_table :artists, :musics
+        end
+        connection.ddl_batch do
+          connection.drop_join_table :artists, :musics
+        end
+
+        assert_not connection.table_exists?("artists_musics")
+      end
+
+      def test_drop_join_table_with_strings
+        connection.ddl_batch do
+          connection.create_join_table :artists, :musics
+        end
+        connection.ddl_batch do
+          connection.drop_join_table "artists", "musics"
+        end
+
+        assert_not connection.table_exists?("artists_musics")
+      end
+
+      def test_drop_join_table_with_the_proper_order
+        connection.ddl_batch do
+          connection.create_join_table :videos, :musics
+        end
+        connection.ddl_batch do
+          connection.drop_join_table :videos, :musics
+        end
+
+        assert_not connection.table_exists?("musics_videos")
+      end
+
+      def test_drop_join_table_with_the_table_name
+        connection.ddl_batch do
+          connection.create_join_table :artists, :musics, table_name: :catalog
+        end
+        connection.ddl_batch do
+          connection.drop_join_table :artists, :musics, table_name: :catalog
+        end
+
+        assert_not connection.table_exists?("catalog")
+      end
+
+      def test_drop_join_table_with_the_table_name_as_string
+        connection.ddl_batch do
+          connection.create_join_table :artists, :musics, table_name: "catalog"
+        end
+        connection.ddl_batch do
+          connection.drop_join_table :artists, :musics, table_name: "catalog"
+        end
+
+        assert_not connection.table_exists?("catalog")
+      end
+
+      def test_drop_join_table_with_column_options
+        connection.ddl_batch do
+          connection.create_join_table :artists, :musics, column_options: { null: true }
+        end
+        connection.ddl_batch do
+          connection.drop_join_table :artists, :musics, column_options: { null: true }
+        end
+
+        assert_not connection.table_exists?("artists_musics")
+      end
+
+      def test_create_and_drop_join_table_with_common_prefix
+        with_table_cleanup do
+          connection.create_join_table "audio_artists", "audio_musics"
+          assert connection.table_exists?("audio_artists_musics")
+
+          connection.drop_join_table "audio_artists", "audio_musics"
+          assert_not connection.table_exists?("audio_artists_musics"), "Should have dropped join table, but didn't"
+        end
+      end
+
+      private
+      def with_table_cleanup
+        tables_before = connection.data_sources
+
+        yield
+      ensure
+        tables_after = connection.data_sources - tables_before
+
+        connection.ddl_batch do
+          tables_after.each do |table|
+            connection.drop_table table
+          end
+        end
+      end
+    end
+  end
+end

data/acceptance/cases/migration/ddl_batching_test.rb

@@ -0,0 +1,80 @@
+# Copyright 2020 Google LLC
+#
+# Use of this source code is governed by an MIT-style
+# license that can be found in the LICENSE file or at
+# https://opensource.org/licenses/MIT.
+
+# frozen_string_literal: true
+
+require "test_helper"
+
+module ActiveRecord
+  class Migration
+    class DDLBatchingTest < SpannerAdapter::TestCase
+      include SpannerAdapter::Migration::TestHelper
+      include ActiveSupport::Testing::Stream
+
+      class Box < ActiveRecord::Base
+      end
+
+      class CreateBoxMigration < ActiveRecord::Migration::Current
+        def change
+          connection.ddl_batch do
+            create_table("boxes") do |t|
+              t.string :name
+            end
+
+            add_column :boxes, :length, :integer
+          end
+
+          Box.create(name: "Box1", length: 10)
+        end
+      end
+
+      def setup
+        skip_test_table_create!
+
+        super
+      end
+
+      def teardown
+        super
+
+        connection.ddl_batch do
+          [:boxes, :ddl_batch_test].each do |name|
+            if connection.table_exists?(name)
+              connection.drop_table name
+            end
+          end
+        end
+      end
+
+      def test_ddl_batching
+        information_schema = connection.send :information_schema
+
+        connection.ddl_batch do
+          connection.create_table("ddl_batch_test") do |t|
+            t.string :name
+          end
+          connection.add_column :ddl_batch_test, :created_at, :time
+
+          assert_not information_schema.table(:ddl_batch_test)
+        end
+
+        assert information_schema.table(:ddl_batch_test)
+      end
+
+      def test_ddl_batching_with_dml_statement
+        migration = CreateBoxMigration.new
+        silence_stream($stdout) { migration.migrate(:up) }
+
+        assert connection.table_exists?(:boxes)
+        assert connection.column_exists?(:boxes, :length, :integer)
+
+        assert_equal 1, Box.count
+        box = Box.first
+        assert_equal 10, box.length
+      end
+    end
+  end
+end

data/acceptance/cases/migration/foreign_key_test.rb

@@ -0,0 +1,297 @@
+# Copyright 2020 Google LLC
+#
+# Use of this source code is governed by an MIT-style
+# license that can be found in the LICENSE file or at
+# https://opensource.org/licenses/MIT.
+
+# frozen_string_literal: true
+
+require "test_helper"
+
+module ActiveRecord
+  class Migration
+    class ForeignKeyTest < SpannerAdapter::TestCase
+      include SpannerAdapter::Migration::TestHelper
+      include ActiveSupport::Testing::Stream
+
+      class Rocket < ActiveRecord::Base
+      end
+
+      class Astronaut < ActiveRecord::Base
+      end
+
+      def setup
+        skip_test_table_create!
+        super
+
+        connection.ddl_batch do
+          connection.create_table "rockets", force: true do |t|
+            t.string :name
+          end
+
+          connection.create_table "astronauts", force: true do |t|
+            t.string :name
+            t.references :rocket
+          end
+        end
+      end
+
+      def teardown
+        connection.ddl_batch do
+          connection.drop_table "astronauts", if_exists: true
+          connection.drop_table "rockets", if_exists: true
+          connection.drop_table "fk_test_has_fk", if_exists: true
+          connection.drop_table "fk_test_has_pk", if_exists: true
+        end
+      end
+
+      def test_foreign_keys
+        connection.ddl_batch do
+          connection.create_table :fk_test_has_pk, primary_key: "pk_id", force: :cascade do |t|
+          end
+
+          connection.create_table :fk_test_has_fk, force: true do |t|
+            t.references :fk, null: false
+            t.foreign_key :fk_test_has_pk, column: "fk_id", name: "fk_name", primary_key: "pk_id"
+          end
+        end
+
+        foreign_keys = connection.foreign_keys("fk_test_has_fk")
+        assert_equal 1, foreign_keys.size
+
+        fk = foreign_keys.first
+        assert_equal "fk_test_has_fk", fk.from_table
+        assert_equal "fk_test_has_pk", fk.to_table
+        assert_equal "fk_id", fk.column
+        assert_equal "pk_id", fk.primary_key
+      end
+
+      def test_add_foreign_key_inferes_column
+        connection.ddl_batch do
+          connection.add_foreign_key :astronauts, :rockets
+        end
+
+        foreign_keys = connection.foreign_keys("astronauts")
+        assert_equal 1, foreign_keys.size
+
+        fk = foreign_keys.first
+        assert_equal "astronauts", fk.from_table
+        assert_equal "rockets", fk.to_table
+        assert_equal "rocket_id", fk.column
+        assert_equal "id", fk.primary_key
+      end
+
+      def test_add_foreign_key_with_column
+        connection.ddl_batch do
+          connection.add_foreign_key :astronauts, :rockets, column: "rocket_id"
+        end
+
+        foreign_keys = connection.foreign_keys("astronauts")
+        assert_equal 1, foreign_keys.size
+
+        fk = foreign_keys.first
+        assert_equal "astronauts", fk.from_table
+        assert_equal "rockets", fk.to_table
+        assert_equal "rocket_id", fk.column
+        assert_equal "id", fk.primary_key
+      end
+
+      def test_add_foreign_key_with_non_standard_primary_key
+        connection.ddl_batch do
+          connection.create_table :space_shuttles, id: false, force: true do |t|
+            t.integer :pk, primary_key: true
+          end
+
+          connection.add_foreign_key(:astronauts, :space_shuttles,
+            column: "rocket_id", primary_key: "pk", name: "custom_pk")
+        end
+
+        foreign_keys = connection.foreign_keys("astronauts")
+        assert_equal 1, foreign_keys.size
+
+        fk = foreign_keys.first
+        assert_equal "astronauts", fk.from_table
+        assert_equal "space_shuttles", fk.to_table
+        assert_equal "pk", fk.primary_key
+      ensure
+        connection.ddl_batch do
+          connection.remove_foreign_key :astronauts, name: "custom_pk", to_table: "space_shuttles"
+          connection.drop_table :space_shuttles
+        end
+      end
+
+      def test_foreign_key_exists
+        connection.ddl_batch do
+          connection.add_foreign_key :astronauts, :rockets
+        end
+
+        assert connection.foreign_key_exists?(:astronauts, :rockets)
+        assert_not connection.foreign_key_exists?(:astronauts, :stars)
+      end
+
+      def test_foreign_key_exists_by_column
+        connection.ddl_batch do
+          connection.add_foreign_key :astronauts, :rockets, column: "rocket_id"
+        end
+
+        assert connection.foreign_key_exists?(:astronauts, column: "rocket_id")
+        assert_not connection.foreign_key_exists?(:astronauts, column: "star_id")
+      end
+
+      def test_foreign_key_exists_by_name
+        connection.ddl_batch do
+          connection.add_foreign_key :astronauts, :rockets, column: "rocket_id", name: "fancy_named_fk"
+        end
+
+        assert connection.foreign_key_exists?(:astronauts, name: "fancy_named_fk")
+        assert_not connection.foreign_key_exists?(:astronauts, name: "other_fancy_named_fk")
+      end
+
+      def test_foreign_key_exists_in_change_table
+        connection.change_table(:astronauts) do |t|
+          t.foreign_key :rockets, column: "rocket_id", name: "fancy_named_fk"
+
+          assert t.foreign_key_exists?(column: "rocket_id")
+          assert_not t.foreign_key_exists?(column: "star_id")
+        end
+      end
+
+      def test_remove_foreign_key_inferes_column
+        connection.ddl_batch do
+          connection.add_foreign_key :astronauts, :rockets
+        end
+
+        assert_equal 1, connection.foreign_keys("astronauts").size
+        connection.ddl_batch do
+          @connection.remove_foreign_key :astronauts, :rockets
+        end
+        assert_equal [], connection.foreign_keys("astronauts")
+      end
+
+      def test_remove_foreign_key_by_column
+        connection.ddl_batch do
+          connection.add_foreign_key :astronauts, :rockets, column: "rocket_id"
+        end
+
+        assert_equal 1, connection.foreign_keys("astronauts").size
+        connection.ddl_batch do
+          @connection.remove_foreign_key :astronauts, column: "rocket_id"
+        end
+        assert_equal [], connection.foreign_keys("astronauts")
+      end
+
+      def test_remove_foreign_key_by_symbol_column
+        connection.ddl_batch do
+          connection.add_foreign_key :astronauts, :rockets, column: :rocket_id
+        end
+
+        assert_equal 1, connection.foreign_keys("astronauts").size
+        connection.ddl_batch do
+          connection.remove_foreign_key :astronauts, column: :rocket_id
+        end
+        assert_equal [], connection.foreign_keys("astronauts")
+      end
+
+      def test_remove_foreign_key_by_name
+        connection.ddl_batch do
+          connection.add_foreign_key :astronauts, :rockets, column: "rocket_id", name: "fancy_named_fk"
+        end
+
+        assert_equal 1, connection.foreign_keys("astronauts").size
+        connection.ddl_batch do
+          connection.remove_foreign_key :astronauts, name: "fancy_named_fk"
+        end
+        assert_equal [], connection.foreign_keys("astronauts")
+      end
+
+      def test_remove_foreign_non_existing_foreign_key_raises
+        e = assert_raises ArgumentError do
+          connection.remove_foreign_key :astronauts, :rockets
+        end
+        assert_equal "Table 'astronauts' has no foreign key for rockets", e.message
+      end
+
+      def test_remove_foreign_key_by_the_select_one_on_the_same_table
+        connection.ddl_batch do
+          connection.add_foreign_key :astronauts, :rockets
+          connection.add_reference :astronauts, :myrocket, foreign_key: { to_table: :rockets }
+        end
+
+        assert_equal 2, connection.foreign_keys("astronauts").size
+
+        connection.ddl_batch do
+          connection.remove_foreign_key :astronauts, :rockets, column: "myrocket_id"
+        end
+
+        assert_equal [["astronauts", "rockets", "rocket_id"]],
+          connection.foreign_keys("astronauts").map { |fk| [fk.from_table, fk.to_table, fk.column] }
+      end
+
+      class CreateCitiesAndHousesMigration < ActiveRecord::Migration::Current
+        def change
+          connection.ddl_batch do
+            create_table("cities") { |t| }
+
+            create_table("houses") do |t|
+              t.references :city
+            end
+            add_foreign_key :houses, :cities, column: "city_id"
+          end
+
+          # remove and re-add to test that schema is updated and not accidentally cached
+          remove_foreign_key :houses, :cities
+          add_foreign_key :houses, :cities, column: "city_id"
+        end
+      end
+
+      def test_add_foreign_key_is_reversible
+        migration = CreateCitiesAndHousesMigration.new
+        silence_stream($stdout) { migration.migrate(:up) }
+        assert_equal 1, connection.foreign_keys("houses").size
+      ensure
+        silence_stream($stdout) { migration.migrate(:down) }
+      end
+
+      class CreateSchoolsAndClassesMigration < ActiveRecord::Migration::Current
+        def up
+          connection.ddl_batch do
+            create_table(:schools)
+
+            create_table(:classes) do |t|
+              t.references :school
+            end
+            add_foreign_key :classes, :schools
+          end
+        end
+
+        def down
+          connection.ddl_batch do
+            drop_table :classes, if_exists: true
+            drop_table :schools, if_exists: true
+          end
+        end
+      end
+
+      def test_add_foreign_key_with_prefix
+        ActiveRecord::Base.table_name_prefix = "p_"
+        migration = CreateSchoolsAndClassesMigration.new
+        silence_stream($stdout) { migration.migrate(:up) }
+        assert_equal 1, connection.foreign_keys("p_classes").size
+      ensure
+        silence_stream($stdout) { migration.migrate(:down) }
+        ActiveRecord::Base.table_name_prefix = nil
+      end
+
+      def test_add_foreign_key_with_suffix
+        ActiveRecord::Base.table_name_suffix = "_s"
+        migration = CreateSchoolsAndClassesMigration.new
+        silence_stream($stdout) { migration.migrate(:up) }
+        assert_equal 1, connection.foreign_keys("classes_s").size
+      ensure
+        silence_stream($stdout) { migration.migrate(:down) }
+        ActiveRecord::Base.table_name_suffix = nil
+      end
+    end
+  end
+end
+