cassandra_object_rails 0.0.1

data/test/unit/inspect_test.rb

@@ -0,0 +1,26 @@
+require 'test_helper'
+
+class CassandraObject::InspectTest < CassandraObject::TestCase
+  test 'attribute_for_inspect' do
+    object = temp_object do
+      string :long_string
+      time :the_time
+      integer :other
+    end
+
+    assert_equal "#{'x' * 51}...".inspect, object.new.attribute_for_inspect('x' * 100)
+    assert_equal "2012-02-14 12:01:02".inspect, object.new.attribute_for_inspect(Time.new(2012, 02, 14, 12, 01, 02))
+    assert_equal "\"foo\"", object.new.attribute_for_inspect('foo')
+  end
+
+  test 'inspect' do
+    object = temp_object do
+      string :description
+      integer :price
+    end.new(description: "yeah buddy", price: nil)
+
+    assert_match /id/, object.inspect
+    assert_match /description/, object.inspect
+    assert_no_match /price/, object.inspect
+  end
+end

data/test/unit/log_subscriber_test.rb

@@ -0,0 +1,22 @@
+require "test_helper"
+require "active_support/log_subscriber/test_helper"
+require "cassandra_object/log_subscriber"
+
+class CassandraObject::LogSubscriberTest < CassandraObject::TestCase
+  include ActiveSupport::LogSubscriber::TestHelper
+
+  def setup
+    super
+
+    CassandraObject::LogSubscriber.attach_to :cassandra_object
+  end
+
+  def test_cql_notification
+    Issue.execute_cql "SELECT * FROM Issues"
+
+    wait
+
+    assert_equal 1, @logger.logged(:debug).size
+    assert_match "SELECT * FROM Issues", @logger.logged(:debug)[0]
+  end
+end

data/test/unit/persistence_test.rb

@@ -0,0 +1,187 @@
+#!/bin/env ruby
+# encoding: utf-8
+
+require 'test_helper'
+
+class CassandraObject::PersistenceTest < CassandraObject::TestCase
+  test 'encode_attributes' do
+    klass = temp_object do
+      string :description
+    end
+
+    assert_equal(
+      {},
+      klass.encode_attributes({})
+    )
+
+    assert_equal(
+      {},
+      klass.encode_attributes({description: nil})
+    )
+
+    assert_equal(
+      {'description' => 'lol'},
+      klass.encode_attributes({description: 'lol'})
+    )
+  end
+
+  test "batch" do
+    first_issue = second_issue = nil
+
+    Issue.batch do
+      assert Issue.batching?
+
+      first_issue = Issue.create
+      second_issue = Issue.create
+
+      assert_raise(CassandraObject::RecordNotFound) { Issue.find(first_issue.id) }
+      assert_raise(CassandraObject::RecordNotFound) { Issue.find(second_issue.id) }
+    end
+
+    assert !Issue.batching?
+    assert_nothing_raised(CassandraObject::RecordNotFound) { Issue.find(first_issue.id) }
+    assert_nothing_raised(CassandraObject::RecordNotFound) { Issue.find(second_issue.id) }
+  end
+
+  test 'persistance inquiries' do
+    issue = Issue.new
+    assert issue.new_record?
+    assert !issue.persisted?
+
+    issue.save
+    assert issue.persisted?
+    assert !issue.new_record?
+  end
+
+  test 'create' do
+    issue = Issue.create { |i| i.description = 'foo' }
+    assert_equal 'foo', issue.description
+    assert_equal 'foo', Issue.find(issue.id).description
+  end
+
+  test 'read and write UTF' do
+    utf = "\ucba1\ucba2\ucba3 ƒ´∑ƒ©√åµ≈√ˆअनुच्छेद´µøµø¬≤ 汉语漢語".force_encoding(Encoding::UTF_8)
+
+    issue = Issue.create { |i| i.description = utf }
+    assert_equal utf, issue.description
+    reloaded = Issue.find(issue.id).description
+    assert_equal utf, reloaded
+  end
+
+  test 'save' do
+    issue = Issue.new
+    issue.save
+
+    assert_equal issue, Issue.find(issue.id)
+  end
+
+  test 'save!' do
+    klass = temp_object do
+      string :description
+      validates :description, presence: true
+    end
+
+    record = klass.new(description: 'bad')
+    record.save!
+
+    assert_raise CassandraObject::RecordInvalid do
+      record = klass.new
+      record.save!
+    end
+  end
+
+  test 'destroy' do
+    issue = Issue.create
+    issue.destroy
+
+    assert issue.destroyed?
+    assert !issue.persisted?
+    assert !issue.new_record?
+  end
+
+  test 'update_attribute' do
+    issue = Issue.create
+    issue.update_attribute(:description, 'lol')
+
+    assert !issue.changed?
+    assert_equal 'lol', issue.description
+  end
+
+  test 'update_attributes' do
+    issue = Issue.create
+    issue.update_attributes(description: 'lol')
+
+    assert !issue.changed?
+    assert_equal 'lol', issue.description
+  end
+
+  test 'update_attributes!' do
+    begin
+      Issue.validates(:description, presence: true)
+
+      issue = Issue.new(description: 'bad')
+      issue.save!
+
+      assert_raise CassandraObject::RecordInvalid do
+        issue.update_attributes! description: ''
+      end
+    ensure
+      Issue.reset_callbacks(:validate)
+    end
+  end
+
+  test 'update nil attributes' do
+    issue = Issue.create(title: 'I rule', description: 'lololol')
+
+    issue.update_attributes title: nil
+
+    issue = Issue.find issue.id
+    assert_nil issue.title
+  end
+
+  test 'becomes' do
+    klass = temp_object do
+    end
+
+    assert_kind_of klass, Issue.new.becomes(klass)
+  end
+
+  test 'reload' do
+    persisted_issue = Issue.create
+    fresh_issue = Issue.find(persisted_issue.id)
+    fresh_issue.update_attribute(:description, 'say what')
+
+    reloaded_issue = persisted_issue.reload
+    assert_equal 'say what', persisted_issue.description
+    assert_equal persisted_issue, reloaded_issue
+  end
+
+  test 'delete with consistency' do
+    issue = Issue.create
+    Issue.with_consistency 'QUORUM' do
+      issue.destroy
+    end
+  end
+
+  test 'insert with consistency' do
+    Issue.with_consistency 'QUORUM' do
+      Issue.create
+    end
+  end
+
+  test 'allow CQL keyword in column name' do
+    assert_nothing_raised do
+      Issue.string :text
+      issue = Issue.create :text => 'hello'
+      issue.text = 'world'
+      issue.save!
+      issue.text = nil
+      issue.save!
+    end
+  end
+
+  test 'quote_columns' do
+    klass = Class.new { include CassandraObject::Persistence }
+    assert_equal %w{'a' 'b'}, klass.__send__(:quote_columns, %w{a b})
+  end
+end

data/test/unit/savepoints_test.rb

@@ -0,0 +1,35 @@
+require 'test_helper'
+
+class CassandraObject::SavepointsTest < CassandraObject::TestCase
+  test 'rollback create' do
+    Issue.savepoint do
+      issue = Issue.create description: 'foo'
+      raise 'lol'
+    end
+
+    assert_nil Issue.first
+  end
+
+  test 'rollback update' do
+    issue = Issue.create description: 'foo'
+
+    Issue.savepoint do
+      issue.update_attributes description: 'bar'
+      raise 'lol'
+    end
+
+    issue = Issue.find issue.id
+    assert_equal 'foo', issue.description
+  end
+
+  test 'rollback destroy' do
+    issue = Issue.create description: 'foo'
+
+    Issue.savepoint do
+      issue.destroy
+      raise 'lol'
+    end
+
+    assert_nothing_raised { Issue.find issue.id }
+  end
+end

data/test/unit/schema/tasks_test.rb

@@ -0,0 +1,29 @@
+require 'test_helper'
+
+class CassandraObject::Schema::TasksTest < CassandraObject::TestCase
+  test "column_families" do
+    assert_equal ['Issues'], CassandraObject::Schema.column_families
+  end
+
+  test "dump" do
+    io = StringIO.new
+
+    CassandraObject::Schema.dump(io)
+    io.rewind
+
+    assert_match /Issues/, io.read
+  end
+
+  test "load" do
+    CassandraObject::Base.expects(:execute_cql).with("DO STUFF;")
+    CassandraObject::Base.expects(:execute_cql).with("AND MORE;")
+
+    CassandraObject::Schema.load StringIO.new(
+      "DO\n" +
+      " STUFF;\n" +
+      "\n" +
+      "AND\n" +
+      " MORE;\n"
+    )
+  end
+end

data/test/unit/schema_test.rb

@@ -0,0 +1,47 @@
+require 'test_helper'
+
+class CassandraObject::SchemaTest < CassandraObject::TestCase
+  test "create_column_family" do
+    CassandraObject::Schema.create_column_family 'TestRecords', 'compression_parameters:sstable_compression' => 'SnappyCompressor'
+
+    begin
+      CassandraObject::Schema.create_column_family 'TestRecords'
+      assert false, 'TestRecords should already exist'
+    rescue Exception => e
+    end
+  end
+
+  test "drop_column_family" do
+    CassandraObject::Schema.create_column_family 'TestCFToDrop'
+
+    CassandraObject::Schema.drop_column_family 'TestCFToDrop'
+
+    begin
+      CassandraObject::Schema.drop_column_family 'TestCFToDrop'
+      assert false, 'TestCFToDrop should not exist'
+    rescue Exception => e
+    end
+  end
+
+  test "create_index" do
+    CassandraObject::Schema.create_column_family 'TestIndexed'
+
+    CassandraObject::Schema.alter_column_family 'TestIndexed', "ADD id_value varchar"
+
+    CassandraObject::Schema.add_index 'TestIndexed', 'id_value'
+  end
+
+  test "drop_index" do
+    CassandraObject::Schema.create_column_family 'TestDropIndexes'
+
+    CassandraObject::Schema.alter_column_family 'TestDropIndexes', "ADD id_value1 varchar"
+    CassandraObject::Schema.alter_column_family 'TestDropIndexes', "ADD id_value2 varchar"
+
+    CassandraObject::Schema.add_index 'TestDropIndexes', 'id_value1'
+    CassandraObject::Schema.add_index 'TestDropIndexes', 'id_value2', 'special_name'
+
+    CassandraObject::Schema.drop_index 'TestDropIndexes_id_value1_idx'
+    CassandraObject::Schema.drop_index 'special_name'
+  end
+
+end

data/test/unit/scope/batches_test.rb

@@ -0,0 +1,30 @@
+require 'test_helper'
+
+class CassandraObject::BatchesTest < CassandraObject::TestCase
+  test 'find_each' do
+    Issue.create
+    Issue.create
+
+    issues = []
+    Issue.find_each do |issue|
+      issues << issue
+    end
+
+    assert_equal Issue.all.to_set, issues.to_set
+  end
+
+  test 'find_in_batches' do
+    Issue.create
+    Issue.create
+    Issue.create
+
+    issue_batches = []
+    Issue.find_in_batches(batch_size: 2) do |issues|
+      issue_batches << issues
+    end
+
+    assert_equal 2, issue_batches.size
+    assert issue_batches.any? { |issues| issues.size == 2 }
+    assert issue_batches.any? { |issues| issues.size == 1 }
+  end
+end

data/test/unit/scope/finder_methods_test.rb

@@ -0,0 +1,51 @@
+require 'test_helper'
+
+class CassandraObject::FinderMethodsTest < CassandraObject::TestCase
+  test 'find' do
+    Issue.create.tap do |issue|
+      assert_equal issue, Issue.find(issue.id)
+    end
+
+    begin
+      Issue.find(nil)
+      assert false
+    rescue => e
+      assert_equal "Couldn't find Issue with key nil", e.message
+    end
+
+    assert_raise CassandraObject::RecordNotFound do
+      Issue.find('what')
+    end
+  end
+
+  test 'find with ids' do
+    first_issue = Issue.create
+    second_issue = Issue.create
+    third_issue = Issue.create
+
+    assert_equal [], Issue.find([])
+    assert_equal [first_issue, second_issue], Issue.find([first_issue.id, second_issue.id])
+  end
+
+  test 'find_by_id' do
+    Issue.create.tap do |issue|
+      assert_equal issue, Issue.find_by_id(issue.id)
+    end
+
+    assert_nil Issue.find_by_id('what')
+  end
+
+  test 'all' do
+    first_issue = Issue.create
+    second_issue = Issue.create
+
+    assert_equal [first_issue, second_issue].to_set, Issue.all.to_set
+  end
+
+  test 'first' do
+    first_issue = Issue.create
+    second_issue = Issue.create
+
+    assert [first_issue, second_issue].include?(Issue.first)
+  end
+end

data/test/unit/scope/query_methods_test.rb

@@ -0,0 +1,26 @@
+require 'test_helper'
+
+class CassandraObject::Scope::QueryMethodsTest < CassandraObject::TestCase
+  test "select" do
+    original_issue = Issue.create title: 'foo', description: 'bar'
+
+    found_issue = Issue.select(:title).find(original_issue.id)
+
+    assert_equal 'foo', found_issue.title
+    assert_equal original_issue.id, found_issue.id
+    assert_nil found_issue.description
+  end
+
+  test "select with block" do
+    foo_issue = Issue.create title: 'foo'
+    bar_issue = Issue.create title: 'bar'
+
+    assert_equal [foo_issue], Issue.select { |issue| issue.title == 'foo' }
+  end
+
+  test "select with consistency" do
+    Issue.with_consistency 'ONE' do
+      Issue.all
+    end
+  end
+end