cequel 0.0.0 → 0.4.0

data/spec/examples/model/dirty_spec.rb

@@ -0,0 +1,39 @@
+require File.expand_path('../spec_helper', __FILE__)
+
+describe Cequel::Model::Dirty do
+  let(:post) { Post.new(:id => 1) }
+
+  it 'should not have changed attributes by default' do
+    post.changed_attributes.should be_empty
+  end
+
+  it 'should have changed attributes if attributes change' do
+    post.title = 'Cequel'
+    post.changed_attributes.should == {:title => nil}.with_indifferent_access
+  end
+
+  it 'should not have changed attributes if attribute set to the same thing' do
+    post.title = nil
+    post.changed_attributes.should be_empty
+  end
+
+  it 'should support *_changed? method' do
+    post.title = 'Cequel'
+    post.title_changed?.should be_true
+  end
+
+  it 'should not have changed attributes after save' do
+    connection.stub(:execute)
+    post.title = 'Cequel'
+    post.save
+    post.changed_attributes.should be_empty
+  end
+
+  it 'should have previous changes after save' do
+    connection.stub(:execute)
+    post.title = 'Cequel'
+    post.save
+    post.previous_changes.
+      should == { :title => [nil, 'Cequel'] }.with_indifferent_access
+  end
+end

data/spec/examples/model/dynamic_spec.rb

@@ -0,0 +1,41 @@
+require File.expand_path('../spec_helper', __FILE__)
+
+describe Cequel::Model::Dynamic do
+
+  let(:category) { Category.new(:id => 1, :name => 'Big Data') }
+
+  it 'should allow getting and setting of dynamic attributes' do
+    category[:tag] = 'bigdata'
+    category[:tag].should == 'bigdata'
+  end
+
+  it 'should insert dynamic values' do
+    category[:tag] = 'bigdata'
+    connection.should_receive(:execute).
+      with "INSERT INTO categories (?) VALUES (?)", ['id', 'name', 'tag'], [1, 'Big Data', 'bigdata']
+    category.save
+  end
+
+  it 'should update dynamic values' do
+    category[:tag] = 'bigdata'
+    connection.stub(:execute).
+      with "INSERT INTO categories (?) VALUES (?)", ['id', 'name', 'tag'], [1, 'Big Data', 'bigdata']
+    category.save
+    category[:tag] = 'big-data'
+    category[:color] = 'blue'
+    connection.should_receive(:execute).
+      with "UPDATE categories SET ? = ?, ? = ? WHERE ? = ?", 'tag', 'big-data', 'color', 'blue', :id, 1
+    category.save
+  end
+
+  it 'should delete dynamic values' do
+    category[:tag] = 'bigdata'
+    connection.stub(:execute).
+      with "INSERT INTO categories (?) VALUES (?)", ['id', 'name', 'tag'], [1, 'Big Data', 'bigdata']
+    category.save
+    category[:tag] = nil
+    connection.should_receive(:execute).
+      with "DELETE ? FROM categories WHERE ? = ?", ['tag'], :id, 1
+    category.save
+  end
+end

data/spec/examples/model/inheritable_spec.rb

@@ -0,0 +1,45 @@
+require File.expand_path('../spec_helper', __FILE__)
+
+describe Cequel::Model::Inheritable do
+  it 'should inherit key from superclass' do
+    Photo.key_alias.should == :id
+  end
+
+  it 'should inherit columns from superclass' do
+    Photo.column_names.should == [:id, :class_name, :label, :checksum, :url]
+  end
+
+  it 'should return ::base_class for base class' do
+    Asset.base_class.should == Asset
+  end
+
+  it 'should return ::base_class for subclass' do
+    Photo.base_class.should == Asset
+  end
+
+  it 'should not allow overriding a class without a type column' do
+    base = Class.new do
+      include Cequel::Model
+      key :id, :integer
+    end
+    expect { sub = Class.new(base) }.to raise_error(ArgumentError)
+  end
+
+  it 'should set type on initialization' do
+    Photo.new.class_name.should == 'Photo'
+  end
+
+  it 'should query correct column family when querying subclass' do
+    connection.stub(:execute).
+      with("SELECT * FROM assets WHERE ? = ? LIMIT 1", :id, 1).
+      and_return result_stub(:id => 1, :label => 'Cequel')
+    Photo.find(1)
+  end
+
+  it 'should hydrate correct model class when querying base class' do
+    connection.stub(:execute).
+      with('SELECT * FROM assets WHERE ? = ? LIMIT 1', :id, 1).
+      and_return result_stub(:id => 1, :class_name => 'Photo', :label => 'Cequel')
+    Asset.find(1).should be_a(Photo)
+  end
+end

data/spec/examples/model/magic_spec.rb

@@ -0,0 +1,199 @@
+require File.expand_path('../spec_helper', __FILE__)
+
+describe Cequel::Model::Magic do
+  describe '::find_by_*' do
+    it 'should magically look up one record by given value' do
+      connection.stub(:execute).
+        with("SELECT * FROM posts WHERE ? = ? LIMIT 1", :title, 'Cequel').
+        and_return result_stub(:id => 1, :title => 'Cequel')
+
+      Post.find_by_title('Cequel').id.should == 1
+    end
+
+    it 'should magically look up one record by multiple values' do
+      connection.stub(:execute).
+        with("SELECT * FROM posts WHERE ? = ? AND ? = ? LIMIT 1", :title, 'Cequel', :published, true).
+        and_return result_stub(:id => 1, :title => 'Cequel', :published => true)
+
+      Post.find_by_title_and_published('Cequel', true).id.should == 1
+    end
+
+    it 'should magically work on scopes' do
+      connection.stub(:execute).
+        with("SELECT ? FROM posts WHERE ? = ? AND ? = ? LIMIT 1", [:id], :title, 'Cequel', :published, true).
+        and_return result_stub(:id => 1)
+
+      Post.select(:id).find_by_title_and_published('Cequel', true).id.should == 1
+    end
+
+    it 'should raise error if specified columns different from arg count' do
+      expect { Post.find_by_title_and_published('Cequel') }.
+        to raise_error(ArgumentError)
+    end
+  end
+
+  describe '::find_all_by_*' do
+    context 'with existing record specified as args' do
+      it 'should magically look up one record by multiple values' do
+        connection.stub(:execute).
+          with("SELECT * FROM posts WHERE ? = ? AND ? = ?", :title, 'Cequel', :published, true).
+          and_return result_stub(
+            {:id => 1, :title => 'Cequel', :published => true},
+            {:id => 2, :title => 'Cequel2', :published => true }
+          )
+
+        Post.find_all_by_title_and_published('Cequel', true).map(&:id).
+          should == [1, 2]
+      end
+
+      it 'should magically work on scopes' do
+        connection.stub(:execute).
+          with("SELECT ? FROM posts WHERE ? = ? AND ? = ?", [:id], :title, 'Cequel', :published, true).
+          and_return result_stub(
+            {:id => 1},
+            {:id => 2}
+          )
+
+        Post.select(:id).find_all_by_title_and_published('Cequel', true).map(&:id).
+          should == [1, 2]
+      end
+    end
+  end
+
+  describe '::find_or_create_by_*' do
+    it 'should return existing record from args' do
+      connection.stub(:execute).
+        with("SELECT * FROM posts WHERE ? = ? AND ? = ? LIMIT 1", :title, 'Cequel', :published, true).
+        and_return result_stub(:id => 1, :title => 'Cequel', :published => true)
+
+      Post.find_or_create_by_title_and_published('Cequel', true).id.should == 1
+    end
+
+    it 'should create new record from args' do
+      now = Time.now
+      Time.stub!(:now).and_return now
+      connection.stub(:execute).
+        with("SELECT * FROM blogs WHERE ? = ? LIMIT 1", :id, 1).
+        and_return result_stub
+      connection.should_receive(:execute).
+        with(
+          "INSERT INTO blogs (?) VALUES (?)",
+          ['id', 'published', 'updated_at', 'created_at'],
+          [1, true, now, now]
+        )
+
+      Blog.find_or_create_by_id(1).id.should == 1
+    end
+
+    it 'should look up record from attributes' do
+      connection.stub(:execute).
+        with("SELECT * FROM posts WHERE ? = ? AND ? = ? LIMIT 1", :title, 'Cequel', :published, true).
+        and_return result_stub(:id => 1, :title => 'Cequel', :published => true)
+
+      Post.find_or_create_by_title_and_published(
+        :id => 2, :title => 'Cequel', :published => true
+      ).id.should == 1
+    end
+
+    it 'should create record from all attributes specified, including non-lookup ones' do
+      connection.stub(:execute).
+        with("SELECT * FROM posts WHERE ? = ? AND ? = ? LIMIT 1", :title, 'Cequel', :published, true).
+        and_return result_stub
+      connection.should_receive(:execute).
+        with(
+          "INSERT INTO posts (?) VALUES (?)",
+          ['id', 'title', 'published'],
+          [2, 'Cequel', true])
+
+      Post.find_or_create_by_title_and_published(
+        :id => 2, :title => 'Cequel', :published => true
+      ).id.should == 2
+    end
+
+    it 'should yield instance on create if block given' do
+      connection.stub(:execute).
+        with("SELECT * FROM posts WHERE ? = ? AND ? = ? LIMIT 1", :title, 'Cequel', :published, true).
+        and_return result_stub
+      connection.should_receive(:execute).
+        with(
+          "INSERT INTO posts (?) VALUES (?)",
+          ['id', 'title', 'published'],
+          [2, 'Cequel', true]
+        )
+
+      Post.find_or_create_by_title_and_published('Cequel', true) do |post|
+        post.id = 2
+      end.id.should == 2
+    end
+
+    it 'should work on scopes' do
+      connection.stub(:execute).
+        with("SELECT ? FROM posts WHERE ? = ? AND ? = ? LIMIT 1", [:id], :title, 'Cequel', :published, true).
+        and_return result_stub(:id => 1)
+
+      Post.select(:id).find_or_create_by_title_and_published('Cequel', true).id.
+        should == 1
+    end
+  end
+
+  describe '::find_or_initialize_by_*' do
+    it 'should return existing record from args' do
+      connection.stub(:execute).
+        with("SELECT * FROM posts WHERE ? = ? AND ? = ? LIMIT 1", :title, 'Cequel', :published, true).
+        and_return result_stub(:id => 1, :title => 'Cequel', :published => true)
+
+      Post.find_or_initialize_by_title_and_published('Cequel', true).id.should == 1
+    end
+
+    it 'should initialize new record from args' do
+      now = Time.now
+      Time.stub!(:now).and_return now
+      timestamp = (now.to_f * 1000).to_i
+      connection.stub(:execute).
+        with("SELECT * FROM blogs WHERE ? = ? LIMIT 1", :id, 1).
+        and_return result_stub
+
+      Blog.find_or_initialize_by_id(1).id.should == 1
+    end
+
+    it 'should look up record from attributes' do
+      connection.stub(:execute).
+        with("SELECT * FROM posts WHERE ? = ? AND ? = ? LIMIT 1", :title, 'Cequel', :published, true).
+        and_return result_stub(:id => 1, :title => 'Cequel', :published => true)
+
+      Post.find_or_initialize_by_title_and_published(
+        :id => 2, :title => 'Cequel', :published => true
+      ).id.should == 1
+    end
+
+    it 'should create record from all attributes specified, including non-lookup ones' do
+      connection.stub(:execute).
+        with("SELECT * FROM posts WHERE ? = ? AND ? = ? LIMIT 1", :title, 'Cequel', :published, true).
+        and_return result_stub
+
+      Post.find_or_initialize_by_title_and_published(
+        :id => 2, :title => 'Cequel', :published => true
+      ).id.should == 2
+    end
+
+    it 'should yield instance on initialize if block given' do
+      connection.stub(:execute).
+        with("SELECT * FROM posts WHERE ? = ? AND ? = ? LIMIT 1", :title, 'Cequel', :published, true).
+        and_return result_stub
+
+      Post.find_or_initialize_by_title_and_published('Cequel', true) do |post|
+        post.id = 2
+      end.id.should == 2
+    end
+
+    it 'should work on scopes' do
+      connection.stub(:execute).
+        with("SELECT ? FROM posts WHERE ? = ? AND ? = ? LIMIT 1", [:id], :title, 'Cequel', :published, true).
+        and_return result_stub(:id => 1)
+
+      Post.select(:id).find_or_initialize_by_title_and_published('Cequel', true).id.
+        should == 1
+    end
+
+  end
+end

data/spec/examples/model/mass_assignment_security_spec.rb

@@ -0,0 +1,13 @@
+require File.expand_path('../spec_helper', __FILE__)
+
+describe Cequel::Model::MassAssignmentSecurity do
+  let(:post) { Post.new(:id => 1, :title => 'Cequel', :blog_id => 3) }
+
+  it 'should allow setting of unprotected attributes' do
+    post.title.should == 'Cequel'
+  end
+
+  it 'should not allow setting of protected attributes' do
+    post.blog_id.should be_nil
+  end
+end

data/spec/examples/model/naming_spec.rb

@@ -0,0 +1,9 @@
+require File.expand_path('../spec_helper', __FILE__)
+
+describe Cequel::Model::Naming do
+
+  it 'should give correct model_name' do
+    Post.model_name.should == 'Post'
+  end
+
+end

data/spec/examples/model/observer_spec.rb

@@ -0,0 +1,86 @@
+require File.expand_path('../spec_helper', __FILE__)
+
+describe Cequel::Model::Observer do
+  before do
+    Cequel::Model.observers = [:post_observer, :asset_observer]
+    Cequel::Model.instantiate_observers
+  end
+
+  let(:post) do
+    connection.stub(:execute).
+      with("SELECT * FROM posts WHERE ? = ? LIMIT 1", :id, 1).
+      and_return result_stub('id' => 1, 'title' => 'Hey')
+    Post.find(1)
+  end
+
+  shared_examples_for 'observing callbacks' do |*callbacks|
+
+    all_callbacks = [
+      :before_create, :after_create, :before_update, :after_update,
+      :before_save, :after_save, :before_destroy, :after_destroy,
+      :before_validation, :after_validation
+    ]
+
+    callbacks.each do |callback|
+      it "should observe #{callback}" do
+        post.should have_been_observed(callback)
+      end
+    end
+
+    (all_callbacks - callbacks).each do |callback|
+      it "should not observe #{callback}" do
+        post.should_not have_been_observed(callback)
+      end
+    end
+  end
+
+  context 'on create' do
+    let(:post) do
+      connection.stub(:execute).
+        with "INSERT INTO posts (?) VALUES (?)", ['id', 'title'], [1, 'Hey']
+      Post.new(:id => 1, :title => 'Hey')
+    end
+
+    before { post.save }
+
+    it_should_behave_like 'observing callbacks',
+      :before_create, :after_create, :before_save, :after_save, :before_validation, :after_validation
+  end
+
+  context 'on update' do
+
+    before { post.save }
+
+    it_should_behave_like 'observing callbacks',
+      :before_update, :after_update, :before_save, :after_save, :before_validation, :after_validation
+  end
+
+  context 'on destroy 'do
+
+    before do
+      connection.stub(:execute).
+        with "DELETE FROM posts WHERE ? = ?", :id, 1
+
+      post.destroy
+    end
+
+    it_should_behave_like 'observing callbacks', :before_destroy, :after_destroy
+  end
+
+  context 'on validation' do
+    before do
+      post.valid?
+    end
+
+    it_should_behave_like 'observing callbacks', :before_validation, :after_validation
+  end
+
+  context 'with inheritence' do
+    it 'should observe subclass' do
+      connection.stub(:execute).
+        with("INSERT INTO assets (?) VALUES (?)", ['id', 'label', 'class_name'], [1, 'Cequel', 'Photo'])
+      photo = Photo.create!(:id => 1, :label => 'Cequel')
+      photo.should have_observed(:before_save)
+    end
+  end
+end

data/spec/examples/model/persistence_spec.rb

@@ -0,0 +1,201 @@
+require File.expand_path('../spec_helper', __FILE__)
+
+describe Cequel::Model::Persistence do
+  describe '#find' do
+    it 'should return hydrated instance' do
+      connection.stub(:execute).
+        with("SELECT * FROM posts WHERE ? = ? LIMIT 1", :id, 2).
+        and_return result_stub(:id => 2, :title => 'Cequel')
+
+      post = Post.find(2)
+      post.id.should == 2
+      post.title.should == 'Cequel'
+    end
+
+    it 'should not set defaults when hydrating instance' do
+      connection.stub(:execute).
+        with("SELECT ? FROM blogs WHERE ? = ? LIMIT 1", [:id, :name], :id, 2).
+        and_return result_stub(:id => 1, :title => 'Big Data')
+
+      blog = Blog.select(:id, :name).find(2)
+      blog.published.should be_nil
+    end
+
+    it 'should return multiple instances' do
+      connection.stub(:execute).
+        with("SELECT * FROM posts WHERE ? IN (?)", :id, [2, 5]).
+        and_return result_stub(
+          {:id => 2, :title => 'Cequel 2'},
+          {:id => 5, :title => 'Cequel 5'}
+        )
+
+      posts = Post.find(2, 5)
+      posts.map { |post| [post.id, post.title] }.
+        should == [[2, 'Cequel 2'], [5, 'Cequel 5']]
+    end
+
+    it 'should return one-element array if passed one-element array' do
+      connection.stub(:execute).
+        with("SELECT * FROM posts WHERE ? = ? LIMIT 1", :id, 2).
+        and_return result_stub(:id => 2, :title => 'Cequel')
+
+      post = Post.find([2]).first
+      post.id.should == 2
+      post.title.should == 'Cequel'
+    end
+
+    it 'should raise RecordNotFound if row has no data' do
+      connection.stub(:execute).
+        with("SELECT * FROM posts WHERE ? = ? LIMIT 1", :id, 2).
+        and_return result_stub(:id => 2)
+
+      expect { Post.find(2) }.to raise_error Cequel::Model::RecordNotFound
+    end
+
+    it 'should raise RecordNotFound if row has nil data' do
+      connection.stub(:execute).
+        with("SELECT ? FROM posts WHERE ? = ? LIMIT 1", [:title], :id, 2).
+        and_return result_stub(:title => nil)
+
+      expect { Post.select(:title).find(2) }.to raise_error Cequel::Model::RecordNotFound
+    end
+
+    it 'should raise RecordNotFound if some rows in multi-row query have no data' do
+      connection.stub(:execute).
+        with("SELECT * FROM posts WHERE ? IN (?)", :id, [2, 5]).
+        and_return result_stub(
+          {:id => 2, :title => 'Cequel 2'},
+          {:id => 5}
+        )
+
+      expect { Post.find(2, 5) }.to raise_error(Cequel::Model::RecordNotFound)
+    end
+  end
+
+  describe '#reload' do
+    let(:post) do
+      connection.stub(:execute).
+        with("SELECT * FROM posts WHERE ? = ? LIMIT 1", :id, 2).
+        and_return result_stub(:id => 2, :title => 'Cequel')
+      Post.find(2)
+    end
+
+    it 'should reload attributes from Cassandra' do
+      post.title = 'Donkeys'
+      connection.should_receive(:execute).
+        with("SELECT * FROM posts WHERE ? = ? LIMIT 1", :id, 2).
+        and_return result_stub(:id => 2, :title => 'Cequel')
+      post.reload
+      post.title.should == 'Cequel'
+    end
+  end
+
+  describe '#save' do
+    describe 'with new record' do
+      let(:post) { Post.new(:id => 1) }
+
+      it 'should persist only columns with values' do
+        connection.should_receive(:execute).
+          with("INSERT INTO posts (?) VALUES (?)", ['id', 'title'], [1, 'Cequel'])
+
+        post.title = 'Cequel'
+        post.save
+      end
+
+      it 'should mark instance as persisted' do
+        connection.stub(:execute).
+          with("INSERT INTO posts (?) VALUES (?)", ['id', 'title'], [1, 'Cequel'])
+
+        post.title = 'Cequel'
+        post.save
+        post.should be_persisted
+      end
+
+      it 'should not send anything to Cassandra if no column values are set' do
+        post.save
+        post.should_not be_persisted
+      end
+
+      it 'should raise MissingKey if no key set' do
+        expect { Post.new.save }.to raise_error(Cequel::Model::MissingKey)
+      end
+    end
+
+    describe 'with persisted record' do
+      let(:post) do
+        connection.stub(:execute).with("SELECT * FROM posts WHERE ? = ? LIMIT 1", :id, 1).
+          and_return result_stub(:id => 1, :blog_id => 1, :title => 'Cequel')
+        Post.find(1)
+      end
+
+      it 'should send UPDATE statement with changed columns using underlying attributes' do
+        connection.should_receive(:execute).
+          with "UPDATE posts SET ? = ?, ? = ? WHERE ? = ?", 'body', 'Cequel cequel', 'author_name', 'nworB taM', :id, 1
+        post.body = 'Cequel cequel'
+        post.author_name = 'Mat Brown'
+        post.save
+      end
+
+      it 'should send DELETE statement with removed columns' do
+        connection.should_receive(:execute).
+          with "DELETE ? FROM posts WHERE ? = ?", ['title'], :id, 1
+        post.title = nil
+        post.save
+      end
+
+      it 'should mark record as transient if all attributes removed' do
+        connection.stub(:execute).
+          with "DELETE ? FROM posts WHERE ? = ?", ['title', 'blog_id'], :id, 1
+        post.title = nil
+        post.blog_id = nil
+        post.save
+        post.should_not be_persisted
+      end
+    end
+  end
+
+  describe '#update_attributes' do
+    let(:post) do
+      connection.stub(:execute).with("SELECT * FROM posts WHERE ? = ? LIMIT 1", :id, 1).
+        and_return result_stub(:id => 1, :blog_id => 1, :title => 'Cequel')
+      Post.find(1)
+    end
+
+    it 'should change attributes and save them' do
+      connection.should_receive(:execute).
+        with "UPDATE posts SET ? = ? WHERE ? = ?", 'body', 'Cequel cequel', :id, 1
+      post.update_attributes(:body => 'Cequel cequel')
+    end
+  end
+
+  describe '#destroy' do
+    let(:post) do
+      connection.stub(:execute).with("SELECT * FROM posts WHERE ? = ? LIMIT 1", :id, 1).
+        and_return result_stub(:id => 1, :blog_id => 1, :title => 'Cequel')
+      Post.find(1)
+    end
+
+    it 'should delete all columns from column family' do
+      connection.should_receive(:execute).
+        with "DELETE FROM posts WHERE ? = ?", :id, 1
+
+      post.destroy
+    end
+  end
+
+  describe '::create' do
+    it 'should persist only columns with values' do
+      connection.should_receive(:execute).
+        with("INSERT INTO posts (?) VALUES (?)", ['id', 'title'], [1, 'Cequel'])
+
+      Post.create(:id => 1, :title => 'Cequel')
+    end
+
+    it 'should return post instance and mark it as persisted' do
+      connection.stub(:execute).
+        with("INSERT INTO posts (?) VALUES (?)", ['id', 'title'], [1, 'Cequel'])
+
+      Post.create(:id => 1, :title => 'Cequel').should be_persisted
+    end
+  end
+end