citrusbyte-schemer 0.0.5 → 0.0.6

data/lib/schemer.rb

@@ -2,11 +2,13 @@ require 'activerecord'
 
 module Schemer
   def schema(*args)
-    extend  ClassMethods
-    include InstanceMethods
+    extend ClassMethods
     
     class_inheritable_accessor :schema_columns
     self.schema_columns = args.collect(&:to_s)
+
+    update_schema
+    update_methods
   end
 
   module ClassMethods
@@ -35,14 +37,6 @@ module Schemer
       (column_names - protected_columns - schema_columns).each { |column| ActiveRecord::Migration.remove_column(table_name, column) }
     end    
   end
-  
-  module InstanceMethods
-    def initialize(*args)
-      self.class.update_schema
-      self.class.update_methods
-      super
-    end
-  end
 end
 
 ActiveRecord::Base.extend(Schemer)

data/test/schemer_test.rb

@@ -12,26 +12,30 @@ class FooTest < Test::Unit::TestCase
   context "defining the schema" do
     setup do
       Foo.schema :foo, :bar
-      @foo = Foo.new
     end
     
     should "create the foos table" do
       assert Foo.table_exists?
     end
     
-    should "create the foo column" do
-      assert @foo.respond_to?(:foo)
-    end
+    context "with an instance" do
+      setup do
+        @foo = Foo.new
+      end
+      
+      should "create the foo column" do
+        assert @foo.respond_to?(:foo)
+      end
 
-    should "create the bar column" do
-      assert @foo.respond_to?(:bar)
-    end
+      should "create the bar column" do
+        assert @foo.respond_to?(:bar)
+      end
+    end    
   end
   
   context "updating the schema" do
     setup do
       Foo.schema :foo, :bar
-      @foo = Foo.new
       Foo.schema :foo
       @foo = Foo.new
     end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: citrusbyte-schemer
 version: !ruby/object:Gem::Version 
-  version: 0.0.5
+  version: 0.0.6
 platform: ruby
 authors: 
 - Ben Alavi
@@ -22,10 +22,10 @@ extensions: []
 extra_rdoc_files: []
 
 files: 
-- lib/schemer.rb
 - README.markdown
 - LICENSE
 - Rakefile
+- lib/schemer.rb
 - rails/init.rb
 - test/schemer_test.rb
 has_rdoc: false