cequel 0.0.0 → 0.4.0

data/spec/examples/model/serialization_spec.rb

@@ -0,0 +1,20 @@
+require File.expand_path('../spec_helper', __FILE__)
+
+describe 'Serialization' do
+  let(:post) { Post.new(:id => 1, :title => 'Cequel') }
+
+  it 'should serialize to JSON' do
+    post.to_json.should ==
+      {'post' => {'id' => 1, 'title' => 'Cequel'}}.to_json
+  end
+
+  it 'should serialize to XML' do
+    post.to_xml.should == <<XML
+<?xml version="1.0" encoding="UTF-8"?>
+<post>
+  <id type="integer">1</id>
+  <title>Cequel</title>
+</post>
+XML
+  end
+end

data/spec/examples/model/spec_helper.rb

@@ -0,0 +1,12 @@
+require File.expand_path('../../spec_helper', __FILE__)
+require 'cequel/model'
+
+Dir.glob(File.join(File.dirname(__FILE__), '../../models/**/*.rb')).each do |file|
+  require file
+end
+
+RSpec.configure do |config|
+  config.before :each do
+    Cequel::Model.keyspace.connection = connection
+  end
+end

data/spec/examples/model/timestamps_spec.rb

@@ -0,0 +1,52 @@
+require File.expand_path('../spec_helper', __FILE__)
+
+describe Cequel::Model::Timestamps do
+  let!(:now) { Time.now }
+  before { Time.stub(:now).and_return now }
+
+  it 'should add created_at column' do
+    Blog.column_names.should include(:created_at)
+  end
+
+  it 'should add updated_at column' do
+    Blog.column_names.should include(:updated_at)
+  end
+
+  it 'should populate created_at and updated_at column on create' do
+    connection.should_receive(:execute).
+      with "INSERT INTO blogs (?) VALUES (?)",
+        ['id', 'published', 'updated_at', 'created_at'],
+        [1, true, now, now]
+    Blog.create!(:id => 1)
+  end
+
+  it 'should populate updated_at column on update' do
+    connection.stub(:execute).
+      with("SELECT * FROM blogs WHERE ? = ? LIMIT 1", :id, 1).
+      and_return result_stub(
+        :id => 1,
+        :name => 'Blogtime',
+        :created_at => now - 1.day,
+        :updated_at => now - 1.day
+      )
+    blog = Blog.find(1)
+    blog.name = 'Bloggy'
+    connection.should_receive(:execute).
+      with("UPDATE blogs SET ? = ?, ? = ? WHERE ? = ?", 'name', 'Bloggy', 'updated_at', now, :id, 1)
+    blog.save
+  end
+
+  it 'should not update updated_at when calling save with no dirty attributes' do
+    connection.stub(:execute).
+      with("SELECT * FROM blogs WHERE ? = ? LIMIT 1", :id, 1).
+      and_return result_stub(
+        :id => 1,
+        :name => 'Blogtime',
+        :created_at => now - 1.day,
+        :updated_at => now - 1.day
+      )
+    blog = Blog.find(1)
+    blog.save
+    blog.updated_at.should == now - 1.day
+  end
+end

data/spec/examples/model/translation_spec.rb

@@ -0,0 +1,23 @@
+require File.expand_path('../spec_helper', __FILE__)
+
+describe Cequel::Model::Translation do
+  before do
+    I18n.backend.load_translations(
+      File.expand_path('../../../support/en.yml', __FILE__))
+    I18n.locale = 'en'
+  end
+
+  it 'should translate model names' do
+    Post.model_name.human.should == 'Blog post'
+  end
+
+  it 'should translate attribute names' do
+    Post.human_attribute_name(:title).should == 'Post title'
+  end
+
+  it 'should translate error messages' do
+    post = Post.new(:id => 1, :require_title => true)
+    post.valid?
+    post.errors.full_messages.should include("Post title is a required field")
+  end
+end

data/spec/examples/model/validations_spec.rb

@@ -0,0 +1,86 @@
+require File.expand_path('../spec_helper', __FILE__)
+
+describe Cequel::Model::Validations do
+  describe '#valid?' do
+    it 'should be false if model is not valid' do
+      Post.new(:id => 1, :require_title => true).should_not be_valid
+    end
+
+    it 'should be true if model is valid' do
+      Post.new(:id => 1, :require_title => true, :title => 'Cequel').should be_valid
+    end
+  end
+
+  describe '#save' do
+    it 'should return false and not persist model if invalid' do
+      Post.new(:id => 1, :body => 'Cequel', :require_title => true).save.should be_false
+    end
+
+    it 'should return true and persist model if valid' do
+      connection.should_receive(:execute).
+        with "INSERT INTO posts (?) VALUES (?)", ['id', 'title'], [1, 'Cequel']
+
+      Post.new(:id => 1, :title => 'Cequel', :require_title => true).save.should be_true
+    end
+  end
+
+  describe '#save!' do
+    it 'should raise error and not persist model if invalid' do
+      expect do
+        Post.new(:id => 1, :body => 'Cequel', :require_title => true).save!
+      end.to raise_error(Cequel::Model::RecordInvalid)
+    end
+
+    it 'should persist model and return self if valid' do
+      connection.should_receive(:execute).
+        with "INSERT INTO posts (?) VALUES (?)", ['id', 'title'], [1, 'Cequel']
+
+      post = Post.new(:id => 1, :title => 'Cequel', :require_title => true)
+      post.save!.should == post
+    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 if valid' do
+      connection.should_receive(:execute).
+        with "UPDATE posts SET ? = ? WHERE ? = ?", 'body', 'Cequel cequel', :id, 1
+      post.update_attributes!(:body => 'Cequel cequel')
+    end
+
+    it 'should raise error if not valid' do
+      post.require_title = true
+      expect { post.update_attributes!(:title => nil) }.
+        to raise_error(Cequel::Model::RecordInvalid)
+    end
+  end
+
+  describe '::create!' do
+    it 'should raise RecordInvalid and not persist model if invalid' do
+      expect do
+        Post.create!(:id => 1, :body => 'Cequel', :require_title => true)
+      end.to raise_error(Cequel::Model::RecordInvalid)
+    end
+
+    it 'should and return model if valid' do
+      connection.should_receive(:execute).
+        with "INSERT INTO posts (?) VALUES (?)", ['id', 'title'], [1, 'Cequel']
+
+      Post.create!(:id => 1, :title => 'Cequel', :require_title => true).
+        title.should == 'Cequel'
+    end
+  end
+
+  describe 'callbacks' do
+    it 'should call validation callbacks' do
+      post = Post.new(:id => 1)
+      post.valid?
+      post.should have_callback(:validation)
+    end
+  end
+end

data/spec/examples/spec_helper.rb

@@ -0,0 +1,9 @@
+require File.expand_path('../../environment', __FILE__)
+
+Dir.glob(File.expand_path('../../support/**/*.rb', __FILE__)).each do |file|
+  require file
+end
+
+RSpec.configure do |config|
+  config.include(Cequel::SpecSupport::Helpers)
+end

data/spec/models/asset.rb

@@ -0,0 +1,21 @@
+class Asset
+
+  include Cequel::Model
+
+  key :id, :integer
+  column :class_name, :ascii
+  column :label, :varchar
+  column :checksum, :varchar
+
+  index_preference :checksum, :class_name
+
+  def observed!(callback)
+    @observed ||= []
+    @observed << callback
+  end
+
+  def has_observed?(callback)
+    @observed.include?(callback)
+  end
+
+end

data/spec/models/asset_observer.rb

@@ -0,0 +1,5 @@
+class AssetObserver < Cequel::Model::Observer
+  def before_save(asset)
+    asset.observed!(:before_save)
+  end
+end

data/spec/models/blog.rb

@@ -0,0 +1,14 @@
+class Blog
+
+  include Cequel::Model
+  include Timestamps
+
+  key :id, :integer
+  column :name, :varchar
+  column :published, :boolean, :default => true
+
+  has_many :posts, :dependent => :destroy
+
+  default_scope limit(100)
+
+end

data/spec/models/blog_posts.rb

@@ -0,0 +1,6 @@
+class BlogPosts < Cequel::Model::Dictionary
+  key :blog_id, :int
+  maps :uuid => :int
+
+  self.default_batch_size = 2
+end

data/spec/models/category.rb

@@ -0,0 +1,9 @@
+class Category
+
+  include Cequel::Model
+  include Cequel::Model::Dynamic
+
+  key :id, :int
+  column :name, :text
+
+end

data/spec/models/comment.rb

@@ -0,0 +1,12 @@
+class Comment
+  include Cequel::Model
+
+  key :id, :uuid
+  column :body, :text
+
+  private
+
+  def generate_key
+    SimpleUUID::UUID.new
+  end
+end

data/spec/models/photo.rb

@@ -0,0 +1,5 @@
+require File.expand_path('../asset', __FILE__)
+
+class Photo < Asset
+  column :url, :varchar
+end

data/spec/models/post.rb

@@ -0,0 +1,88 @@
+class Post
+
+  include Cequel::Model
+
+  key :id, :int
+  column :title, :varchar
+  column :body, :varchar
+  column :published, :boolean
+  column :author_name, :text
+
+  before_save :record_save_callback
+  before_create :record_create_callback
+  before_update :record_update_callback
+  before_destroy :record_destroy_callback
+  before_validation :record_validation_callback
+
+  belongs_to :blog
+  has_one :thumbnail, :class_name => 'Photo'
+
+  validates :title, :presence => true, :if => :require_title?
+
+  attr_protected :blog_id
+
+  attr_writer :require_title
+
+  def self.for_blog(blog_id)
+    where(:blog_id => blog_id)
+  end
+
+  def has_callback?(callback)
+    callbacks.include?(callback)
+  end
+
+  def has_been_observed?(callback)
+    observed.include?(callback)
+  end
+
+  def observed!(callback)
+    observed << callback
+  end
+
+  def require_title?
+    !!@require_title
+  end
+
+  def author_name=(author_name)
+    write_attribute(:author_name, author_name.try(:reverse))
+  end
+
+  def author_name
+    read_attribute(:author_name).try(:reverse)
+  end
+
+  private
+
+  def record_save_callback
+    record_callback(:save)
+  end
+
+  def record_create_callback
+    record_callback(:create)
+  end
+
+  def record_update_callback
+    record_callback(:update)
+  end
+
+  def record_destroy_callback
+    record_callback(:destroy)
+  end
+
+  def record_validation_callback
+    record_callback(:validation)
+  end
+
+  def record_callback(callback)
+    callbacks << callback
+  end
+
+  def callbacks
+    @callbacks ||= Set[]
+  end
+
+  def observed
+    @observed ||= Set[]
+  end
+
+end

data/spec/models/post_comments.rb

@@ -0,0 +1,14 @@
+class PostComments < Cequel::Model::Dictionary
+  key :post_id, :int
+  maps :int => :text
+
+  private
+
+  def serialize_value(data)
+    data.to_json
+  end
+
+  def deserialize_value(post_id, json)
+    JSON.parse(json)
+  end
+end

data/spec/models/post_observer.rb

@@ -0,0 +1,43 @@
+class PostObserver < Cequel::Model::Observer
+
+  def before_create(post)
+    post.observed!(:before_create)
+  end
+
+  def after_create(post)
+    post.observed!(:after_create)
+  end
+
+  def before_save(post)
+    post.observed!(:before_save)
+  end
+
+  def after_save(post)
+    post.observed!(:after_save)
+  end
+
+  def before_update(post)
+    post.observed!(:before_update)
+  end
+
+  def after_update(post)
+    post.observed!(:after_update)
+  end
+
+  def before_destroy(post)
+    post.observed!(:before_destroy)
+  end
+
+  def after_destroy(post)
+    post.observed!(:after_destroy)
+  end
+
+  def before_validation(post)
+    post.observed!(:before_validation)
+  end
+
+  def after_validation(post)
+    post.observed!(:after_validation)
+  end
+
+end

data/spec/support/helpers.rb

@@ -0,0 +1,26 @@
+module Cequel
+
+  module SpecSupport
+
+    Connection = Object.new
+
+    module Helpers
+
+      def result_stub(*results)
+        ResultStub.new(results)
+      end
+
+      def connection
+        Connection
+      end
+
+      def cequel
+        @cequel ||= Cequel::Keyspace.new({}).tap do |keyspace|
+          keyspace.connection = connection
+        end
+      end
+    end
+
+  end
+
+end