cassandra_object_rails 0.0.1

data/test/unit/attribute_methods/primary_key_test.rb

@@ -0,0 +1,26 @@
+require 'test_helper'
+
+class CassandraObject::AttributeMethods::PrimaryKeyTest < CassandraObject::TestCase
+  test 'get id' do
+    model = temp_object do
+      key do
+        "foo"
+      end
+    end
+    record = model.new
+
+    assert_equal 'foo', record.id
+  end
+
+  test 'set id' do
+    issue = Issue.new id: 'foo'
+
+    assert_equal 'foo', issue.id
+  end
+
+  test 'attributes' do
+    issue = Issue.new
+
+    assert_not_nil issue.attributes['id']
+  end
+end

data/test/unit/attribute_methods/typecasting_test.rb

@@ -0,0 +1,112 @@
+require 'test_helper'
+
+class CassandraObject::AttributeMethods::TypecastingTest < CassandraObject::TestCase
+  class CustomType
+  end
+
+  class CustomCoder < CassandraObject::Types::BaseType
+  end
+
+  class TestIssue < CassandraObject::Base
+    self.column_family = 'Issues'
+
+    attribute :custom_column, type: CustomType, coder: CustomCoder
+    boolean :enabled
+    float   :rating
+    integer :price
+    json    :orders
+    string  :name
+  end
+
+  class TestChildIssue < TestIssue
+    string :description
+  end
+
+  test 'attributes not shared' do
+    assert_nothing_raised { Issue.new.description }
+    assert_raise(NoMethodError) { TestIssue.new.description }
+    assert_nothing_raised { TestChildIssue.new.description }
+  end
+
+  test 'custom attribute definer' do
+    model_attribute = TestIssue.attribute_definitions[:custom_column]
+
+    assert_kind_of CustomCoder, model_attribute.coder
+    assert_equal 'custom_column', model_attribute.name
+  end
+
+  test 'typecast_attribute' do
+    assert_equal 1, TestIssue.typecast_attribute(TestIssue.new, 'price', 1)
+    assert_equal 1, TestIssue.typecast_attribute(TestIssue.new, :price, 1)
+
+    assert_raise NoMethodError do
+      TestIssue.typecast_attribute(TestIssue.new, 'wtf', 1)
+    end
+  end
+
+  test 'boolean attribute' do
+    issue = TestIssue.create! enabled: '1'
+    assert_equal true, issue.enabled
+
+    issue = TestIssue.find issue.id
+    assert_equal true, issue.enabled
+  end
+
+  test 'float attribute' do
+    issue = TestIssue.create! rating: '4.5'
+    assert_equal 4.5, issue.rating
+
+    issue = TestIssue.find issue.id
+    assert_equal(4.5, issue.rating)
+  end
+
+  test 'integer attribute' do
+    issue = TestIssue.create! price: '101'
+    assert_equal 101, issue.price
+
+    issue = TestIssue.find issue.id
+    assert_equal(101, issue.price)
+  end
+
+  test 'json attribute' do
+    issue = TestIssue.create! orders: {'a' => 'b'}
+    assert_equal({'a' => 'b'}, issue.orders)
+
+    issue = TestIssue.find issue.id
+    assert_equal({'a' => 'b'}, issue.orders)
+  end
+
+  test 'string attribute' do
+    issue = TestIssue.create! name: 'hola'
+    assert_equal('hola', issue.name)
+
+    issue = TestIssue.find issue.id
+    assert_equal('hola', issue.name)
+  end
+
+  test 'multiple attributes definition' do
+    class MultipleAttributesIssue < CassandraObject::Base
+      self.column_family = 'Issues'
+    end
+
+    assert_nothing_raised {
+      MultipleAttributesIssue.string :hello, :greetings, :bye
+    }
+    issue = MultipleAttributesIssue.new :hello => 'hey', :greetings => 'how r u', :bye => 'see ya'
+
+    assert_equal 'how r u', issue.greetings
+  end
+
+  test 'multiple attributes with options' do
+    class MultipleAttributesIssue < CassandraObject::Base
+      self.column_family = 'Issues'
+    end
+
+    MultipleAttributesIssue.expects(:attribute).with(:hello, { :unique => :true, :type => :string })
+    MultipleAttributesIssue.expects(:attribute).with(:world, { :unique => :true, :type => :string })
+
+    class MultipleAttributesIssue < CassandraObject::Base
+      string :hello, :world, :unique => :true
+    end
+  end
+end

data/test/unit/attribute_methods_test.rb

@@ -0,0 +1,39 @@
+require 'test_helper'
+
+class CassandraObject::AttributeMethodsTest < CassandraObject::TestCase
+  test 'read and write attributes' do
+    issue = Issue.new
+    assert_nil issue.read_attribute(:description)
+
+    issue.write_attribute(:description, nil)
+    assert_nil issue.read_attribute(:description)
+
+    issue.write_attribute(:description, 'foo')
+    assert_equal 'foo', issue.read_attribute(:description)
+  end
+
+  test 'hash accessor aliases' do
+    issue = Issue.new
+
+    issue[:description] = 'bar'
+
+    assert_equal 'bar', issue[:description]
+  end
+
+  # test 'attribute_exists' do
+  #   assert !Issue.new.attribute_exists?(:description)
+  #   assert Issue.new(description: nil).attribute_exists?(:description)
+  #   assert Issue.new(description: false).attribute_exists?(:description)
+  #   assert Issue.new(description: 'hey').attribute_exists?(:description)
+  # end
+
+  test 'attributes setter' do
+    issue = Issue.new
+
+    issue.attributes = {
+      description: 'foo'
+    }
+
+    assert_equal 'foo', issue.description
+  end
+end

data/test/unit/base_test.rb

@@ -0,0 +1,20 @@
+require 'test_helper'
+
+class CassandraObject::BaseTest < CassandraObject::TestCase
+  class Son < CassandraObject::Base
+  end
+
+  class Grandson < Son
+  end
+
+  test 'base_class' do
+    assert_equal CassandraObject::Base, CassandraObject::Base
+    assert_equal Son, Son.base_class
+    assert_equal Son, Grandson.base_class
+  end
+
+  test 'column family' do
+    assert_equal 'CassandraObject::BaseTest::Sons', Son.column_family
+    assert_equal 'CassandraObject::BaseTest::Sons', Grandson.column_family
+  end
+end

data/test/unit/belongs_to/reflection_test.rb

@@ -0,0 +1,12 @@
+require 'test_helper'
+
+class CassandraObject::BelongsTo::ReflectionTest < CassandraObject::TestCase
+  class ::Status < CassandraObject::Base; end
+  class ::User < CassandraObject::Base
+    belongs_to :status
+  end
+
+  test 'class_name' do
+    assert_equal 'Status', User.new.belongs_to_reflections[:status].class_name
+  end
+end

data/test/unit/belongs_to_test.rb

@@ -0,0 +1,62 @@
+require 'test_helper'
+
+class CassandraObject::BelongsToTest < CassandraObject::TestCase
+  class TestObject < Issue
+    string :issue_id
+    belongs_to :issue
+
+    string :widget_id
+    belongs_to :widget, class_name: 'Issue'
+
+    string :target_id
+    string :target_type
+    belongs_to :target, polymorphic: true
+  end
+
+  test 'belongs_to' do
+    issue = Issue.create
+
+    record = TestObject.create(issue: issue)
+
+    assert_equal issue, record.issue
+    assert_equal issue.id, record.issue_id
+
+    record = TestObject.find(record.id)
+    assert_equal issue, record.issue
+  end
+
+  test 'belongs_to with class_name' do
+    issue = Issue.create
+
+    record = TestObject.create(widget: issue)
+
+    assert_equal issue, record.widget
+    assert_equal issue.id, record.widget_id
+
+    record = TestObject.find(record.id)
+    assert_equal issue, record.widget
+  end
+
+  test 'belongs_to with polymorphic' do
+    issue = Issue.create
+
+    record = TestObject.create(target: issue)
+
+    assert_equal issue, record.target
+    assert_equal issue.id, record.target_id
+    assert_equal 'Issue', record.target_type
+
+    record = TestObject.find(record.id)
+    assert_equal issue, record.target
+  end
+
+  test 'belongs_to clear cache after reload' do
+    issue = Issue.create
+    record = TestObject.create(issue: issue)
+    issue.destroy
+
+    assert_not_nil record.issue
+    assert_nil TestObject.find(record.id).issue
+    assert_nil record.reload.issue
+  end
+end

data/test/unit/callbacks_test.rb

@@ -0,0 +1,46 @@
+require 'test_helper'
+
+class CassandraObject::CallbacksTest < CassandraObject::TestCase
+  class TestIssue < CassandraObject::Base
+    self.column_family = 'Issues'
+    string :description
+
+    %w(before_validation after_validation after_save after_create after_update after_destroy).each do |method|
+      send(method) do
+        callback_history << method
+      end
+    end
+
+    def reset_callback_history
+      @callback_history = []
+    end
+
+    def callback_history
+      @callback_history ||= []
+    end
+  end
+
+  test 'create' do
+    issue = TestIssue.create
+
+    assert_equal ['before_validation', 'after_validation', 'after_save', 'after_create'], issue.callback_history
+  end
+
+  test 'update' do
+    issue = TestIssue.create
+    issue.reset_callback_history
+
+    issue.update_attribute :description, 'foo'
+
+    assert_equal ['after_save', 'after_update'], issue.callback_history
+  end
+
+  test 'destroy' do
+    issue = TestIssue.create
+    issue.reset_callback_history
+
+    issue.destroy
+
+    assert_equal ['after_destroy'], issue.callback_history
+  end
+end

data/test/unit/config_test.rb

@@ -0,0 +1,23 @@
+require 'test_helper'
+
+class CassandraObject::ConfigTest < CassandraObject::TestCase
+  test 'config writer' do
+    config =  CassandraObject::Config.new(
+      keyspace: 'place_directory_development',
+      servers: '192.168.0.100:9160',
+      thrift: {'timeout' => 10},
+      keyspace_options: {'placement_strategy' => 'NetworkTopologyStrategy'}
+    )
+
+    assert_equal ['192.168.0.100:9160'], config.servers
+    assert_equal 'place_directory_development', config.keyspace
+    assert_equal 10, config.thrift_options[:timeout]
+    assert_equal 'NetworkTopologyStrategy', config.keyspace_options[:placement_strategy]
+  end
+
+  test 'defaults' do
+    config = CassandraObject::Config.new(keyspace: 'widget_factory')
+
+    assert_equal ["127.0.0.1:9160"], config.servers
+  end
+end

data/test/unit/connection_test.rb

@@ -0,0 +1,10 @@
+require 'test_helper'
+
+class CassandraObject::ConnectionTest < CassandraObject::TestCase
+  class TestObject < CassandraObject::Base
+  end
+
+  test "sanitize supports question marks" do
+    assert_equal 'hello ?', CassandraCQL::Statement.sanitize('hello ?')
+  end
+end

data/test/unit/consistency_test.rb

@@ -0,0 +1,13 @@
+require "test_helper"
+
+class CassandraObject::ConsistencyTest < CassandraObject::TestCase
+  test "with_consistency" do
+    original = CassandraObject::Base.default_consistency
+
+    CassandraObject::Base.with_consistency 'LOL' do
+      assert_equal 'LOL', CassandraObject::Base.default_consistency
+    end
+
+    assert_equal original, CassandraObject::Base.default_consistency
+  end
+end

data/test/unit/core_test.rb

@@ -0,0 +1,55 @@
+require 'test_helper'
+
+class CassandraObject::CoreTest < CassandraObject::TestCase
+  test 'initialiaze' do
+    issue = Issue.new
+
+    assert issue.new_record?
+    assert !issue.destroyed?
+  end
+
+  test 'initialize yields self' do
+    issue = Issue.new { |i| i.description = 'bar' }
+    assert_equal 'bar', issue.description
+  end
+
+  test 'dup' do
+    issue = Issue.create description: 'foo'
+
+    dup_issue = issue.dup
+
+    assert dup_issue.new_record?
+    assert_not_equal issue.id, dup_issue.id
+    assert_nil dup_issue.created_at
+    assert_nil dup_issue.updated_at
+    assert_equal 'foo', issue.description
+  end
+
+  test 'equality of new records' do
+    assert_not_equal Issue.new, Issue.new
+  end
+
+  test 'equality' do
+    first_issue = Issue.create
+    second_issue = Issue.create
+
+    assert_equal first_issue, first_issue
+    assert_equal first_issue, Issue.find(first_issue.id)
+    assert_not_equal first_issue, second_issue
+  end
+
+  test 'to_param' do
+    issue = Issue.new
+    assert_equal issue.id, issue.to_param
+  end
+
+  test 'hash' do
+    issue = Issue.create
+    assert_equal issue.id.hash, issue.hash
+  end
+
+  test 'inspect' do
+    issue = Issue.create
+    assert issue.inspect =~ /^#<Issue id: \"\w+\", created_at: \".+\", updated_at: \".+\", description: \".+\">$/
+  end
+end

data/test/unit/identity_test.rb

@@ -0,0 +1,26 @@
+require 'test_helper'
+
+class CassandraObject::IdentityTest < CassandraObject::TestCase
+  test 'primary_key' do
+    assert_equal 'id', Issue.primary_key
+  end
+
+  test 'default _generate_key' do
+    issue = Issue.new
+
+    assert_not_nil Issue._generate_key(issue)
+  end
+
+  test 'custom key' do
+    model = temp_object do
+      key do
+        "name:#{name}"
+      end
+      attr_accessor :name
+    end
+    record = model.new
+    record.name = 'bar'
+
+    assert_equal 'name:bar', model._generate_key(record)
+  end
+end