mongify 0.1.0 → 0.1.1

data/CHANGELOG.rdoc

@@ -1,3 +1,9 @@
+== 0.1.1 / 28 Jan 2011
+* Renamed GenerateDatabase to DatabaseGenerator
+* Renamed Table#embed? to Table#embedded?
+* Renamed Table#embed_as_object? to Table#embedded_as_object?
+* Added support for polymorphic tables
+* Did some file cleanup for unused files
 == 0.1.0 / 21 Jan 2011
 * Moved from Alpha to Beta! :)
 * Improved internal RDocs

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    mongify (0.1.0)
+    mongify (0.1.1)
       activerecord (>= 2.3.10)
       activesupport (>= 2.3.10)
       bson_ext (>= 1.1.5)

data/README.rdoc

@@ -143,11 +143,18 @@ Table Options are as follow:
   table "table_name", :ignore => true                       # This will ignore the whole table (like it doesn't exist)
                                                             # This option is good for tables like: schema_migrations
                                                             
+  table "table_name",                                       # This allows you to specify the table as being polymorphic
+    :polymorphic => 'notable',                              # and provide the name of the polymorphic relationship.
+    :embed_in => true                                       # Setting embed_in => true allows the relationship to be
+                                                            # embedded directly into the parent class.
+                                                            # If you do not embed it, the polymorphic table will be copied in to
+                                                            # MongoDB and the notable_id will be updated to the new BSON::ObjectID
+                                                            
   table "table_name" do                                     # A table can take a before_save block that will be called just
     before_save do |row|                                    # before the row is saved to the no sql database.
       row.admin = row.delete('permission').to_i > 50        # This gives you the ability to do very powerful things like:
     end                                                     # Moving records around, renaming records, changing values in row based on
-  end                                                       # some values! Checkout {Mongify::Database::DataRow} to learn more
+  end                                                       # some values! Checkout Mongify::Database::DataRow to learn more
 
 === Columns
 
@@ -196,9 +203,7 @@ More testing will be done once the basic functionality is done
 == TODO
 
 * Allow you to set a :force => true for mongodb_connection, so that it drops existing database before importing
-* Work with polymorphic tables
 * Update user on what is being processed
-* Improve RDoc
 
 == Development
 

data/features/process.feature

@@ -6,7 +6,7 @@ Feature: Processing a translation
   Scenario: Process
   Given a database exists
   And a blank mongodb
-  When I run mongify process spec/files/simple_translation.rb -c spec/files/base_configuration.rb
+  When I run mongify process spec/files/translation.rb -c spec/files/base_configuration.rb
   Then it succeeds
   And there should be 3 users in mongodb
   And there should be 3 posts in mongodb

data/features/step_definitions/mongify_steps.rb

@@ -34,7 +34,7 @@ end
 
 
 Given /^a database exists$/ do
-  GenerateDatabase.sqlite
+  DatabaseGenerator.sqlite
 end
 
 Then /^it should print out the database schema$/ do

data/features/step_definitions/mongo_steps.rb

@@ -1,15 +1,15 @@
 Given /^a blank mongodb$/ do
-  GenerateDatabase.clear_mongodb
+  DatabaseGenerator.clear_mongodb
 end
 
 Then /^there should be (\d+) (.*) in mongodb$/ do |count, collection|
-  GenerateDatabase.mongo_connection.db[collection].count.should == count.to_i
+  DatabaseGenerator.mongo_connection.db[collection].count.should == count.to_i
 end
 
 Then /^the (first|sencond|third) (.+)'s (.+) should be (first|second|thrid) (.+)$/ do |collection_place, collection, field, target_place, target|
-  GenerateDatabase.mongo_connection.db[collection.pluralize].find.to_a.send(collection_place.to_sym)[field].should == GenerateDatabase.mongo_connection.db[target.pluralize].find.to_a.send(target_place)['_id']
+  DatabaseGenerator.mongo_connection.db[collection.pluralize].find.to_a.send(collection_place.to_sym)[field].should == DatabaseGenerator.mongo_connection.db[target.pluralize].find.to_a.send(target_place)['_id']
 end
 
 Then /^the (first|sencond|third) (.+) should have (\d+) (.+)$/ do |collection_place, collection, count, target|
-  GenerateDatabase.mongo_connection.db[collection.pluralize].find.to_a.send(collection_place.to_sym)[target.pluralize].count == count
+  DatabaseGenerator.mongo_connection.db[collection.pluralize].find.to_a.send(collection_place.to_sym)[target.pluralize].count == count
 end

data/features/support/env.rb

@@ -5,7 +5,7 @@ require 'tempfile'
 require 'fileutils'
 require 'mongify/cli/application'
 require 'mongify'
-require 'spec/support/generate_database'
+require 'spec/support/database_generator'
 require 'spec/support/config_reader'
 
 ::CONNECTION_CONFIG = ConfigReader.new(File.dirname(File.dirname(File.dirname(File.expand_path(__FILE__)))) + '/spec/support/database.yml')

data/lib/mongify/database/table.rb

@@ -32,6 +32,14 @@ module Mongify
     #   table "table_name", :ignore => true                       # This will ignore the whole table (like it doesn't exist)
     #                                                             # This option is good for tables like: schema_migrations
     # 
+    #   table "table_name", :polymorphic => 'commentable'         # This will identfiy the table as polymorphic and the new ids
+    #                                                             # will be updated as needed.
+    # 
+    #   table "table_name",                                       # This allows you to specify the table as being polymorphic
+    #     :polymorphic => 'notable',                              # and provide the name of the polymorphic relationship.
+    #     :embed_in => true                                       # Setting embed_in => true allows the relationship to be
+    #                                                             # embedded directly into the parent class.
+    #
     #   table "table_name" do                                     # A table can take a before_save block that will be called just
     #     before_save do |row|                                    # before the row is saved to the no sql database.
     #       row.admin = row.delete('permission').to_i > 50        # This gives you the ability to do very powerful things like:
@@ -66,6 +74,16 @@ module Mongify
         @options['ignore']
       end
       
+      # Returns true if table is marked as polymorphic
+      def polymorphic?
+        !!@options['polymorphic']
+      end
+      
+      # Returns the name of the polymorphic association
+      def polymorphic_as
+        @options['polymorphic'].to_s
+      end
+      
       # Add a Database Column to the table
       # This expects to get a {Mongify::Database::Column} or it will raise {Mongify::DatabaseColumnExpected} otherwise
       def add_column(column)
@@ -110,24 +128,24 @@ module Mongify
       
       # Returns the type of embed it will be [object or array]
       def embed_as
-        return nil unless embed?
+        return nil unless embedded?
         return 'object' if @options['as'].to_s.downcase == 'object'
         'array'
       end
 
       # Returns true if table is being embed as an object
-      def embed_as_object?
+      def embedded_as_object?
         embed_as == 'object'
       end
       
       # Returns true if this is an embedded table
-      def embed?
+      def embedded?
         embed_in.present?
       end
       
       # Returns the name of the target column to embed on
       def embed_on
-        return nil unless embed?
+        return nil unless embedded?
         (@options['on'] || "#{@options['embed_in'].to_s.singularize}_id").to_s
       end
       

data/lib/mongify/translation.rb

@@ -69,6 +69,11 @@ module Mongify
       @all_tables = []
     end
     
+    # finds table by name
+    def find(name)
+      all_tables.find{ |t| t.name == name }
+    end
+    
     # Creates a {Mongify::Database::Table} from the given input and adds it to the list of tables
     def table(table_name, options={}, &block)
       table = Mongify::Database::Table.new(table_name, options, &block)
@@ -88,17 +93,21 @@ module Mongify
     
     # Returns an array of all tables that have not been ingored
     def tables
-      all_tables.reject{ |t| t.ignored? }
+      all_tables.reject{ |t| t.ignored? || t.polymorphic? }
     end
     
     # Returns an array of all tables that have not been ignored and are just straight copy tables
     def copy_tables
-      tables.reject{|t| t.embed?}
+      tables.reject{|t| t.embedded?}
+    end
+    
+    def polymorphic_tables
+      all_tables.reject{ |t| t.ignored? || !t.polymorphic? }
     end
     
     # Returns an array of all tables that have not been ignored and are to be embedded
     def embed_tables
-      tables.reject{|t| !t.embed?}
+      tables.reject{|t| !t.embedded?}
     end
     
   end

data/lib/mongify/translation/process.rb

@@ -20,6 +20,7 @@ module Mongify
         copy_data
         copy_embedded_tables
         update_reference_ids
+        copy_polymorphic_tables
         remove_pre_mongified_ids
         nil
       end
@@ -47,12 +48,40 @@ module Mongify
             row.delete(t.embed_on)
             row.merge!(fetch_reference_ids(t, row))
             row.delete('pre_mongified_id')
-            save_function_call = t.embed_as_object? ? '$set' : '$addToSet'
+            save_function_call = t.embedded_as_object? ? '$set' : '$addToSet'
             no_sql_connection.update(t.embed_in, target_row['_id'], {save_function_call => {t.name => row}})
           end
         end
       end
       
+      # Moves over polymorphic data
+      def copy_polymorphic_tables
+        self.polymorphic_tables.each do |t|
+          polymorphic_id_col, polymorphic_type_col = "#{t.polymorphic_as}_id", "#{t.polymorphic_as}_type"
+          sql_connection.select_rows(t.sql_name).each do |row|
+            table_name = row[polymorphic_type_col].tableize            
+            new_id = no_sql_connection.get_id_using_pre_mongified_id(table_name, row[polymorphic_id_col])
+            puts "getting new id for #{table_name}, #{row[polymorphic_id_col]} and getting #{new_id}"
+            if new_id
+              row = t.translate(row)
+              row.merge!(fetch_reference_ids(t, row))
+              row[polymorphic_id_col] = new_id
+              row.delete('pre_mongified_id')
+              if t.embedded?
+                row.delete(polymorphic_id_col)
+                row.delete(polymorphic_type_col)
+                save_function_call = t.embedded_as_object? ? '$set' : '$addToSet'
+                no_sql_connection.update(table_name, new_id, {save_function_call => {t.name => row}})
+              else
+                no_sql_connection.insert_into(t.name, row)
+              end
+            else
+              puts "#{table_name} table not found on #{t.sql_name} polymorphic import"
+            end
+          end
+        end
+      end
+      
       # Updates the reference ids in the no sql database
       def update_reference_ids
         self.tables.each do |t|

data/lib/mongify/version.rb

@@ -1,4 +1,4 @@
 module Mongify
   # Mongify's Current Version Number
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

data/spec/files/{simple_translation.rb → translation.rb}

@@ -33,3 +33,15 @@ table "preferences", :embed_in => :users, :as => :object do
 	column "updated_at", :datetime, :ignore => true
 end
 
+table "notes", :embed_in => true, :polymorphic => 'notable' do
+	column "id", :key
+	column "user_id", :integer, :references => "users"
+	column "notable_id", :integer
+	column "notable_type", :string
+	column "body", :text
+	column "created_at", :datetime
+	column "updated_at", :datetime
+end
+
+
+

data/spec/mongify/cli/worker_command_spec.rb

@@ -12,7 +12,7 @@ describe Mongify::CLI::WorkerCommand do
     @no_sql_connection.stub(:has_connection?).and_return(true)
     @config.stub(:no_sql_connection).and_return(@no_sql_connection)
     
-    @translation_file = 'spec/files/simple_translation.rb'
+    @translation_file = 'spec/files/translation.rb'
     
     Mongify::Translation.stub(:load).and_return(stub(:print => 'worked'))
     @view = mock('view').as_null_object
@@ -82,7 +82,7 @@ describe Mongify::CLI::WorkerCommand do
   
   context "process command" do
     before(:each) do
-      @command = Mongify::CLI::WorkerCommand.new('process', @config, 'spec/files/simple_translation.rb')
+      @command = Mongify::CLI::WorkerCommand.new('process', @config, 'spec/files/translation.rb')
       Mongify::Translation.stub(:parse).and_return(mock(:process => true))
     end
     it "should report success" do

data/spec/mongify/database/no_sql_connection_spec.rb

@@ -127,7 +127,7 @@ describe Mongify::Database::NoSqlConnection do
   
   describe "working connection" do
     before(:each) do
-      @mongodb_connection = GenerateDatabase.mongo_connection
+      @mongodb_connection = DatabaseGenerator.mongo_connection
     end
     
     it "should work" do

data/spec/mongify/database/sql_connection_spec.rb

@@ -2,7 +2,7 @@ require 'spec_helper'
 
 describe Mongify::Database::SqlConnection do
   before(:all) do
-    @db_path = GenerateDatabase.sqlite
+    @db_path = DatabaseGenerator.sqlite
   end
 
   before(:each) do
@@ -30,7 +30,7 @@ describe Mongify::Database::SqlConnection do
 
   context "MySql config" do
     before(:each) do
-      @sql_connection = GenerateDatabase.mysql_connection
+      @sql_connection = DatabaseGenerator.mysql_connection
     end
 
     context "valid?" do
@@ -68,13 +68,13 @@ describe Mongify::Database::SqlConnection do
 
     context "tables" do
       it "should be able to get a list" do
-        @sql_connection.tables.sort.should == ['comments', 'posts', 'preferences', 'users'].sort
+        @sql_connection.tables.should =~ ['comments', 'notes', 'posts', 'preferences', 'users']
       end
     end
 
     context "columns" do
       it "should see columns for a table" do
-        @sql_connection.columns_for(:users).map{ |column| column.name }.sort.should == ['id', 'first_name', 'last_name', 'created_at', 'updated_at'].sort
+        @sql_connection.columns_for(:users).map{ |column| column.name }.should =~ ['id', 'first_name', 'last_name', 'created_at', 'updated_at']
       end
     end
   end

data/spec/mongify/database/table_spec.rb

@@ -150,21 +150,21 @@ describe Mongify::Database::Table do
       it "should allow Object as a value" do
         Mongify::Database::Table.new('comments', :embed_in => 'posts', :on => 'post_id', :as => :object).embed_as.should == 'object'
       end
-      context "embed_as_object?" do
+      context "embedded_as_object?" do
         it "should be true" do
-          Mongify::Database::Table.new('comments', :embed_in => 'posts', :on => 'post_id', :as => :object).should be_embed_as_object
+          Mongify::Database::Table.new('comments', :embed_in => 'posts', :on => 'post_id', :as => :object).should be_embedded_as_object
         end
         it "should be false" do
-          Mongify::Database::Table.new('comments', :embed_in => 'posts', :on => 'post_id', :as => :array).should_not be_embed_as_object
+          Mongify::Database::Table.new('comments', :embed_in => 'posts', :on => 'post_id', :as => :array).should_not be_embedded_as_object
         end
       end
     end
-    context "embed?" do
+    context "embedded?" do
       it "should be true" do
-        Mongify::Database::Table.new('comments', :embed_in => 'posts', :on => 'post_id').should be_embed
+        Mongify::Database::Table.new('comments', :embed_in => 'posts', :on => 'post_id').should be_embedded
       end
       it "should be false" do
-        Mongify::Database::Table.new('users').should_not be_embed
+        Mongify::Database::Table.new('users').should_not be_embedded
       end
     end
     
@@ -172,7 +172,7 @@ describe Mongify::Database::Table do
       it "should be post_id" do
         Mongify::Database::Table.new('comments', :embed_in => 'posts', :on => 'post_id').embed_on.should == 'post_id'
       end
-      it "should be nil when not embed?" do
+      it "should be nil when not embedded?" do
         Mongify::Database::Table.new('users', :on => 'test').embed_on.should be_nil
       end
       it "should calculate embed_on from embed_in" do
@@ -185,7 +185,6 @@ describe Mongify::Database::Table do
     before(:each) do
       @table = Mongify::Database::Table.new('users')
       @table.before_save do |row|
-        puts "my keys are: #{row.keys}"
         row.admin = row.delete('permission').to_i > 50
       end
     end
@@ -202,6 +201,31 @@ describe Mongify::Database::Table do
     end
   end
   
+  context "polymorphic" do
+    before(:each) do
+      @table = Mongify::Database::Table.new('comments', :polymorphic => :commentable)
+    end
+    it "should be true" do
+      @table.should be_polymorphic
+    end
+    it "should be false" do
+      Mongify::Database::Table.new('users').should_not be_polymorphic
+    end
+    it "should return 'commentable'" do
+      @table.polymorphic_as.should == 'commentable'
+    end
+    
+    context "embed" do
+      before(:each) do
+        @table = Mongify::Database::Table.new('comments', :polymorphic => :commentable, :embed_in => true)
+      end
+      it "should be embedded" do
+        @table.should be_embedded
+      end
+      
+    end
+  end
+  
   context "translate" do
     before(:each) do
       @column1 = mock(:translate => {'first_name' => 'Timmy'}, :name => 'first_name')

data/spec/mongify/translation/process_spec.rb

@@ -2,9 +2,9 @@ require 'spec_helper'
 
 describe Mongify::Translation::Process do
   before(:each) do
-    GenerateDatabase.clear_mongodb
-    @sql_connection = GenerateDatabase.sqlite_connection
-    @no_sql_connection = GenerateDatabase.mongo_connection
+    DatabaseGenerator.clear_mongodb
+    @sql_connection = DatabaseGenerator.sqlite_connection
+    @no_sql_connection = DatabaseGenerator.mongo_connection
     @translation = Mongify::Translation.new
   end
   
@@ -47,7 +47,7 @@ describe Mongify::Translation::Process do
     it "should get correct information" do
       @no_sql_connection = mock()
       @translation.stub(:no_sql_connection).and_return(@no_sql_connection)
-      @table = mock(:translate => {}, :name => 'users', :embed? => false)
+      @table = mock(:translate => {}, :name => 'users', :embedded? => false)
       @column = mock(:name => 'user_id', :references => 'users')
       @table.stub(:reference_columns).and_return([@column])
       @no_sql_connection.should_receive(:get_id_using_pre_mongified_id).with('users', 1).once.and_return(500)
@@ -65,7 +65,7 @@ describe Mongify::Translation::Process do
       
       @table = mock(:translate => {'first_name'=> 'Timmy', 'last_name' => 'Zuza', 'preference_id' => 1},
                     :name => 'users', 
-                    :embed? => false, 
+                    :embedded? => false, 
                     :sql_name => 'users')
                     
       @translation.stub(:tables).and_return([@table])
@@ -85,8 +85,8 @@ describe Mongify::Translation::Process do
     
     context "copy_embed_tables" do
       before(:each) do
-        @target_table = mock(:name => 'posts', :embed? => false, :sql_name => 'posts')
-        @embed_table = mock(:translate => {}, :name => 'comments', :embed? => true, :embed_on => 'post_id', :embed_in => 'posts', :embed_as_object? => false, :sql_name => 'comments')
+        @target_table = mock(:name => 'posts', :embedded? => false, :sql_name => 'posts')
+        @embed_table = mock(:translate => {}, :name => 'comments', :embedded? => true, :embed_on => 'post_id', :embed_in => 'posts', :embedded_as_object? => false, :sql_name => 'comments')
         @no_sql_connection.stub(:find_one).and_return({'_id' => 500})
         @translation.stub(:tables).and_return([@target_table, @embed_table])
         @translation.stub(:fetch_reference_ids).and_return({})
@@ -98,25 +98,25 @@ describe Mongify::Translation::Process do
         @translation.send(:copy_embedded_tables)
       end
       it "should remove the pre_mongified_id before embedding" do
-        @embed_table = mock(:translate => {'first_name' => 'bob', 'pre_mongified_id' => 1}, :name => 'comments', :sql_name => 'comments', :embed? => true, :embed_on => 'post_id', :embed_in => 'posts', :embed_as_object? => false)
+        @embed_table = mock(:translate => {'first_name' => 'bob', 'pre_mongified_id' => 1}, :name => 'comments', :sql_name => 'comments', :embedded? => true, :embed_on => 'post_id', :embed_in => 'posts', :embedded_as_object? => false)
         @translation.stub(:tables).and_return([@target_table, @embed_table])
         @no_sql_connection.should_receive(:update).with("posts", 500, {"$addToSet"=>{"comments"=>{'first_name' => 'bob'}}})
         @translation.send(:copy_embedded_tables)
       end
       it "should remove the parent_id from the embedding row" do
-        @embed_table = mock(:translate => {'first_name' => 'bob', 'post_id' => 1}, :name => 'comments', :sql_name => 'comments', :embed? => true, :embed_on => 'post_id', :embed_in => 'posts', :embed_as_object? => false)
+        @embed_table = mock(:translate => {'first_name' => 'bob', 'post_id' => 1}, :name => 'comments', :sql_name => 'comments', :embedded? => true, :embed_on => 'post_id', :embed_in => 'posts', :embedded_as_object? => false)
         @translation.stub(:tables).and_return([@target_table, @embed_table])
         @no_sql_connection.should_receive(:update).with("posts", 500, {"$addToSet"=>{"comments"=>{'first_name' => 'bob'}}})
         @translation.send(:copy_embedded_tables)
       end
       it "should call $addToSet on update of an embed_as_object table" do
-        @embed_table = mock(:translate => {'first_name' => 'bob', 'post_id' => 1}, :name => 'comments', :sql_name => 'comments', :embed? => true, :embed_on => 'post_id', :embed_in => 'posts', :embed_as_object? => true)
+        @embed_table = mock(:translate => {'first_name' => 'bob', 'post_id' => 1}, :name => 'comments', :sql_name => 'comments', :embedded? => true, :embed_on => 'post_id', :embed_in => 'posts', :embedded_as_object? => true)
         @translation.stub(:tables).and_return([@target_table, @embed_table])
         @no_sql_connection.should_receive(:update).with("posts", 500, {"$set"=>{"comments"=>{'first_name' => 'bob'}}})
         @translation.send(:copy_embedded_tables)
       end
       it "should allow rename of table" do
-        @embed_table = mock(:translate => {'first_name' => 'bob', 'post_id' => 1}, :name => 'notes', :sql_name => 'comments', :embed? => true, :embed_on => 'post_id', :embed_in => 'posts', :embed_as_object? => true)
+        @embed_table = mock(:translate => {'first_name' => 'bob', 'post_id' => 1}, :name => 'notes', :sql_name => 'comments', :embedded? => true, :embed_on => 'post_id', :embed_in => 'posts', :embedded_as_object? => true)
         @translation.stub(:tables).and_return([@target_table, @embed_table])
         @no_sql_connection.should_receive(:update).with("posts", 500, {"$set"=>{"notes"=>{'first_name' => 'bob'}}})
         @translation.send(:copy_embedded_tables)
@@ -140,6 +140,55 @@ describe Mongify::Translation::Process do
       end
     end
     
+    context "copy_polymorphic_tables" do
+      before(:each) do
+        @ref_table = mock(:name => 'user_accounts', 
+                          :embedded? => false,
+                          :ignored? => false,
+                          :sql_name => 'user_accounts')
+        @translation.stub(:find).with('user_accounts').and_return([@ref_table])
+        
+        @sql_connection.should_receive(:select_rows).with('comments').and_return([{'commentable_id' => 1, 'commentable_type' => 'UserAccount', 'data' => 'good'}])
+        @no_sql_connection.should_receive(:get_id_using_pre_mongified_id).with('user_accounts', 1).and_return(500)
+      end
+      context "embedded" do
+        it "should work correctly" do
+           @table = mock(:translate => {'data' => 123},
+                          :name => 'comments', 
+                          :embedded? => true,
+                          :polymorphic_as => 'commentable',
+                          :polymorphic? => true, 
+                          :ignored? => false,
+                          :embedded_as_object? => false,
+                          :sql_name => 'comments',
+                          :reference_columns => [])
+
+            @translation.stub(:all_tables).and_return([@table])
+          
+          @no_sql_connection.should_receive(:update).with('user_accounts', 500, {'$addToSet' => {'comments' => {'data' => 123}}})
+          @translation.send(:copy_polymorphic_tables)
+        end
+      end
+      context "not embedded" do
+        it "should work" do
+          @table = mock(:translate => {'data' => 123, 'commentable_type' => 'UserAccount', 'commentable_id' => 1},
+                          :name => 'comments', 
+                          :embedded? => false,
+                          :polymorphic_as => 'commentable',
+                          :polymorphic? => true, 
+                          :ignored? => false,
+                          :embedded_as_object? => false,
+                          :sql_name => 'comments',
+                          :reference_columns => [])
+
+          @translation.stub(:all_tables).and_return([@table])
+            
+          @no_sql_connection.should_receive(:insert_into).with('comments', {'data' => 123, 'commentable_type' => 'UserAccount', 'commentable_id' => 500})
+          @translation.send(:copy_polymorphic_tables)
+        end
+      end
+    end
+    
     context "remove_pre_mongified_ids" do
       it "should remove_pre_mongified_ids on no_sql_connection" do
         @no_sql_connection.should_receive(:remove_pre_mongified_ids).with(anything)

data/spec/mongify/translation_spec.rb

@@ -2,7 +2,7 @@ require 'spec_helper'
 
 describe Mongify::Translation do
   before(:all) do
-    @file_path = File.expand_path(File.dirname(__FILE__) + '/../files/simple_translation.rb')
+    @file_path = File.expand_path(File.dirname(__FILE__) + '/../files/translation.rb')
     @translation = Mongify::Translation.parse(@file_path)
   end
   
@@ -48,17 +48,35 @@ describe Mongify::Translation do
     end
   end
   
+  context "find" do
+    before(:each) do
+      @user_table = mock(:name => 'users')
+      @translation.stub(:all_tables).and_return([mock(:name => 'comments'),
+                                                 @user_table, 
+                                                 mock(:name => 'posts')])
+    end
+    it "should work" do
+      
+      @translation.find('users').should == @user_table
+    end
+    it "should return nil if nothing is found" do
+      @translation.find('apples').should be_nil
+    end
+  end
+  
   context "tables reference" do
     before(:each) do
-      @copy_table = mock(:name => 'users', :embed? => false, :ignored? => false)
-      @embed_table = mock(:name => 'comments', :embed? => true, :ignored? => false)
-      @ignored_table = mock(:name => 'apples', :ignored? => true, :embed? => false)
+      @copy_table = mock(:name => 'users', :ignored? => false, :embedded? => false, :polymorphic? => false)
+      @embed_table = mock(:name => 'comments', :ignored? => false, :embedded? => true, :polymorphic? => false)
+      @ignored_table = mock(:name => 'apples', :ignored? => true, :embedded? => false, :polymorphic? => false)
+      @polymorphic_table = mock(:name => 'comments', :ignored? => false, :embedded? => false, :polymorphic? => true)
       @translation = Mongify::Translation.new()
-      @translation.stub(:all_tables).and_return([@copy_table, @embed_table, @ignored_table])
+      @all_tables = [@copy_table, @embed_table, @ignored_table, @polymorphic_table]
+      @translation.stub(:all_tables).and_return(@all_tables)
     end
     context "tables" do
       it "should not show ignored" do
-        
+        @translation.tables.count == @all_tables.count - 1
       end
     end
     context "copy_tables" do
@@ -71,6 +89,9 @@ describe Mongify::Translation do
         @translation.embed_tables.should == [@embed_table]
       end
     end
+    it "should return only polymorphic tables" do
+      @translation.polymorphic_tables.should == [@polymorphic_table]
+    end
   end
   
   context "add_table" do

data/spec/support/{generate_database.rb → database_generator.rb}

@@ -1,5 +1,5 @@
 # Used during testing to generate and load database information and connection strings
-class GenerateDatabase
+class DatabaseGenerator
   # Returns a mysql connection (using the database.yml)
   def self.mysql_connection
     @sql_connection ||= Mongify::Database::SqlConnection.new( :adapter => CONNECTION_CONFIG.mysql['adapter'], 
@@ -10,6 +10,7 @@ class GenerateDatabase
                                                             :database => CONNECTION_CONFIG.mysql['database']
                                                           )
   end
+  
   # Returns a sqlite connection (using the database.yml )
   def self.sqlite_connection
     @db_path = File.join(File.dirname(File.dirname(File.dirname(File.expand_path(__FILE__)))), CONNECTION_CONFIG.sqlite['database'])
@@ -50,6 +51,14 @@ class GenerateDatabase
       t.timestamps
     end
     
+    conn.create_table(:notes) do |t|
+      t.integer :user_id
+      t.integer :notable_id
+      t.string :notable_type
+      t.text :body
+      t.timestamps
+    end
+    
     if include_data
       
       #Users
@@ -75,21 +84,33 @@ class GenerateDatabase
       [
         {:post_id => 1, :user_id => 1, :body => 'First Comment Body'},
         {:post_id => 2, :user_id => 1, :body => 'Second Comment Body'},
-        {:post_id => 2, :user_id => 2, :body => 'Thrid Comment Body'}
+        {:post_id => 2, :user_id => 2, :body => 'Thrid Comment Body'},
       ].each do |v|
         conn.insert("INSERT INTO comments (body, post_id, user_id, created_at, updated_at) 
                     VALUES ('#{v[:body]}', #{v[:post_id]}, #{v[:user_id]}, '#{Time.now.to_s(:db)}', '#{Time.now.to_s(:db)}')")
       end
       
+      #Preferences
       [
         {:user_id => 1, :notify_by_email => true},
         {:user_id => 2, :notify_by_email => true},
-        {:user_id => 3, :notify_by_email => false}
+        {:user_id => 3, :notify_by_email => false},
       ].each do |v|
           conn.insert("INSERT INTO preferences (user_id, notify_by_email, created_at, updated_at) 
                       VALUES (#{v[:user_id]}, '#{v[:notify_by_email]}', '#{Time.now.to_s(:db)}', '#{Time.now.to_s(:db)}')")
       end
       
+      #Notes
+      [
+        {:user_id => 1, :body => 'note 1', :notable_id => 1, :notable_type => 'Post'},
+        {:user_id => 2, :body => 'note 2', :notable_id => 2, :notable_type => 'Post'},
+        {:user_id => 2, :body => 'note 3', :notable_id => 1, :notable_type => 'User'},
+        {:user_id => 3, :body => 'note 4', :notable_id => 2, :notable_type => 'User'},
+      ].each do |v|
+          conn.insert("INSERT INTO notes (user_id, body, notable_id, notable_type, created_at, updated_at) 
+                      VALUES (#{v[:user_id]}, '#{v[:body]}', #{v[:notable_id]}, '#{v[:notable_type]}', '#{Time.now.to_s(:db)}', '#{Time.now.to_s(:db)}')")
+      end
+      
     end
     
     sqlite_connection.database

data/spec/support/database_output.txt

@@ -33,3 +33,13 @@ table "preferences" do
 	column "updated_at", :datetime
 end
 
+table "notes" do
+	column "id", :key
+	column "user_id", :integer, :references => "users"
+	column "notable_id", :integer, :references => "notables"
+	column "notable_type", :string
+	column "body", :text
+	column "created_at", :datetime
+	column "updated_at", :datetime
+end
+

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: mongify
 version: !ruby/object:Gem::Version 
-  hash: 27
+  hash: 25
   prerelease: 
   segments: 
   - 0
   - 1
-  - 0
-  version: 0.1.0
+  - 1
+  version: 0.1.1
 platform: ruby
 authors: 
 - Andrew Kalek
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-01-21 00:00:00 -05:00
+date: 2011-01-28 00:00:00 -05:00
 default_executable: mongify
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -284,8 +284,7 @@ files:
 - mongify.gemspec
 - spec/default.watch
 - spec/files/base_configuration.rb
-- spec/files/empty_translation.rb
-- spec/files/simple_translation.rb
+- spec/files/translation.rb
 - spec/mongify/cli/application_spec.rb
 - spec/mongify/cli/help_command_spec.rb
 - spec/mongify/cli/options_spec.rb
@@ -307,8 +306,8 @@ files:
 - spec/spec_helper.rb
 - spec/support/config_reader.rb
 - spec/support/database.example
+- spec/support/database_generator.rb
 - spec/support/database_output.txt
-- spec/support/generate_database.rb
 - spec/tmp/.gitignore
 has_rdoc: true
 homepage: http://github.com/anlek/mongify
@@ -345,7 +344,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
 requirements: []
 
 rubyforge_project: 
-rubygems_version: 1.4.1
+rubygems_version: 1.4.2
 signing_key: 
 specification_version: 3
 summary: Translate your SQL data to MongoDB
@@ -358,8 +357,7 @@ test_files:
 - features/support/env.rb
 - spec/default.watch
 - spec/files/base_configuration.rb
-- spec/files/empty_translation.rb
-- spec/files/simple_translation.rb
+- spec/files/translation.rb
 - spec/mongify/cli/application_spec.rb
 - spec/mongify/cli/help_command_spec.rb
 - spec/mongify/cli/options_spec.rb
@@ -381,6 +379,6 @@ test_files:
 - spec/spec_helper.rb
 - spec/support/config_reader.rb
 - spec/support/database.example
+- spec/support/database_generator.rb
 - spec/support/database_output.txt
-- spec/support/generate_database.rb
 - spec/tmp/.gitignore