extendi-cassandra_object 1.0.0

data/test/support/issue.rb

@@ -0,0 +1,12 @@
+class Issue < CassandraObject::BaseSchemaless
+  string :description
+  string :title
+
+  before_create { self.description ||= 'funny' }
+
+  self.allow_filtering = true
+
+  def self.for_key key
+    where_ids(key)
+  end
+end

data/test/support/issue_dynamic.rb

@@ -0,0 +1,12 @@
+class IssueDynamic < CassandraObject::BaseSchemalessDynamic
+  string :description
+  string :title
+
+  before_create { self.description ||= 'funny' }
+
+  self.allow_filtering = true
+
+  def self.for_key key
+    where_ids(key)
+  end
+end

data/test/support/issue_schema.rb

@@ -0,0 +1,17 @@
+class IssueSchema < CassandraObject::BaseSchema
+  string :id
+  string :description
+  string :title
+  float :field
+  integer :intero
+
+  before_create { self.description ||= 'funny' }
+
+  validates :title, presence: true
+
+  self.keys = '(id)'
+
+  def self.for_key key
+    where_ids(key)
+  end
+end

data/test/support/issue_schema_child.rb

@@ -0,0 +1,17 @@
+class IssueSchemaChild < CassandraObject::BaseSchema
+  string :id
+  string :description
+  string :title
+  float :field
+
+  belongs_to :issue_schema_father
+  string :issue_schema_father_id
+
+  validates :title, presence: true
+
+  self.keys = '(id)'
+
+  def self.for_key key
+    where_ids(key)
+  end
+end

data/test/support/issue_schema_father.rb

@@ -0,0 +1,13 @@
+class IssueSchemaFather < CassandraObject::BaseSchema
+  string :id
+  string :title
+  float :field
+
+  validates :title, presence: true
+
+  self.keys = '(id)'
+
+  def self.for_key key
+    where_ids(key)
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,41 @@
+require 'bundler/setup'
+Bundler.require(:default, :test)
+
+require 'rails/test_help'
+require 'mocha/setup'
+
+require 'cassandra_object'
+require 'support/cassandra'
+require 'support/issue'
+require 'support/issue_dynamic'
+require 'support/issue_schema'
+require 'support/issue_schema_child'
+require 'support/issue_schema_father'
+
+module CassandraObject
+  class TestCase < ActiveSupport::TestCase
+
+    def temp_object(&block)
+      Class.new(CassandraObject::Base) do
+        self.column_family = 'Issues'
+        string :force_save
+        before_save { self.force_save = 'junk' }
+
+        def self.name
+          'Issue'
+        end
+
+        instance_eval(&block) if block_given?
+      end
+    end
+  end
+
+  module Types
+    class TestCase < CassandraObject::TestCase
+      attr_accessor :coder
+      setup do
+        @coder = self.class.name.sub(/Test$/, '').constantize.new
+      end
+    end
+  end
+end

data/test/unit/active_model_test.rb

@@ -0,0 +1,18 @@
+require 'test_helper'
+
+class ActiveModelTest < CassandraObject::TestCase
+
+  include ActiveModel::Lint::Tests
+
+  # overrides ActiveModel::Lint::Tests#test_to_param
+  def test_to_param
+  end
+
+  # overrides ActiveModel::Lint::Tests#test_to_key
+  def test_to_key
+  end
+
+  def setup
+    @model = Issue.new
+  end
+end

data/test/unit/adapters/adapter_test.rb

@@ -0,0 +1,6 @@
+require 'test_helper'
+
+class CassandraObject::Adapters::AdapterTest < CassandraObject::TestCase
+  test 'create_table' do
+  end
+end

data/test/unit/attribute_methods/definition_test.rb

@@ -0,0 +1,13 @@
+require 'test_helper'
+
+class CassandraObject::AttributeMethods::DefinitionTest < CassandraObject::TestCase
+  class TestType < CassandraObject::Types::BaseType
+  end
+
+  test 'typecast' do
+    definition = CassandraObject::AttributeMethods::Definition.new(:foo, TestType, {a: :b})
+
+    assert_equal 'foo', definition.name
+    assert_kind_of TestType, definition.coder
+  end
+end

data/test/unit/attribute_methods/dirty_test.rb

@@ -0,0 +1,72 @@
+# require 'test_helper'
+#
+# class CassandraObject::AttributeMethods::DirtyTest < CassandraObject::TestCase
+#   test 'save clears dirty' do
+#     record = temp_object do
+#       string :name
+#     end.new name: 'foo'
+#
+#     assert record.changed?
+#
+#     record.save!
+#
+#     assert_equal [nil, 'foo'], record.previous_changes['name']
+#     assert !record.changed?
+#   end
+#
+#   test 'reload clears dirty' do
+#     record = temp_object do
+#       string :name
+#     end.create! name: 'foo'
+#
+#     record.name = 'bar'
+#     assert record.changed?
+#
+#     record.reload
+#
+#     assert !record.changed?
+#   end
+#
+#   test 'typecast float before dirty check' do
+#     record = temp_object do
+#       float :price
+#     end.create(price: 5.01)
+#
+#     record.price = '5.01'
+#     assert !record.changed?
+#
+#     record.price = '7.12'
+#     assert record.changed?
+#   end
+#
+#   test 'typecast boolean before dirty check' do
+#     record = temp_object do
+#       boolean :awesome
+#     end.create(awesome: false)
+#
+#     record.awesome = false
+#     assert !record.changed?
+#
+#     record.awesome = true
+#     assert record.changed?
+#   end
+#
+#   test 'write_attribute' do
+#     object = temp_object do
+#       string :name
+#     end
+#
+#     expected = {"name"=>[nil, "foo"]}
+#
+#     object.new.tap do |record|
+#       record.name = 'foo'
+#       assert_equal expected, record.changes
+#     end
+#
+#     object.new.tap do |record|
+#       record[:name] = 'foo'
+#       # record.write_attribute(:name, 'foo')
+#       assert_equal expected, record.changes
+#     end
+#   end
+# end

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,119 @@
+require 'test_helper'
+
+class CassandraObject::AttributeMethods::TypecastingTest < CassandraObject::TestCase
+  class CustomType
+  end
+
+  class CustomCoder < CassandraObject::Types::BaseType
+  end
+
+  class TestIssue < CassandraObject::BaseSchemaless
+    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)
+
+    issue = TestIssue.new price: ''
+    assert_nil 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)
+
+    issue = TestIssue.create! name: 42
+    assert_equal '42', 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,51 @@
+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 'attributes setter' do
+    issue = Issue.new
+
+    issue.attributes = {
+      description: 'foo'
+    }
+
+    assert_equal 'foo', issue.description
+  end
+
+  class ChildIssue < Issue
+    def title=(val)
+      self[:title] = val + ' lol'
+    end
+  end
+
+  test 'override' do
+    issue = ChildIssue.new(title: 'hey')
+
+    assert_equal 'hey lol', issue.title
+  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
+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