move one_of from TypeBuilder to StoreModel itself

DmitryTsepelev · Jun 3, 2020 · 9f06ee6 · 9f06ee6
1 parent c283de3
commit 9f06ee6
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/lib/store_model.rb b/lib/store_model.rb
@@ -10,5 +10,13 @@ class << self
     def config
       @config ||= Configuration.new
     end
+
+    # TODO: add documentation
+    def one_of(&block)
+      Class.new do
+        define_singleton_method(:to_type) { Types::PolymorphicType.new(block) }
+        define_singleton_method(:to_array_type) { Types::PolymorphicArrayType.new(block) }
+      end
+    end
   end
 end
diff --git a/lib/store_model/type_builders.rb b/lib/store_model/type_builders.rb
@@ -14,13 +14,5 @@ def to_type
     def to_array_type
       Types::ArrayType.new(self)
     end
-
-    # TODO: add documentation
-    def one_of(&block)
-      Class.new do
-        define_singleton_method(:to_type) { Types::PolymorphicType.new(block) }
-        define_singleton_method(:to_array_type) { }
-      end
-    end
   end
 end