add mysql record

Author: seuros

test/dummy/app/models/mysql_label.rb

@@ -0,0 +1,4 @@
+# frozen_string_literal: true
+
+class MysqlLabel < MysqlRecord
+end

test/dummy/app/models/mysql_record.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+class MysqlRecord < ActiveRecord::Base
+  self.abstract_class = true
+  if ActiveRecord::Base.configurations.configs_for(env_name: 'test', name: 'secondary')
+    establish_connection :secondary
+  end
+end

test/dummy/app/models/mysql_tag.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+class MysqlTag < MysqlRecord
+  after_save do
+    MysqlTagAudit.create(tag_name: name)
+    MysqlLabel.create(name: name)
+  end
+end

test/dummy/app/models/mysql_tag_audit.rb

@@ -0,0 +1,4 @@
+# frozen_string_literal: true
+
+class MysqlTagAudit < MysqlRecord
+end

test/dummy/config/database.yml

@@ -1,11 +1,12 @@
 default: &default
-  url: <%= ENV['DATABASE_URL'] %>
   pool: 20
   properties:
     allowPublicKeyRetrieval: true
 
-development:
-  <<: *default
-
 test:
-  <<: *default
+  primary:
+    <<: *default
+    url: <%= ENV['DATABASE_URL_PG'] %>
+  secondary:
+    <<: *default
+    url: <%= ENV['DATABASE_URL_MYSQL'] %>

test/test_helper.rb

@@ -8,8 +8,6 @@
 require 'tmpdir'
 require 'securerandom'
 
-ENV['DATABASE_URL'] ||= ENV['DATABASE_URL_PG'] || ENV['DATABASE_URL_MYSQL']
-
 db_config_path = File.expand_path('dummy/config/database.yml', __dir__)
 db_config      = YAML.load(ERB.new(File.read(db_config_path)).result, aliases: true)
 ActiveRecord::Base.configurations = ActiveRecord::DatabaseConfigurations.new(db_config)
@@ -18,10 +16,17 @@
 
 ENV['WITH_ADVISORY_LOCK_PREFIX'] ||= SecureRandom.hex
 
-ActiveRecord::Base.establish_connection(:test)
+ActiveRecord::Base.establish_connection(:primary)
 
 load File.expand_path('dummy/db/schema.rb', __dir__)
 
+require_relative 'dummy/app/models/mysql_record'
+if MysqlRecord.connected?
+  ActiveRecord::Base.establish_connection(:secondary)
+  load File.expand_path('dummy/db/schema.rb', __dir__)
+  ActiveRecord::Base.establish_connection(:primary)
+end
+
 def env_db
   @env_db ||= ActiveRecord::Base.connection_db_config.adapter.to_sym
 end
@@ -32,6 +37,9 @@ def env_db
 require_relative 'dummy/app/models/tag'
 require_relative 'dummy/app/models/tag_audit'
 require_relative 'dummy/app/models/label'
+require_relative 'dummy/app/models/mysql_tag'
+require_relative 'dummy/app/models/mysql_tag_audit'
+require_relative 'dummy/app/models/mysql_label'
 require 'minitest'
 require 'maxitest/autorun'
 require 'mocha/minitest'
@@ -51,6 +59,13 @@ def is_postgresql_adapter?; adapter_support.postgresql?; end
       TagAudit.table_name,
       Label.table_name
     )
+    if MysqlRecord.connected?
+      MysqlRecord.connection.truncate_tables(
+        MysqlTag.table_name,
+        MysqlTagAudit.table_name,
+        MysqlLabel.table_name
+      )
+    end
   end
 
 end