squealer 2.1.1 → 2.2.0

data/spec/spec_helper.rb

@@ -6,37 +6,31 @@ $LOAD_PATH.unshift(File.dirname(__FILE__))
 $LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
 
 require 'squealer'
+require "spec_helper_dbms_#{ENV['EXPORT_DBMS']||'mysql'}"
 
 Spec::Runner.configure do |config|
   config.before(:suite) do
     $db_name = "test_export_#{object_id}"
-    create_test_db($db_name)
+    create_export_db($db_name)
+    create_import_db($db_name)
   end
 
   config.after(:suite) do
-    drop_test_db($db_name)
-  end
-
-  def create_test_db(name)
-    @my = Mysql.connect('localhost', 'root')
-    @my.query("DROP DATABASE IF EXISTS #{name}")
-    @my.query("CREATE DATABASE #{name}")
-    @my.query("USE #{name}")
-    @my.query("SET sql_mode='ANSI_QUOTES'")
-
-    create_export_tables
+    Squealer::Database.instance.send(:dispose_all_connections)
+    drop_export_test_db($db_name)
 
-    Squealer::Database.instance.import_from('localhost', 27017, $db_name)
-    @mongo = Squealer::Database.instance.import.send(:instance_variable_get, '@dbc')
     drop_mongo
-    seed_import
   end
 
-  def drop_test_db(name)
-    @my.query("DROP DATABASE IF EXISTS #{name}")
-    @my.close
+  config.after(:each) do
+    @export_dbc.dispose if @export_dbc
+  end
 
+  def create_import_db(name)
+    Squealer::Database.instance.import_from('localhost', 27017, name)
+    @mongo = Squealer::Database.instance.import.instance_variable_get('@dbc')
     drop_mongo
+    seed_import
   end
 
   def drop_mongo
@@ -51,24 +45,24 @@ Spec::Runner.configure do |config|
       { :name => 'Josh Graham', :dob => as_time(Date.parse('01-Jan-1971')), :gender => 'M',
         :organization_id => hashrocket,
         :activities => [
-          { :name => 'Develop squealer', :due_date => as_time(Date.today + 1) },
-          { :name => 'Organize speakerconf.com', :due_date => as_time(Date.today + 30) },
-          { :name => 'Hashrocket party', :due_date => as_time(Date.today + 7) }
+          { :_id => id, :name => 'Develop squealer', :due_date => as_time(Date.today + 1) },
+          { :_id => id, :name => 'Organize speakerconf.com', :due_date => as_time(Date.today + 30) },
+          { :_id => id, :name => 'Hashrocket party', :due_date => as_time(Date.today + 7) }
         ]
       },
       { :name => 'Bernerd Schaefer', :dob => as_time(Date.parse('31-Dec-1985')), :gender => 'M',
         :organization_id => hashrocket,
         :activities => [
-          { :name => 'Retype all of the code Josh wrote in squealer', :due_date => as_time(Date.today + 2) },
-          { :name => 'Listen to rare Thelonius Monk EP', :due_date => as_time(Date.today) },
-          { :name => 'Practice karaoke', :due_date => as_time(Date.today + 7) }
+          { :_id => id, :name => 'Retype all of the code Josh wrote in squealer', :due_date => as_time(Date.today + 2) },
+          { :_id => id, :name => 'Listen to rare Thelonius Monk EP', :due_date => as_time(Date.today) },
+          { :_id => id, :name => 'Practice karaoke', :due_date => as_time(Date.today + 7) }
         ]
       },
       { :name => 'Your momma', :dob => as_time(Date.parse('15-Jun-1955')), :gender => 'F',
         :organization_id => zorganization,
         :activities => [
-          { :name => 'Cook me some pie', :due_date => as_time(Date.today) },
-          { :name => 'Make me a sammich', :due_date => as_time(Date.today) }
+          { :_id => id, :name => 'Cook me some pie', :due_date => as_time(Date.today) },
+          { :_id => id, :name => 'Make me a sammich', :due_date => as_time(Date.today) }
         ]
       }
     ]
@@ -76,37 +70,23 @@ Spec::Runner.configure do |config|
     users.each { |user| @mongo.collection('users').save user }
   end
 
-  def create_export_tables
-    command = <<-COMMAND.gsub(/\n\s*/, " ")
-      CREATE TABLE "users" (
-        "id" INT NOT NULL AUTO_INCREMENT ,
-        "name" VARCHAR(255) NULL ,
-        "gender" CHAR(1) NULL ,
-        "dob" DATE NULL ,
-        PRIMARY KEY ("id") )
-    COMMAND
-    @my.query(command)
-
-    command = <<-COMMAND.gsub(/\n\s*/, " ")
-      CREATE TABLE "activity" (
-        "id" INT NOT NULL AUTO_INCREMENT ,
-        "user_id" INT NULL ,
-        "name" VARCHAR(255) NULL ,
-        "due_date" DATE NULL ,
-        PRIMARY KEY ("id") )
-    COMMAND
-    @my.query(command)
-
-    command = <<-COMMAND.gsub(/\n\s*/, " ")
-      CREATE TABLE "organizations" (
-        "id" INT NOT NULL AUTO_INCREMENT ,
-        "disabed_date" DATE NULL ,
-        PRIMARY KEY ("id") )
-    COMMAND
-    @my.query(command)
+
+  def truncate_export_tables
+    non_query('TRUNCATE TABLE "user"')
+    non_query('TRUNCATE TABLE "activity"')
+    non_query('TRUNCATE TABLE "organization"')
   end
 
   def as_time(date)
     Time.parse(date.to_s)
   end
+
+  def non_query(text)
+    export_dbc.create_command(text).execute_non_query
+  end
+
+  def id
+    require 'digest/sha1'
+    (Digest::SHA1.hexdigest rand.to_s)[0,24]
+  end
 end

data/spec/spec_helper_dbms_mysql.rb

@@ -0,0 +1,56 @@
+def create_export_db(name)
+  $db_adapter = 'mysql'
+  dbc = DataObjects::Connection.new("mysql://root@localhost/mysql")
+  dbc.create_command("DROP DATABASE IF EXISTS #{name}").execute_non_query
+  dbc.create_command("CREATE DATABASE #{name}").execute_non_query
+  dbc.create_command("SET sql_mode='ANSI_QUOTES'").execute_non_query
+  create_export_tables
+  dbc.dispose
+end
+
+def drop_export_test_db(name)
+  @export_dbc.dispose if @export_dbc
+  dbc = DataObjects::Connection.new("mysql://root@localhost/mysql")
+  dbc.create_command("DROP DATABASE IF EXISTS #{name}").execute_non_query
+  dbc.dispose
+end
+
+def create_export_tables
+  command = <<-COMMAND.gsub(/\n\s*/, " ")
+      CREATE TABLE "user" (
+        "id" CHAR(24) NOT NULL ,
+        "organization_id" CHAR(24) NOT NULL ,
+        "name" VARCHAR(255) NULL ,
+        "gender" CHAR(1) NULL ,
+        "dob" DATETIME NULL ,
+        "foreign" BOOLEAN NULL ,
+        "dull" BOOLEAN NULL ,
+        "symbolic" VARCHAR(255) NULL ,
+        "interests" TEXT NULL ,
+        PRIMARY KEY ("id") )
+  COMMAND
+  non_query(command)
+
+  command = <<-COMMAND.gsub(/\n\s*/, " ")
+      CREATE TABLE "activity" (
+        "id" CHAR(24) NOT NULL ,
+        "user_id" CHAR(24) NULL ,
+        "name" VARCHAR(255) NULL ,
+        "due_date" DATETIME NULL ,
+        PRIMARY KEY ("id") )
+  COMMAND
+  non_query(command)
+
+  command = <<-COMMAND.gsub(/\n\s*/, " ")
+      CREATE TABLE "organization" (
+        "id" CHAR(24) NOT NULL ,
+        "disabled_date" DATETIME NULL ,
+        PRIMARY KEY ("id") )
+  COMMAND
+  non_query(command)
+end
+
+def export_dbc
+  $db_user ||= 'root'
+  @export_dbc ||= DataObjects::Connection.new("mysql://root@localhost/#{$db_name}")
+end

data/spec/spec_helper_dbms_postgres.rb

@@ -0,0 +1,55 @@
+def create_export_db(name)
+  $db_adapter = 'postgres'
+  dbc = DataObjects::Connection.new("postgres://localhost/postgres")
+  dbc.create_command("DROP DATABASE IF EXISTS #{name}").execute_non_query
+  dbc.create_command("CREATE DATABASE #{name}").execute_non_query
+  create_export_tables
+  dbc.dispose
+end
+
+def drop_export_test_db(name)
+  @export_dbc.dispose if @export_dbc
+  dbc = DataObjects::Connection.new("postgres://localhost/postgres")
+  dbc.create_command("DROP DATABASE IF EXISTS #{name}").execute_non_query
+  dbc.dispose
+end
+
+def create_export_tables
+  command = <<-COMMAND.gsub(/\n\s*/, " ")
+      CREATE TABLE "user" (
+        "id" CHAR(24) NOT NULL ,
+        "organization_id" CHAR(24) NOT NULL ,
+        "name" VARCHAR(255) NULL ,
+        "gender" CHAR(1) NULL ,
+        "dob" TIMESTAMP NULL ,
+        "foreign" BOOLEAN NULL ,
+        "dull" BOOLEAN NULL ,
+        "symbolic" VARCHAR(255) NULL ,
+        "interests" TEXT NULL ,
+        PRIMARY KEY ("id") )
+  COMMAND
+  non_query(command)
+
+  command = <<-COMMAND.gsub(/\n\s*/, " ")
+      CREATE TABLE "activity" (
+        "id" CHAR(24) NOT NULL ,
+        "user_id" CHAR(24) NULL ,
+        "name" VARCHAR(255) NULL ,
+        "due_date" TIMESTAMP NULL ,
+        PRIMARY KEY ("id") )
+  COMMAND
+  non_query(command)
+
+  command = <<-COMMAND.gsub(/\n\s*/, " ")
+      CREATE TABLE "organization" (
+        "id" CHAR(24) NOT NULL ,
+        "disabled_date" TIMESTAMP NULL ,
+        PRIMARY KEY ("id") )
+  COMMAND
+  non_query(command)
+end
+
+def export_dbc
+  $db_user ||= ''
+  @export_dbc ||= DataObjects::Connection.new("postgres://localhost/#{$db_name}")
+end

data/spec/squealer/database_spec.rb

@@ -1,5 +1,4 @@
 require 'spec_helper'
-require 'mysql'
 require 'mongo'
 
 describe Squealer::Database do
@@ -105,14 +104,25 @@ describe Squealer::Database do
       end
 
       context "real squeal" do
-        before { pending "interactive_view" }
-        subject do
-          source = databases.import.source("users")
-        end
-
-        it "^^^ you saw that progress bar right there" do
-          subject.each do
-            sleep (rand(2) + 1)
+        # before { pending "interactive_view" }
+        it "exports that stuff to SQL" do
+          databases.export_to($db_adapter, 'localhost', $db_user, '', $db_name)
+          databases.import.source("users").each do |user|
+            target(:user) do |target|
+              target.instance_variable_get('@row_id').should == user['_id'].to_s
+              assign(:organization_id)
+              assign(:name)
+
+              #TODO: Update README to highlight that all embedded docs should have an _id
+              # as all Ruby mappers for MongoDB make one. (according to Durran)
+              user.activities.each do |activity|
+                target(:activity) do |target|
+                  assign(:user_id)
+                  assign(:name)
+                  assign(:due_date)
+                end
+              end
+            end
           end
         end
       end
@@ -125,8 +135,28 @@ describe Squealer::Database do
     let(:databases) { Squealer::Database.instance }
 
     it "takes an export database" do
-      databases.export_to('localhost', 'root', '', $db_name)
-      databases.send(:instance_variable_get, '@export_dbc').should be_a_kind_of(Mysql)
+      databases.export_to($db_adapter, 'localhost', $db_user, '', $db_name)
+      databases.instance_variable_get('@export_do').should_not be_nil
+    end
+  end
+
+  describe "upsertable?" do
+    subject { Squealer::Database.instance }
+
+    context "mysql connection" do
+      before do
+        subject.export_to('mysql', 'localhost', 'root', '', 'mysql')
+      end
+
+      it { should be_upsertable }
+    end
+
+    context "postgres connection" do
+      before do
+        subject.export_to('postgres', 'localhost', '', '', 'postgres')
+      end
+
+      it { should_not be_upsertable }
     end
   end
 

data/spec/squealer/object_spec.rb

@@ -112,11 +112,11 @@ describe Object do
   end
 
   def mock_mysql
-    my = mock(Mysql)
-    st = mock(Mysql::Stmt)
+    my = mock(DataObjects::Connection)
+    comm = mock(DataObjects::Command)
     Squealer::Database.instance.should_receive(:export).at_least(:once).and_return(my)
-    my.should_receive(:prepare).at_least(:once).and_return(st)
-    st.should_receive(:execute).at_least(:once)
+    my.should_receive(:create_command).at_least(:once).and_return(comm)
+    comm.should_receive(:execute_non_query).at_least(:once)
   end
 
 end