extendi-cassandra_object 1.0.0

data/test/unit/persistence_test.rb

@@ -0,0 +1,266 @@
+#!/bin/env ruby
+# encoding: utf-8
+
+require 'test_helper'
+
+class CassandraObject::PersistenceTest < CassandraObject::TestCase
+  test 'instantiate removes unknowns' do
+    assert_nil Issue.instantiate('theid', 'z' => 'nooo').attributes['z']
+  end
+
+  test 'encode_attributes' do
+    assert_equal(
+      {},
+      Issue.encode_attributes({})
+    )
+
+    assert_equal(
+      {'description' => nil},
+      Issue.encode_attributes({'description' => nil})
+    )
+
+    assert_equal(
+      {'description' => 'lol'},
+      Issue.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 do
+      Issue.find(first_issue.id)
+    end
+    assert_nothing_raised do
+      Issue.find(second_issue.id)
+    end
+  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
+    begin
+      Issue.validates(:description, presence: true)
+
+      record = Issue.new(description: 'bad')
+      record.save!
+
+      assert_raise CassandraObject::RecordInvalid do
+        record = Issue.new
+        record.save!
+      end
+    ensure
+      Issue.reset_callbacks(:validate)
+    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 '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 'remove' do
+    record = Issue.new(title: 'cool')
+    record.save!
+
+    id = record.id
+    assert_equal id, Issue.find(id).id
+
+    Issue.remove(id)
+
+    assert_raise CassandraObject::RecordNotFound do
+      Issue.find(id)
+    end
+  end
+
+  test 'remove multiple' do
+    ids = []
+    (1..10).each do
+      record = Issue.create!(title: 'cool')
+      ids << record.id
+    end
+
+    Issue.remove(ids)
+
+    assert_equal [], Issue.find(ids)
+  end
+
+  test 'ttl' do
+    record = Issue.create({title: 'name', ttl: 1})
+    assert_nothing_raised do
+      Issue.find(record.id)
+    end
+
+    sleep 2
+
+    assert_raise CassandraObject::RecordNotFound do
+      Issue.find(record.id)
+    end
+  end
+
+  test 'dynamic create' do
+
+    id1 = "1"
+    IssueDynamic.create(key: id1, title: 'tit', dynamic_field1: 'one', dynamic_field2: 'two')
+    id2 = "2"
+    IssueDynamic.create(key: id2, title: 'tit2', dynamic_field1: '1', dynamic_field2: '2')
+    # number of dynamic fields
+
+    assert_equal 3, IssueDynamic.find(id1)[id1].size
+  end
+
+  test 'dynamic update' do
+
+    id = "123"
+    IssueDynamic.create(key: id, title: 'tit', dynamic_field1: 'one', dynamic_field2: 'two')
+    assert_equal 3, IssueDynamic.find(id)[id].size
+
+    IssueDynamic.update(id, {title: 'tit_new', dynamic_field1: 'new_one', dynamic_field2: nil})
+    assert_equal 2, IssueDynamic.find(id)[id].size
+
+  end
+
+  test 'dynamic delete' do
+    id = "123"
+    IssueDynamic.create(key: id, title: 'tit', dynamic_field1: 'one', dynamic_field2: 'two')
+    IssueDynamic.delete(id)
+    assert_raise CassandraObject::RecordNotFound do
+      IssueDynamic.find(id)
+    end
+  end
+
+  test 'dynamic delete attributes' do
+    id = "123"
+    IssueDynamic.create(key: id, title: 'tit', dynamic_field1: 'one', dynamic_field2: 'two')
+    IssueDynamic.delete(id, [:dynamic_field1])
+    assert_equal 2, IssueDynamic.find(id)[id].size
+  end
+
+  test 'paged_request_dynamic' do
+
+    NUMTEST = 21000
+    KEY = '987987'
+
+    NUMTEST.times.each do |i|
+      rand = rand()
+      IssueDynamic.create(key: KEY, rand => rand)
+    end
+    found = IssueDynamic.find_by_id(KEY)
+
+    assert_equal NUMTEST, found[KEY].size
+  end
+
+end

data/test/unit/railties/controller_runtime_test.rb

@@ -0,0 +1,48 @@
+require 'test_helper'
+require 'cassandra_object/railties/controller_runtime'
+
+class CassandraObject::Railties::ControllerRuntimeTest < MiniTest::Test
+  class TestRuntime
+    def self.log_process_action(payload)
+      ['sweet']
+    end
+
+    def cleanup_view_runtime
+      12
+    end
+
+    def append_info_to_payload(payload)
+      payload[:foo] = 42
+    end
+  end
+
+  class CassandraRuntime < TestRuntime
+    include CassandraObject::Railties::ControllerRuntime
+  end
+
+  def test_cleanup_view_runtime
+    runtime = CassandraRuntime.new
+    CassandraObject::LogSubscriber.runtime = 10
+
+    runtime.cleanup_view_runtime
+
+    assert_equal 0, CassandraObject::LogSubscriber.runtime
+  end
+
+  def test_append_info_to_payload
+    runtime = CassandraRuntime.new
+    payload = {}
+    runtime.append_info_to_payload(payload)
+
+    assert_equal 42, payload[:foo]
+    assert payload.key?(:cassandra_object_runtime)
+  end
+
+  def test_log_process_action
+    payload = {cassandra_object_runtime: 12.3}
+    messages = CassandraRuntime.log_process_action(payload)
+
+    assert_equal 2, messages.size
+    assert_equal 'CassandraObject: 12.3ms', messages.last
+  end
+end

data/test/unit/schema/tasks_test.rb

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

data/test/unit/schema_test.rb

@@ -0,0 +1,115 @@
+require 'test_helper'
+
+class CassandraObject::SchemaTest < CassandraObject::TestCase
+
+  SCHEMA = {attributes: 'key text, text_field text, integer_field int ,float_field float, created_at timestamp, updated_at timestamp, PRIMARY KEY (key)'}
+
+  test 'create_keyspace' do
+    CassandraObject::Schema.create_keyspace 'Blah'
+    begin
+      existing_keyspace = false
+      CassandraObject::Schema.create_keyspace 'Blah'
+    rescue Exception => e
+      assert_equal e.message, 'Cannot add existing keyspace "blah"'
+      existing_keyspace = true
+    ensure
+      CassandraObject::Schema.drop_keyspace 'Blah'
+    end
+
+    assert existing_keyspace
+  end
+
+  test 'drop undroppable' do
+    begin
+      CassandraObject::Schema.drop_keyspace 'cassandra_object_test'
+    rescue Exception => e
+      assert_equal e.message, 'Cannot drop keyspace cassandra_object_test. You must delete all tables before'
+    ensure
+    end
+  end
+
+  test 'create_table' do
+    CassandraObject::Schema.create_table 'TestSchemaRecords', SCHEMA
+
+    begin
+      CassandraObject::Schema.create_table 'TestSchemaRecords', SCHEMA
+      assert false, 'TestSchemaRecords should already exist'
+    rescue Exception => e
+      assert_equal e.message.gsub('column family', 'table'), 'Cannot add already existing table "testschemarecords" to keyspace "cassandra_object_test"'
+    end
+  end
+
+  test 'drop_table' do
+    CassandraObject::Schema.create_table 'TestSchemaCFToDrop1', SCHEMA
+
+    CassandraObject::Schema.drop_table 'TestSchemaCFToDrop1'
+
+    begin
+      CassandraObject::Schema.drop_table 'TestSchemaCFToDrop1'
+      assert false, 'TestSchemaCFToDrop1 should not exist'
+    rescue Exception => e
+      assert_equal e.message.gsub('columnfamily', 'table'), 'unconfigured table testschemacftodrop1'
+    end
+  end
+
+  test 'test drop with record' do
+    class TestDrop < CassandraObject::BaseSchema
+      self.column_family = 'TestSchemaCFToDrop2'
+    end
+
+    CassandraObject::Schema.create_table 'TestSchemaCFToDrop2', SCHEMA
+    TestDrop.create
+
+    begin
+      CassandraObject::Schema.drop_table 'TestSchemaCFToDrop2'
+    rescue Exception => e
+      assert_equal e.message, 'The table TestSchemaCFToDrop2 is not empty! If you want to drop it add the option confirm = true'
+    end
+  end
+
+  test 'test drop with confirm' do
+    CassandraObject::Schema.create_table 'TestSchemaCFToDrop3', SCHEMA
+
+    CassandraObject::Schema.drop_table 'TestSchemaCFToDrop3', true
+    begin
+      CassandraObject::Schema.drop_table 'TestSchemaCFToDrop3'
+      assert false, 'TestSchemaCFToDrop should not exist'
+    rescue Exception => e
+      assert_equal e.message.gsub('columnfamily', 'table'), 'unconfigured table testschemacftodrop3'
+    end
+  end
+
+  test 'test drop empty' do
+    CassandraObject::Schema.create_table 'TestSchemaCFToDrop4', SCHEMA
+    # drop empty
+    CassandraObject::Schema.drop_table 'TestSchemaCFToDrop4'
+    begin
+      CassandraObject::Schema.drop_table 'TestSchemaCFToDrop4'
+      assert false, 'TestSchemaCFToDrop4 should not exist'
+    rescue Exception => e
+      assert_equal e.message.gsub('columnfamily', 'table'), 'unconfigured table testschemacftodrop4'
+    end
+  end
+
+  test 'create_index' do
+    CassandraObject::Schema.create_column_family 'TestSchemaIndexed', SCHEMA
+
+    CassandraObject::Schema.alter_column_family 'TestSchemaIndexed', 'ADD id_value varchar'
+
+    CassandraObject::Schema.add_index 'TestSchemaIndexed', 'id_value'
+  end
+
+  test 'drop_index' do
+    CassandraObject::Schema.create_column_family 'TestSchemaDropIndexes', SCHEMA
+
+    CassandraObject::Schema.alter_column_family 'TestSchemaDropIndexes', 'ADD id_value1 varchar'
+    CassandraObject::Schema.alter_column_family 'TestSchemaDropIndexes', 'ADD id_value2 varchar'
+
+    CassandraObject::Schema.add_index 'TestSchemaDropIndexes', 'id_value1'
+    CassandraObject::Schema.add_index 'TestSchemaDropIndexes', 'id_value2', 'special_name'
+
+    CassandraObject::Schema.drop_index 'TestSchemaDropIndexes_id_value1_idx'
+    CassandraObject::Schema.drop_index 'special_name'
+  end
+
+end