superstore 1.0.0

data/test/unit/attribute_methods_test.rb

@@ -0,0 +1,51 @@
+require 'test_helper'
+
+class Superstore::AttributeMethodsTest < Superstore::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 Superstore::BaseTest < Superstore::TestCase
+  class Son < Superstore::Base
+  end
+
+  class Grandson < Son
+  end
+
+  test 'base_class' do
+    assert_equal Superstore::Base, Superstore::Base
+    assert_equal Son, Son.base_class
+    assert_equal Son, Grandson.base_class
+  end
+
+  test 'column family' do
+    assert_equal 'Superstore::BaseTest::Sons', Son.column_family
+    assert_equal 'Superstore::BaseTest::Sons', Grandson.column_family
+  end
+end

data/test/unit/belongs_to/reflection_test.rb

@@ -0,0 +1,12 @@
+require 'test_helper'
+
+class Superstore::BelongsTo::ReflectionTest < Superstore::TestCase
+  class ::Status < Superstore::Base; end
+  class ::User < Superstore::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 Superstore::BelongsToTest < Superstore::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 Superstore::CallbacksTest < Superstore::TestCase
+  class TestIssue < Superstore::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/cassandra_schema/statements_test.rb

@@ -0,0 +1,47 @@
+require 'test_helper'
+
+class Superstore::CassandraSchema::StatementsTest < Superstore::TestCase
+  if Superstore::Base.adapter.is_a?(Superstore::Adapters::CassandraAdapter)
+    test "create_keyspace" do
+      Superstore::CassandraSchema.create_keyspace 'Blah'
+
+      existing_keyspace = false
+      begin
+        Superstore::CassandraSchema.create_keyspace 'Blah'
+      rescue Exception => e
+        existing_keyspace = true
+      ensure
+        Superstore::CassandraSchema.drop_keyspace 'Blah'
+      end
+
+      assert existing_keyspace
+    end
+
+    test "add_index" do
+      begin
+        Superstore::CassandraSchema.create_table 'TestIndexed'
+        Superstore::CassandraSchema.alter_column_family 'TestIndexed', "ADD id_value varchar"
+        Superstore::CassandraSchema.add_index 'TestIndexed', 'id_value'
+      ensure
+        Superstore::CassandraSchema.drop_table 'TestIndexed'
+      end
+    end
+
+    test "drop_index" do
+      begin
+        Superstore::CassandraSchema.create_table 'TestDropIndexes'
+
+        Superstore::CassandraSchema.alter_column_family 'TestDropIndexes', "ADD id_value1 varchar"
+        Superstore::CassandraSchema.alter_column_family 'TestDropIndexes', "ADD id_value2 varchar"
+
+        Superstore::CassandraSchema.add_index 'TestDropIndexes', 'id_value1'
+        Superstore::CassandraSchema.add_index 'TestDropIndexes', 'id_value2', 'special_name'
+
+        Superstore::CassandraSchema.drop_index 'TestDropIndexes_id_value1_idx'
+        Superstore::CassandraSchema.drop_index 'special_name'
+      ensure
+        Superstore::CassandraSchema.drop_table 'TestDropIndexes'
+      end
+    end
+  end
+end

data/test/unit/cassandra_schema/tasks_test.rb

@@ -0,0 +1,31 @@
+require 'test_helper'
+
+class Superstore::CassandraSchema::TasksTest < Superstore::TestCase
+  if Superstore::Base.adapter.is_a?(Superstore::Adapters::CassandraAdapter)
+    test "table_names" do
+      assert_equal ['Issues'], Superstore::CassandraSchema.table_names
+    end
+
+    test "dump" do
+      io = StringIO.new
+
+      Superstore::CassandraSchema.dump(io)
+      io.rewind
+
+      assert_match /Issues/, io.read
+    end
+
+    test "load" do
+      Superstore::CassandraSchema.expects(:keyspace_execute).with("DO STUFF;")
+      Superstore::CassandraSchema.expects(:keyspace_execute).with("AND MORE;")
+
+      Superstore::CassandraSchema.load StringIO.new(
+        "DO\n" +
+        " STUFF;\n" +
+        "\n" +
+        "AND\n" +
+        " MORE;\n"
+      )
+    end
+  end
+end

data/test/unit/connection_test.rb

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

data/test/unit/core_test.rb

@@ -0,0 +1,55 @@
+require 'test_helper'
+
+class Superstore::CoreTest < Superstore::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 Superstore::IdentityTest < Superstore::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

data/test/unit/inspect_test.rb

@@ -0,0 +1,26 @@
+require 'test_helper'
+
+class Superstore::InspectTest < Superstore::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,26 @@
+require "test_helper"
+require "active_support/log_subscriber/test_helper"
+require "superstore/log_subscriber"
+
+class Superstore::LogSubscriberTest < Superstore::TestCase
+  include ActiveSupport::LogSubscriber::TestHelper
+
+  def setup
+    super
+
+    Superstore::LogSubscriber.attach_to :cassandra_object
+  end
+
+  def test_cql_notification
+    Issue.adapter.execute "SELECT * FROM Issues"
+
+    wait
+
+    assert_equal 1, @logger.logged(:debug).size
+    assert_match "SELECT * FROM Issues", @logger.logged(:debug)[0]
+  end
+
+  def test_initializes_runtime
+    Thread.new { assert_equal 0, Superstore::LogSubscriber.runtime }.join
+  end
+end

data/test/unit/persistence_test.rb

@@ -0,0 +1,213 @@
+#!/bin/env ruby
+# encoding: utf-8
+
+require 'test_helper'
+
+class Superstore::PersistenceTest < Superstore::TestCase
+  test 'instantiate removes unknowns' do
+    assert_nil Issue.instantiate('theid', 'z' => 'nooo').attributes['z']
+  end
+
+  test 'encode_attributes' do
+    klass = temp_object do
+      string :description
+    end
+
+    assert_equal(
+      {},
+      klass.encode_attributes({})
+    )
+
+    assert_equal(
+      {'description' => nil},
+      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(Superstore::RecordNotFound) { Issue.find(first_issue.id) }
+      assert_raise(Superstore::RecordNotFound) { Issue.find(second_issue.id) }
+    end
+
+    assert !Issue.batching?
+    assert_nothing_raised(Superstore::RecordNotFound) { Issue.find(first_issue.id) }
+    assert_nothing_raised(Superstore::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 Superstore::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 Superstore::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 'quote_columns' do
+    klass = Class.new { include Superstore::Persistence }
+    assert_equal %w{'a' 'b'}, klass.__send__(:quote_columns, %w{a b})
+  end
+
+  test 'remove' do
+    klass = temp_object do
+      string :name
+    end
+
+    record = klass.new(name: 'cool')
+    record.save!
+
+    id = record.id
+    assert_equal id, klass.find(id).id
+
+    klass.remove(id)
+
+    assert_raise Superstore::RecordNotFound do
+      klass.find(id)
+    end
+  end
+
+  test 'remove multiple' do
+    klass = temp_object do
+      string :name
+    end
+
+    ids = []
+    (1..10).each do
+      record = klass.create!(name: 'cool')
+      ids << record.id
+    end
+
+    klass.remove(ids)
+
+    assert_equal [], klass.find(ids)
+  end
+
+end

data/test/unit/railties/controller_runtime_test.rb

@@ -0,0 +1,48 @@
+require 'test_helper'
+require "superstore/railties/controller_runtime"
+
+class Superstore::Railties::ControllerRuntimeTest < MiniTest::Unit::TestCase
+  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 Superstore::Railties::ControllerRuntime
+  end
+
+  def test_cleanup_view_runtime
+    runtime = CassandraRuntime.new
+    Superstore::LogSubscriber.runtime = 10
+
+    runtime.cleanup_view_runtime
+
+    assert_equal 0, Superstore::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 "Superstore: 12.3ms", messages.last
+  end
+end