cockroach 0.0.0

data/test/fixturers/factory_girl/profiler_test.rb

@@ -0,0 +1,124 @@
+require "test_helper"
+require "user"
+
+module Cockroach
+  class FactoryGirlProfilerTest < Test::Unit::TestCase
+    context "Simple profiler" do
+      setup do
+        before_setup
+        mock_factory_girl
+        Cockroach.setup do |c|
+          c.root = File.expand_path("../../../support/data/dummy_structure", __FILE__)
+          c.config_path = "./config/user_only.yml"
+        end
+      end
+
+      teardown do
+        after_teardown
+      end
+
+      should "inherit default config" do
+        profiler = nil
+
+        assert_nothing_thrown do
+          profiler = Cockroach::FactoryGirl::Profiler.new
+        end
+
+        assert_equal Cockroach.config.profile, profiler.instance_variable_get(:@source)
+      end
+
+      should "load! should preload nodes" do
+        profiler = Cockroach::FactoryGirl::Profiler.new
+
+        profiler.instance_variable_set(:@nodes, {})
+
+        profiler.expects(:load)
+
+        profiler.load!
+      end
+
+      context "Subnodes" do
+
+        context "Access" do
+          should "be from top level" do
+            profiler = Cockroach::FactoryGirl::Profiler.new
+            assert_equal @users_node, profiler['person']
+          end
+        end
+        
+        should "loading" do
+          profiler = Cockroach::FactoryGirl::Profiler.new
+
+          assert_nil profiler.instance_variable_get(:@loaded)
+
+          fake_node = mock()
+          fake_node.stubs(:node_name).returns('name')
+          Cockroach::FactoryGirl::Node.stubs(:new).with(any_parameters).returns(fake_node)
+          Cockroach::FactoryGirl::Node.expects(:new).with("users_amount",1000).returns(fake_node)
+
+          assert profiler.load
+          assert profiler.instance_variable_get(:@loaded)
+        end
+
+        should "loading!" do
+          profiler = Cockroach::FactoryGirl::Profiler.new
+
+          profiler.load
+
+          profiler.nodes.each_value do |node|
+            node.stubs(:load!)
+            node.expects(:load!)
+          end
+
+          assert profiler.load!
+        end
+      end
+    end
+
+    context "Profiling" do
+      setup do
+        before_setup
+        mock_factory_girl
+        ActiveSupport::Inflector.inflections do |inflect|
+          inflect.singular /(ss)$/, '\1'
+        end
+        Cockroach.setup do |c|
+          c.root = File.expand_path("../../../support/data/dummy_structure", __FILE__)
+          c.config_path = "./config/witness.yml"
+        end
+        @profile = Cockroach::FactoryGirl::Profiler.new
+      end
+      
+      should "load without error" do
+        ::FactoryGirl.stubs(:factory_by_name).with(any_parameters)
+
+        assert_nothing_thrown do
+          @profile.load
+        end
+      end
+
+      should "load! without error" do
+        ::FactoryGirl.stubs(:factory_by_name).with(any_parameters)
+
+        mocks = ((1..6).to_a.collect {|i| stub('witness', :id => i) })
+        message_mock = stub('message', :id => 0)
+
+        ::FactoryGirl.expects(:create).with("witness").times(6).returns( *mocks )
+
+        ::FactoryGirl.expects(:create).with("message", {"author" => mocks[0]}).at_least(1).at_most(6).returns(message_mock)
+        ::FactoryGirl.expects(:create).with("message", {"author" => mocks[1]}).at_least(1).at_most(6).returns(message_mock)
+        ::FactoryGirl.expects(:create).with("message", {"author" => mocks[2]}).at_least(1).at_most(6).returns(message_mock)
+        ::FactoryGirl.expects(:create).with("message", {"author" => mocks[3]}).at_least(1).at_most(6).returns(message_mock)
+        ::FactoryGirl.expects(:create).with("message", {"author" => mocks[4]}).at_least(1).at_most(6).returns(message_mock)
+        ::FactoryGirl.expects(:create).with("message", {"author" => mocks[5]}).at_least(1).at_most(6).returns(message_mock)
+
+        Cockroach::FactoryGirl::Node.any_instance.stubs(:allowed_options).returns(['author'])
+
+        @profile.load
+        assert_nothing_thrown do
+          @profile.load!
+        end
+      end
+    end
+  end
+end

data/test/fixturers/source_test.rb

@@ -0,0 +1,153 @@
+require "test_helper"
+require "user"
+
+module Cockroach
+  class FactoryGirlSourceTest < Test::Unit::TestCase
+    setup do
+      before_setup
+      mock_factory_girl
+    end
+
+    teardown do
+      after_teardown
+    end
+
+    context "Getting source" do
+      context "Model" do
+        setup do
+          @old_const = Object.const_get(:Place) if Object.const_defined?(:Place)
+          @place = stub('Place')
+          silence_warnings { Object.const_set(:Place, @place) }
+        end
+
+        teardown do
+          if @old_const
+            silence_warnings { Object.const_set('Place', @old_const) }
+          end
+        end
+
+        should "get model with conditions option" do
+          Cockroach::Source::Model.expects(:new).with(Place, {"conditions" => "column = 'test'"})
+
+          Cockroach::Source.get_source({"model" => "Place", "conditions" => "column = 'test'"})
+        end
+        
+        should "get model with id option" do
+          Cockroach::Source::Model.expects(:new).with(Place, {"id" => "3"})
+          
+          Cockroach::Source.get_source({"id" => "3","model" => "Place"})
+        end
+
+        should "get model without options" do
+          Cockroach::Source::Model.expects(:new).with(Place, {})
+
+          Cockroach::Source.get_source({"model" => "Place"})
+        end
+      end
+    end
+
+    context "Node" do
+      setup do
+        @lands_node = Cockroach::FactoryGirl::Node.new(
+          'places' => {
+            'as' => 'lands',
+            'amount' => '10'
+          })
+        @source = Cockroach::Source::Node.new @lands_node
+      end
+
+      should "find return nil if no record created yet" do
+        assert_nil @source.send(:find, 1)
+      end
+
+      should "random return nil if no record created yet" do
+        assert_nil @source.sample
+      end
+
+      should "return record with certain id" do
+        places = ((1..10).to_a.collect {|i| stub('place', :id => i) })
+        ::FactoryGirl.stubs("create").with("place").returns( *places )
+        @lands_node.__send__(:load!)
+
+        place_class = stub('place_clase')
+        place_class.stubs(:find).with(6).returns(p = places[5])
+        factory = @lands_node.instance_variable_get(:@factory)
+        factory.stubs(:send).with(:class_name).returns(place_class)
+
+        assert_equal p, @source.send(:find, 6)
+      end
+
+      should "return random record" do
+        places = ((1..10).to_a.collect {|i| stub('place', :id => i) })
+        ::FactoryGirl.stubs("create").with("place").returns( *places )
+        @lands_node.__send__(:load!)
+
+        place_class = stub('place_clase')
+        place_class.stubs(:find).with(any_parameters).returns(p = places.sample)
+        factory = @lands_node.instance_variable_get(:@factory)
+        factory.stubs(:send).with(:class_name).returns(place_class)
+
+        assert_equal p, @source.sample
+      end
+    end
+
+    context "Model" do
+      setup do
+        @old_const = Object.const_get(:Place) if Object.const_defined?(:Place)
+        @place = stub('Place')
+        silence_warnings { Object.const_set(:Place, @place) }
+
+        @source = Cockroach::Source::Model.new Place
+      end
+
+      teardown do
+        if @old_const
+          silence_warnings { Object.const_set('Place', @old_const) }
+        end
+      end
+
+      should "return record with certain id" do
+        places = ((1..10).to_a.collect {|i| stub('place', :id => i) })
+        @place.stubs(:find).with(6).returns(p = places[5])
+        
+        assert_equal p, @source.send(:find, 6)
+      end
+
+      should "return random record" do
+        places = ((1..10).to_a.collect {|i| stub('place', :id => i) })
+        @place.stubs(:order).with("RAND()").returns(@place)
+        @place.stubs(:first).returns(p = places.sample)
+
+        assert_equal p, @source.sample
+      end
+
+      context "Specific instance" do
+        setup do
+          @source = Cockroach::Source::Model.new Place, {"id" => "3"}
+        end
+
+        should "return specific record only" do
+          places = ((1..10).to_a.collect {|i| stub('place', :id => i) })
+          @place.expects(:find).with("3").returns(p = places[2])
+
+          assert_equal p, @source.sample
+        end
+      end
+      
+      context "Where condition" do
+        setup do
+          @source = Cockroach::Source::Model.new Place, {"conditions" => "exists = TRUE"}
+        end
+
+        should "return specific record only" do
+          places = ((1..10).to_a.collect {|i| stub('place', :id => i) })
+          @place.stubs(:order).with("RAND()").returns(@place)
+          @place.stubs(:where).with("exists = TRUE").returns(@place)
+          @place.stubs(:first).returns(p = places.sample)
+
+          assert_equal p, @source.sample
+        end
+      end
+    end
+  end
+end

data/test/generators/install_generator_test.rb

@@ -0,0 +1,14 @@
+require "test_helper"
+require "generators/cockroach/install_generator"
+
+class InstallGeneratorTest < Rails::Generators::TestCase
+  tests Cockroach::Generators::InstallGenerator
+  destination File.expand_path("../../../tmp", __FILE__)
+  setup :prepare_destination
+
+  test "Assert all files are properly created" do
+    run_generator
+    assert_file "config/faker.yml"
+    assert_file "config/initializers/cockroach.rb"
+  end
+end

data/test/support/active_record.rb

@@ -0,0 +1,11 @@
+require 'logger'
+require "active_record"
+
+ActiveRecord::Base.establish_connection(
+  :adapter => "sqlite3",
+  :database => ':memory:'
+)
+
+ActiveRecord::Migration.verbose = false
+#ActiveRecord::Migration.verbose = true
+#ActiveRecord::Base.logger = Logger.new(STDOUT)

data/test/support/data/correct_with_option.yml

@@ -0,0 +1,13 @@
+fixture_gem: :factory_girl
+fixtures_path: test/fixtures
+
+users_amount: 1000
+feeds_ratio: 1000
+ships:
+  amount: 200
+boats:
+  ratio: 2000
+corsair:
+  ratio:
+    upper_limit: 200
+    lower_limit: 100

data/test/support/data/correct_without_option.yml

@@ -0,0 +1,10 @@
+users_amount: 1000
+feeds_ratio: 1000
+ships:
+  amount: 200
+boats:
+  ratio: 2000
+corsair:
+  ratio:
+    upper_limit: 200
+    lower_limit: 100

data/test/support/data/dummy_structure/config/faker.yml

@@ -0,0 +1,13 @@
+fixture_gem: :factory_girl
+fixtures_path: test/fixtures
+
+users_amount: 1000
+feeds_ratio: 1000
+ships:
+  amount: 200
+boats:
+  ratio: 2000
+corsair:
+  ratio:
+    upper_limit: 200
+    lower_limit: 100

data/test/support/data/dummy_structure/config/user_only.yml

@@ -0,0 +1 @@
+users_amount: 1000

data/test/support/data/dummy_structure/config/witness.yml

@@ -0,0 +1,6 @@
+witness:
+  amount: 6
+  message:
+    ratio: 3
+    author_as: witness
+    

data/test/support/data/dummy_structure/test/factories/user_factory.rb

@@ -0,0 +1,16 @@
+FactoryGirl.define do
+  factory(:user) do
+    sequence(:email) { |n| "user_number#{n}@test.com" }
+    sequence(:first_name) { |n| "user_number#{n}" }
+    sequence(:last_name) { |n| "user_number#{n}" }
+
+    factory :user_admin do
+      role :admin
+    end
+
+    factory :user_other  do
+      role :other
+    end
+  end
+end
+

data/test/support/database_cleaner.rb

@@ -0,0 +1,13 @@
+require 'database_cleaner'
+
+DatabaseCleaner.strategy = :truncation
+
+module OrmSetup
+  def before_setup
+    DatabaseCleaner.start
+  end
+
+  def after_teardown
+    DatabaseCleaner.clean
+  end
+end

data/test/support/factory_girl_mocked.rb

@@ -0,0 +1,32 @@
+#Hack to mock factory girl after it was loaded
+require 'factory_girl'
+Object.__send__(:remove_const, :FactoryGirl)
+module FactoryGirl
+  class << self
+    attr_accessor :definition_file_paths
+  end
+
+  self.definition_file_paths = %w(factories test/factories spec/factories)
+
+  def self.define
+  end
+
+  module Mock
+    def sequence_number_for(factory)
+      if (@sequences ||= {})[factory].nil?
+        @sequences[factory] ||= 0
+      else
+        @sequences[factory] += 1
+      end
+    end
+
+    def mock_factory_girl
+      ::FactoryGirl.stubs(:factories)
+      ::FactoryGirl.stubs(:find_definitions)
+      ::FactoryGirl.stubs(:factory_by_name).with("user").returns(stub('user'))
+      ::FactoryGirl.stubs(:factory_by_name).with("place").returns(stub('place'))
+      ::FactoryGirl.stubs(:factory_by_name).with("bird").returns(stub('bird'))
+      ::FactoryGirl.stubs(:factory_by_name).with("test").raises(ArgumentError.new("Factory not registered: test"))
+    end
+  end
+end

data/test/support/models/user.rb

@@ -0,0 +1,23 @@
+require 'active_record'
+
+class CreateUsersTables < ActiveRecord::Migration
+  def change
+    create_table(:users) do |t|
+      t.string :email
+      t.string :first_name
+      t.string :last_name
+      t.string :role, :default => 'user'
+
+      t.timestamps
+    end
+  end
+end
+
+CreateUsersTables.migrate(:up)
+
+class User < ActiveRecord::Base
+  validates  :first_name, :last_name, presence: true,  length: {:within => 2..100}
+  validates  :email, presence: true, uniqueness: true,  length: {:maximum => 100}
+
+  attr_accessible :email, :first_name, :last_name
+end