temporary_model 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a1dd9e5df7a2d28a160b195643174d4a196ca06e0e60c8da2d3888033e27249a
-  data.tar.gz: b1d709cc0f71716ee2e139de14f4e4c3d91222b905b83c15b7e11c2470998a2b
+  metadata.gz: 630b4c206db73a2140031839721833e1d06205772c4a162293d40aa48171dc75
+  data.tar.gz: 72b4939a7e14aaf4872d2cb00a6a529f20bdbd9a4238c532ad404fe0ba2bf667
 SHA512:
-  metadata.gz: 42d762e3ff3585c3b5860af4db223b0c94c45054666915bc759c4c933838377922b7cd9d462daeab17125c692dc15ae0970e84a7df95c58ad8b82985531b4cd1
-  data.tar.gz: 0b66e0d9587d61941fa82a496328820b43324a52b5287560b1d2e01d342f31aab3e16570798878bab18a77cbaf8161e0f2c561a05bd16e380bb04099b3cb787b
+  metadata.gz: 6fb34bee99a0323b765506ea04382ecf6b8aaaae64c32c8dd141e973b6f8b5705f956d40ece9d756479f66e159cccb1662e4fba8a492badd692fc20d072907ce
+  data.tar.gz: fd3790a3958adb022e85f347218e83b28f8187bc5039855d9c6c2e1bf25a74d0008ad13283e99d139264545682f7ad11a48e10cdeefbc36afd46f2955e9c925a

data/lib/temporary_model/test_helper.rb

@@ -3,28 +3,29 @@ module TemporaryModel::TestHelper
 
   included do
     def self.temporary_classes
-      @temporary_classes ||= []
+      @temporary_classes ||= {}
     end
 
     def self.temporary_model(model_name, &block)
       raise(ArgumentError, "#{model_name} has already been defined") if Object.const_defined?(model_name)
-      klass = Class.new(TemporaryModel::Record, &block)
-      Object.const_set(model_name, klass)
-      temporary_classes.push(klass)
+      temporary_classes[model_name] = Class.new(TemporaryModel::Record, &block)
     end
   end
 
   # Override ActiveSupport::TestCase#run
+  # FIXME: こんなにループさせんでもできるんじゃないか？
+  # FIXME: テストごとに設定じゃなくてテストクラス全体でできないかな
   def run
-    self.class.temporary_classes.each do |temporary_class|
+    self.class.temporary_classes.each do |model_name, temporary_class|
+      Object.const_set(model_name, temporary_class)
       create_temporary_table temporary_class.table_name, &temporary_class.define_table
     end
     super
   ensure
     TemporaryModel::Record.connection.disable_referential_integrity do
-      self.class.temporary_classes.each do |temporary_class|
+      self.class.temporary_classes.each do |model_name, temporary_class|
         drop_temporary_table temporary_class.table_name
-        Object.send(:remove_const, temporary_class.name)
+        Object.send(:remove_const, model_name)
       end
     end
   end

data/lib/temporary_model/version.rb

@@ -1,3 +1,3 @@
 module TemporaryModel
-  VERSION = '0.1.0'
+  VERSION = '0.2.0'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: temporary_model
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Yuichi Takeuchi