activerecord-spanner-adapter 0.3.0 → 1.0.0

data/acceptance/cases/type/numeric_test.rb

@@ -0,0 +1,27 @@
+# Copyright 2021 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
+  module Type
+    class NumericTest < SpannerAdapter::TestCase
+      include SpannerAdapter::Types::TestHelper
+
+      def test_convert_to_sql_type
+        assert_equal "NUMERIC", connection.type_to_sql(:numeric)
+      end
+
+      def test_set_numeric_value_in_create
+        record = TestTypeModel.create(price: 9750.99)
+        record.reload
+        assert_equal 9750.99, record.price
+      end
+    end
+  end
+end

data/acceptance/cases/type/string_test.rb

@@ -0,0 +1,79 @@
+# 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
+  module Type
+    class StringTest < SpannerAdapter::TestCase
+      include SpannerAdapter::Types::TestHelper
+
+      def test_convert_to_sql_type
+        assert_equal "STRING(MAX)", connection.type_to_sql(:string)
+        assert_equal "STRING(1024)", connection.type_to_sql(:string, limit: 1024)
+      end
+
+      def test_set_string_in_create
+        description = "a" * 1000
+        name = "Test name"
+        record = TestTypeModel.create(description: description, name: name)
+        record.reload
+
+        assert_equal description, record.description
+        assert_equal name, record.name
+      end
+
+      def test_set_string_with_max_limit_in_create
+        str = "a" * 256
+
+        assert_raise(ActiveRecord::StatementInvalid) {
+          TestTypeModel.create(name: str)
+        }
+      end
+
+      def test_escape_special_charaters_and_save
+        str = [
+          "Newline \n",
+          "`Backtick`",
+          "'Quote'",
+          "Bell \a",
+          "Backspace \b",
+          "Formfeed \f",
+          "Carriage Return \r",
+          "Tab \t",
+          "Vertical Tab \v",
+          "Backslash \\",
+          "Question Mark \?",
+          "Double Quote \"",
+        ].join (" ")
+
+        record = TestTypeModel.new(description: str)
+        assert_equal str, record.description
+
+        record.save!
+        assert_equal str, record.description
+
+        record.reload
+        assert_equal str, record.description
+      end
+
+      def test_save_special_charaters
+        str = "Hello Seocial Chars : € à ö ¿ © 😎"
+
+        record = TestTypeModel.new(description: str)
+        assert_equal str, record.description
+
+        record.save!
+        assert_equal str, record.description
+
+        record.reload
+        assert_equal str, record.description
+      end
+    end
+  end
+end

data/acceptance/cases/type/text_test.rb

@@ -0,0 +1,30 @@
+# 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
+  module Type
+    class TextTest < SpannerAdapter::TestCase
+      include SpannerAdapter::Types::TestHelper
+
+      def test_convert_to_sql_type
+        assert_equal "STRING(MAX)", connection.type_to_sql(:text)
+        assert_equal "STRING(1024)", connection.type_to_sql(:text, limit: 1024)
+      end
+
+      def test_set_text_in_create
+        text = "a" * 1000
+        record = TestTypeModel.create(bio: text)
+        record.reload
+
+        assert_equal text, record.bio
+      end
+    end
+  end
+end

data/acceptance/cases/type/time_test.rb

@@ -0,0 +1,87 @@
+# 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
+  module Type
+    class TimeTest < SpannerAdapter::TestCase
+      include SpannerAdapter::Types::TestHelper
+
+      def test_convert_to_sql_type
+        assert_equal "TIMESTAMP", connection.type_to_sql(:time)
+      end
+
+      def test_assign_time
+        expected_time = ::Time.now
+        record = TestTypeModel.new start_time: expected_time
+
+        assert_equal expected_time, record.start_time
+      end
+
+      def test_assign_empty_time
+        record = TestTypeModel.new start_time: ""
+        assert_nil record.start_time
+      end
+
+      def test_assign_nil_time
+        record = TestTypeModel.new start_time: nil
+        assert_nil record.start_time
+      end
+
+      def test_set_and_save_time
+        expected_time = ::Time.now.utc
+        record = TestTypeModel.create! start_time: expected_time
+
+        assert_equal expected_time, record.start_time
+
+        record.reload
+        assert_equal expected_time, record.start_time
+      end
+
+      def test_date_time_string_value_with_subsecond_precision
+        expected_time = ::Time.local 2017, 07, 4, 14, 19, 10, 897761
+
+        string_value = "2017-07-04 14:19:10.897761"
+
+        record = TestTypeModel.new start_time: string_value
+        assert_equal expected_time, record.start_time.utc
+
+        record.save!
+        assert_equal expected_time, record.start_time.utc
+
+        assert_equal record, TestTypeModel.find_by(start_time: string_value)
+      end
+
+      def test_date_time_with_string_value_with_non_iso_format
+        string_value = "04/07/2017 2:19pm"
+        expected_time = ::Time.local 2017, 07, 4, 14, 19
+
+        record = TestTypeModel.new start_time: string_value
+        assert_equal expected_time, record.start_time
+
+        record.save!
+        assert_equal expected_time, record.start_time.utc
+
+        assert_equal record, TestTypeModel.find_by(start_time: string_value)
+      end
+
+      def test_default_year_is_correct
+        expected_time = ::Time.utc(2000, 1, 1, 10, 30, 0)
+        record = TestTypeModel.new start_time: { 4 => 10, 5 => 30 }
+
+        assert_equal expected_time, record.start_time
+
+        record.save!
+        record.reload
+
+        assert_equal expected_time, record.start_time
+      end
+    end
+  end
+end

data/acceptance/models/account.rb

@@ -0,0 +1,13 @@
+# 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.
+
+class Account < ActiveRecord::Base
+  belongs_to :firm
+  belongs_to :customer
+  has_many :transactions
+
+  alias_attribute :available_credit, :credit_limit
+end

data/acceptance/models/address.rb

@@ -0,0 +1,9 @@
+# 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.
+
+class Address < ActiveRecord::Base
+  has_one :author
+end

data/acceptance/models/album.rb

@@ -0,0 +1,12 @@
+# Copyright 2021 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.
+
+class Album < ActiveRecord::Base
+  # The relationship with singer is not really a foreign key, but an INTERLEAVE IN relationship. We still need to
+  # use the `foreign_key` attribute to indicate which column to use for the relationship.
+  belongs_to :singer, foreign_key: "singerid"
+  has_many :tracks, foreign_key: "albumid", dependent: :delete_all
+end

data/acceptance/models/all_types.rb

@@ -0,0 +1,8 @@
+# Copyright 2021 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.
+
+class AllTypes < ActiveRecord::Base
+end

data/acceptance/models/author.rb

@@ -0,0 +1,11 @@
+# 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.
+
+class Author < ActiveRecord::Base
+  has_many :posts
+  has_many :comments, through: :posts
+  belongs_to :organization
+end

data/acceptance/models/club.rb

@@ -0,0 +1,12 @@
+# 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.
+
+class Club < ActiveRecord::Base
+  has_many :memberships
+  has_many :members, through: :memberships
+  has_many :favourites, -> { where(memberships: { favourite: true }) },
+           through: :memberships, source: :member
+end

data/acceptance/models/comment.rb

@@ -0,0 +1,9 @@
+# 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.
+
+class Comment < ActiveRecord::Base
+  belongs_to :post, counter_cache: true
+end

data/acceptance/models/customer.rb

@@ -0,0 +1,9 @@
+# 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.
+
+class Customer < ActiveRecord::Base
+  has_many :accounts
+end

data/acceptance/models/department.rb

@@ -0,0 +1,9 @@
+# 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.
+
+class Department < ActiveRecord::Base
+  belongs_to :resource, polymorphic: true
+end

data/acceptance/models/firm.rb

@@ -0,0 +1,10 @@
+# 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.
+
+class Firm < ActiveRecord::Base
+  has_one :account
+  has_many :departments, as: :resource
+end

data/acceptance/models/member.rb

@@ -0,0 +1,13 @@
+# 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.
+
+class Member < ActiveRecord::Base
+  has_one :membership
+  has_one :club, through: :membership
+  has_one :favourite_club, -> { where "memberships.favourite = ?", true },
+          through: :membership, source: :club
+  belongs_to :member_type
+end

data/acceptance/models/member_type.rb

@@ -0,0 +1,9 @@
+# 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.
+
+class MemberType < ActiveRecord::Base
+  has_many :members
+end

data/acceptance/models/membership.rb

@@ -0,0 +1,10 @@
+# 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.
+
+class Membership < ActiveRecord::Base
+  belongs_to :member
+  belongs_to :club
+end

data/acceptance/models/organization.rb

@@ -0,0 +1,9 @@
+# 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.
+
+class Organization < ActiveRecord::Base
+  has_many :authors, dependent: :destroy
+end

data/acceptance/models/post.rb

@@ -0,0 +1,10 @@
+# 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.
+
+class Post < ActiveRecord::Base
+  belongs_to :author
+  has_many :comments
+end

data/acceptance/models/singer.rb

@@ -0,0 +1,10 @@
+# Copyright 2021 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.
+
+class Singer < ActiveRecord::Base
+  has_many :albums, foreign_key: "singerid", dependent: :delete_all
+  has_many :tracks, foreign_key: "singerid"
+end

data/acceptance/models/track.rb

@@ -0,0 +1,20 @@
+# Copyright 2021 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.
+
+class Track < ActiveRecord::Base
+  belongs_to :album, foreign_key: "albumid"
+  belongs_to :singer, foreign_key: "singerid", counter_cache: true
+
+  def initialize attributes = nil
+    super
+    self.singer ||= self.album&.singer
+  end
+
+  def album=value
+    super
+    self.singer = value&.singer
+  end
+end

data/acceptance/models/transaction.rb

@@ -0,0 +1,9 @@
+# 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.
+
+class Transaction < ActiveRecord::Base
+  belongs_to :account, counter_cache: true
+end

data/acceptance/schema/schema.rb

@@ -0,0 +1,147 @@
+# 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
+
+ActiveRecord::Schema.define do
+  ActiveRecord::Base.connection.ddl_batch do
+    create_table :all_types do |t|
+      t.column :col_string, :string
+      t.column :col_int64, :bigint
+      t.column :col_float64, :float
+      t.column :col_numeric, :numeric
+      t.column :col_bool, :boolean
+      t.column :col_bytes, :binary
+      t.column :col_date, :date
+      t.column :col_timestamp, :datetime
+      t.column :col_json, :json unless ENV["SPANNER_EMULATOR_HOST"]
+      t.column :col_json, :string if ENV["SPANNER_EMULATOR_HOST"]
+
+      t.column :col_array_string, :string, array: true
+      t.column :col_array_int64, :bigint, array: true
+      t.column :col_array_float64, :float, array: true
+      t.column :col_array_numeric, :numeric, array: true
+      t.column :col_array_bool, :boolean, array: true
+      t.column :col_array_bytes, :binary, array: true
+      t.column :col_array_date, :date, array: true
+      t.column :col_array_timestamp, :datetime, array: true
+      t.column :col_array_json, :json, array: true unless ENV["SPANNER_EMULATOR_HOST"]
+      t.column :col_array_json, :string, array: true if ENV["SPANNER_EMULATOR_HOST"]
+    end
+
+    create_table :firms do |t|
+      t.string  :name
+      t.integer :rating
+      t.string :description
+      t.references :account
+    end
+
+    create_table :customers do |t|
+      t.string  :name
+    end
+
+    create_table :accounts do |t|
+      t.references :customer, index: false
+      t.references :firm, index: false
+      t.string  :name
+      t.integer :credit_limit
+      t.integer :transactions_count
+    end
+
+    create_table :transactions do |t|
+      t.float :amount
+      t.references :account, index: false
+    end
+
+    create_table :departments do |t|
+      t.string :name
+      t.references :resource, polymorphic: true
+    end
+
+    create_table :member_types do |t|
+      t.string :name
+    end
+
+    create_table :members do |t|
+      t.string :name
+      t.references :member_type, index: false
+      t.references :admittable, polymorphic: true, index: false
+    end
+
+    create_table :memberships do |t|
+      t.datetime :joined_on
+      t.references :club, index: false
+      t.references :member, index: false
+      t.boolean :favourite
+    end
+
+    create_table :clubs do |t|
+      t.string :name
+    end
+
+    create_table :authors do |t|
+      t.string :name, null: false
+      t.date :registered_date
+      t.references :organization, index: false
+    end
+
+    create_table :posts do |t|
+      t.string :title
+      t.string :content
+      t.references :author
+      t.integer :comments_count
+      t.date :post_date
+      t.time :published_time
+    end
+
+    create_table :comments do |t|
+      t.string :comment
+      t.references :post, index: false, foreign_key: true
+    end
+
+    create_table :addresses do |t|
+      t.string :line1
+      t.string :postal_code
+      t.string :city
+      t.references :author, index: false
+    end
+
+    create_table :organizations do |t|
+      t.string :name
+      t.datetime :last_updated, allow_commit_timestamp: true
+    end
+
+    create_table :singers, id: false do |t|
+      t.primary_key :singerid
+      t.column :first_name, :string, limit: 200
+      t.string :last_name
+      t.integer :tracks_count
+      t.integer :lock_version
+      t.string :full_name, as: "COALESCE(first_name || ' ', '') || last_name", stored: true
+    end
+
+    create_table :albums, id: false do |t|
+      t.interleave_in :singers
+      t.primary_key :albumid
+      # `singerid` is part of the primary key in the table definition, but it is not visible to ActiveRecord as part of
+      # the primary key, to prevent ActiveRecord from considering this to be an entity with a composite primary key.
+      t.parent_key :singerid
+      t.string :title
+      t.integer :lock_version
+    end
+
+    create_table :tracks, id: false do |t|
+      # `:cascade` causes all tracks that belong to an album to automatically be deleted when an album is deleted.
+      t.interleave_in :albums, :cascade
+      t.primary_key :trackid
+      t.parent_key :singerid
+      t.parent_key :albumid
+      t.string :title
+      t.numeric :duration
+      t.integer :lock_version
+    end
+  end
+end