@@ -1,2 +1,4 @@ pkg *.gem +*.log +*.db .gitignore @@ -1,3 +1,5 @@ +require 'activerecord' + module Schemer def schema(*args) extend ClassMethods @@ -8,27 +10,39 @@ module Schemer end module ClassMethods + # Columns which we don't touch regardless of not being defined in schema + # (and are ignored if they are defined in schema) def protected_columns %w( id ) end + + # Create the underlying table for this class + def create_table + ActiveRecord::Migration.create_table(table_name) do |t|;end; + end + + # Update ActiveRecord's automatically generated methods so we don't have to + # reload for schema changes to take effect + def update_methods + generated_methods.each { |method| remove_method(method) } + @columns = @column_names = @columns_hash = @generated_methods = nil + end + + # Update the underlying schema as defined by schema call + def update_schema + create_table unless table_exists? + (schema_columns - column_names).each { |column| ActiveRecord::Migration.add_column(table_name, column, :string) } + (column_names - protected_columns - schema_columns).each { |column| ActiveRecord::Migration.remove_column(table_name, column) } + end end module InstanceMethods def initialize(*args) - update_schema + self.class.update_schema + self.class.update_methods super end - - private - - def update_schema - ActiveRecord::Migration.create_table(self.class.table_name) do |t|;end; unless self.class.table_exists? - self.class.schema_columns.each do |column| - ActiveRecord::Migration.add_column(self.class.table_name, column, :string) unless respond_to?(column.to_sym) - (self.class.column_names - self.class.protected_columns).each do |column| - ActiveRecord::Migration.remove_column(self.class.table_name, column) unless self.class.schema_columns.include?(column.to_s) - end - end - end end end + +ActiveRecord::Base.extend(Schemer) lib/schemer.rb @@ -1,3 +1,2 @@ +raise "We highly suggest you only use Schemer in development mode!" unless Rails.env.to_s == 'development' require 'schemer' -raise "We would highly suggest you only use Schemer in development mode!" unless Rails.env.to_s == 'development' -ActiveRecord::Base.extend(Schemer) \ No newline at end of file rails/init.rb @@ -9,7 +9,7 @@ Gem::Specification.new do |s| s.specification_version = 2 if s.respond_to? :specification_version= - s.files = <%= FileList['lib/**/*.rb', 'README*', 'LICENSE', 'Rakefile', 'rails/**/*', 'test/**/*.*'].inspect %> + s.files = <%= FileList['lib/**/*.rb', 'README*', 'LICENSE', 'Rakefile', 'rails/**/*', 'test/**/*.rb'].inspect %> s.require_paths = ['lib'] schemer.gemspec.erb @@ -1,5 +1,43 @@ -require File.dirname(__FILE__) + "/../lib/contest" +require 'rubygems' +require 'contest' +require File.join(File.dirname(__FILE__), '..', 'lib', 'schemer') -class FooTest < Contest::Unit::TestCase - # coming soon, tests! +ActiveRecord::Base.establish_connection :adapter => 'sqlite3', :database => 'test/schemer_test.db' +ActiveRecord::Base.logger = Logger.new(File.join(File.dirname(__FILE__), "debug.log")) + +class Foo < ActiveRecord::Base;end; +ActiveRecord::Migration.drop_table(Foo.table_name) if Foo.table_exists? + +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 + + should "create the bar column" do + assert @foo.respond_to?(:bar) + end + end + + context "updating the schema" do + setup do + Foo.schema :foo, :bar + @foo = Foo.new + Foo.schema :foo + @foo = Foo.new + end + + should "remove the bar column" do + assert !@foo.respond_to?(:bar) + end + end end test/schemer_test.rb a16894a402f67a4d6bf657b73c25227c99e7a753 benalavi benalavi@gmail.com http://github.com/citrusbyte/schemer/commit/3d22e9c15fcc5034e7d74e0f2b2a52dee288cb31 3d22e9c15fcc5034e7d74e0f2b2a52dee288cb31 2009-04-03T18:50:24-07:00 2009-04-03T18:50:24-07:00 added tests, refactored quite a bit, fixed some bugs 76adbee9b08ed4fbb2e807ee306af7e771609fcd benalavi benalavi@gmail.com