adept_dynamoid 0.5.0.6

data/lib/dynamoid/validations.rb

@@ -0,0 +1,36 @@
+# encoding: utf-8
+module Dynamoid
+  
+  # Provide ActiveModel validations to Dynamoid documents.
+  module Validations
+    extend ActiveSupport::Concern
+
+    include ActiveModel::Validations
+    include ActiveModel::Validations::Callbacks
+
+    # Override save to provide validation support.
+    #
+    # @since 0.2.0
+    def save(options = {})
+      options.reverse_merge!(:validate => true)
+      return false if options[:validate] and (not valid?)
+      super
+    end
+
+    # Is this object valid?
+    #
+    # @since 0.2.0
+    def valid?(context = nil)
+      context ||= (new_record? ? :create : :update)
+      super(context)
+    end
+
+    # Raise an error unless this object is valid.
+    #
+    # @since 0.2.0
+    def save!
+      raise Dynamoid::Errors::DocumentNotValid.new(self) unless valid?
+      save(:validate => false)
+    end
+  end
+end

data/spec/app/models/address.rb

@@ -0,0 +1,10 @@
+class Address
+  include Dynamoid::Document
+  
+  field :city
+  field :options, :serialized
+
+  def zip_code=(zip_code)
+    self.city = "Chicago"
+  end
+end

data/spec/app/models/camel_case.rb

@@ -0,0 +1,24 @@
+class CamelCase
+  include Dynamoid::Document
+
+  field :color
+
+  belongs_to :magazine
+  has_many :users
+  has_one :sponsor
+  has_and_belongs_to_many :subscriptions
+  
+  before_create :doing_before_create
+  after_create :doing_after_create
+  
+  private
+  
+  def doing_before_create
+    true
+  end
+  
+  def doing_after_create
+    true
+  end
+
+end

data/spec/app/models/magazine.rb

@@ -0,0 +1,11 @@
+class Magazine
+  include Dynamoid::Document
+  
+  field :title
+  
+  has_many :subscriptions
+  has_many :camel_cases
+  has_one :sponsor
+
+  belongs_to :owner, :class_name => 'User', :inverse_of => :books
+end

data/spec/app/models/message.rb

@@ -0,0 +1,9 @@
+class Message
+  include Dynamoid::Document
+
+  table name: :messages, key: :message_id, read_capacity: 200, write_capacity: 200
+
+  range :time, :datetime
+
+  field :text
+end

data/spec/app/models/sponsor.rb

@@ -0,0 +1,8 @@
+class Sponsor
+  include Dynamoid::Document
+  
+  belongs_to :magazine
+  has_many :subscriptions
+
+  belongs_to :camel_case
+end

data/spec/app/models/subscription.rb

@@ -0,0 +1,12 @@
+class Subscription
+  include Dynamoid::Document
+  
+  field :length, :integer
+  
+  belongs_to :magazine
+  has_and_belongs_to_many :users
+
+  belongs_to :customer, :class_name => 'User', :inverse_of => :monthly
+
+  has_and_belongs_to_many :camel_cases
+end

data/spec/app/models/tweet.rb

@@ -0,0 +1,12 @@
+class Tweet
+  include Dynamoid::Document
+
+  table name: :twitters, key: :tweet_id, read_capacity: 200, write_capacity: 200
+
+  range :group, :string
+
+  field :msg
+  field :count, :integer
+  field :tags, :set
+  field :user_name
+end

data/spec/app/models/user.rb

@@ -0,0 +1,26 @@
+class User
+  include Dynamoid::Document
+  
+  field :name
+  field :email
+  field :password
+  field :last_logged_in_at, :datetime
+  
+  index :name
+  index :email
+  index [:name, :email]
+  index :name, :range_key => :created_at
+  index :name, :range_key => :last_logged_in_at
+  index :created_at, :range => true
+  
+  has_and_belongs_to_many :subscriptions
+
+  has_many :books, :class_name => 'Magazine', :inverse_of => :owner
+  has_one :monthly, :class_name => 'Subscription', :inverse_of => :customer
+
+  has_and_belongs_to_many :followers, :class_name => 'User', :inverse_of => :following
+  has_and_belongs_to_many :following, :class_name => 'User', :inverse_of => :followers
+
+  belongs_to :camel_case
+
+end

data/spec/dynamoid/adapter/aws_sdk_spec.rb

@@ -0,0 +1,186 @@
+require 'dynamoid/adapter/aws_sdk'
+require File.expand_path(File.dirname(__FILE__) + '../../../spec_helper')
+
+describe Dynamoid::Adapter::AwsSdk do
+
+  if ENV['ACCESS_KEY'] && ENV['SECRET_KEY']
+
+    context 'without a preexisting table' do
+      # CreateTable and DeleteTable
+      it 'performs CreateTable and DeleteTable' do
+        table = Dynamoid::Adapter.create_table('CreateTable', :id, :range_key =>  { :created_at => :number })
+
+        Dynamoid::Adapter.connection.tables.collect{|t| t.name}.should include 'CreateTable'
+
+        Dynamoid::Adapter.delete_table('CreateTable')
+      end
+    end
+
+    context 'with a preexisting table' do
+      before(:all) do
+        Dynamoid::Adapter.create_table('dynamoid_tests_TestTable1', :id) unless Dynamoid::Adapter.list_tables.include?('dynamoid_tests_TestTable1')
+        Dynamoid::Adapter.create_table('dynamoid_tests_TestTable2', :id) unless Dynamoid::Adapter.list_tables.include?('dynamoid_tests_TestTable2')
+        Dynamoid::Adapter.create_table('dynamoid_tests_TestTable3', :id, :range_key => { :range => :number }) unless Dynamoid::Adapter.list_tables.include?('dynamoid_tests_TestTable3')
+      end
+
+      # GetItem, PutItem and DeleteItem
+      it "performs GetItem for an item that does not exist" do
+        Dynamoid::Adapter.get_item('dynamoid_tests_TestTable1', '1').should be_nil
+      end
+
+      it "performs GetItem for an item that does exist" do
+        Dynamoid::Adapter.put_item('dynamoid_tests_TestTable1', {:id => '1', :name => 'Josh'})
+
+        Dynamoid::Adapter.get_item('dynamoid_tests_TestTable1', '1').should == {:name => 'Josh', :id => '1'}
+
+        Dynamoid::Adapter.delete_item('dynamoid_tests_TestTable1', '1')
+
+        Dynamoid::Adapter.get_item('dynamoid_tests_TestTable1', '1').should be_nil
+      end
+
+      it 'performs GetItem for an item that does exist with a range key' do
+        Dynamoid::Adapter.put_item('dynamoid_tests_TestTable3', {:id => '1', :name => 'Josh', :range => 2.0})
+
+        Dynamoid::Adapter.get_item('dynamoid_tests_TestTable3', '1', :range_key => 2.0).should == {:name => 'Josh', :id => '1', :range => 2.0}
+
+        Dynamoid::Adapter.delete_item('dynamoid_tests_TestTable3', '1', :range_key => 2.0)
+
+        Dynamoid::Adapter.get_item('dynamoid_tests_TestTable3', '1', :range_key => 2.0).should be_nil
+      end
+
+      it 'performs DeleteItem for an item that does not exist' do
+        Dynamoid::Adapter.delete_item('dynamoid_tests_TestTable1', '1')
+
+        Dynamoid::Adapter.get_item('dynamoid_tests_TestTable1', '1').should be_nil
+      end
+
+      it 'performs PutItem for an item that does not exist' do
+        Dynamoid::Adapter.put_item('dynamoid_tests_TestTable1', {:id => '1', :name => 'Josh'})
+
+        Dynamoid::Adapter.get_item('dynamoid_tests_TestTable1', '1').should == {:id => '1', :name => 'Josh'}
+      end
+
+      # BatchGetItem
+      it "performs BatchGetItem with singular keys" do
+        Dynamoid::Adapter.put_item('dynamoid_tests_TestTable1', {:id => '1', :name => 'Josh'})
+        Dynamoid::Adapter.put_item('dynamoid_tests_TestTable2', {:id => '1', :name => 'Justin'})
+
+        results = Dynamoid::Adapter.batch_get_item('dynamoid_tests_TestTable1' => '1', 'dynamoid_tests_TestTable2' => '1')
+        results.size.should == 2
+        results['dynamoid_tests_TestTable1'].should include({:name => 'Josh', :id => '1'})
+        results['dynamoid_tests_TestTable2'].should include({:name => 'Justin', :id => '1'})
+      end
+
+      it "performs BatchGetItem with multiple keys" do
+        Dynamoid::Adapter.put_item('dynamoid_tests_TestTable1', {:id => '1', :name => 'Josh'})
+        Dynamoid::Adapter.put_item('dynamoid_tests_TestTable1', {:id => '2', :name => 'Justin'})
+
+        results = Dynamoid::Adapter.batch_get_item('dynamoid_tests_TestTable1' => ['1', '2'])
+        results.size.should == 1
+        results['dynamoid_tests_TestTable1'].should include({:name => 'Josh', :id => '1'})
+        results['dynamoid_tests_TestTable1'].should include({:name => 'Justin', :id => '2'})
+      end
+
+      it 'performs BatchGetItem with one ranged key' do
+        Dynamoid::Adapter.put_item('dynamoid_tests_TestTable3', {:id => '1', :name => 'Josh', :range => 1.0})
+        Dynamoid::Adapter.put_item('dynamoid_tests_TestTable3', {:id => '2', :name => 'Justin', :range => 2.0})
+
+        results = Dynamoid::Adapter.batch_get_item('dynamoid_tests_TestTable3' => [['1', 1.0]])
+        results.size.should == 1
+        results['dynamoid_tests_TestTable3'].should include({:name => 'Josh', :id => '1', :range => 1.0})
+      end
+
+      it 'performs BatchGetItem with multiple ranged keys' do
+        Dynamoid::Adapter.put_item('dynamoid_tests_TestTable3', {:id => '1', :name => 'Josh', :range => 1.0})
+        Dynamoid::Adapter.put_item('dynamoid_tests_TestTable3', {:id => '2', :name => 'Justin', :range => 2.0})
+
+        results = Dynamoid::Adapter.batch_get_item('dynamoid_tests_TestTable3' => [['1', 1.0],['2', 2.0]])
+        results.size.should == 1
+        results['dynamoid_tests_TestTable3'].should include({:name => 'Josh', :id => '1', :range => 1.0})
+        results['dynamoid_tests_TestTable3'].should include({:name => 'Justin', :id => '2', :range => 2.0})
+      end
+
+      # ListTables
+      it 'performs ListTables' do
+        Dynamoid::Adapter.list_tables.should include 'dynamoid_tests_TestTable1'
+        Dynamoid::Adapter.list_tables.should include 'dynamoid_tests_TestTable2'
+      end
+
+      # Query
+      it 'performs query on a table and returns items' do
+        Dynamoid::Adapter.put_item('dynamoid_tests_TestTable1', {:id => '1', :name => 'Josh'})
+
+        Dynamoid::Adapter.query('dynamoid_tests_TestTable1', :hash_value => '1').should == { :id=> '1', :name=>"Josh" }
+      end
+
+      it 'performs query on a table and returns items if there are multiple items' do
+        Dynamoid::Adapter.put_item('dynamoid_tests_TestTable1', {:id => '1', :name => 'Josh'})
+        Dynamoid::Adapter.put_item('dynamoid_tests_TestTable1', {:id => '2', :name => 'Justin'})
+
+        Dynamoid::Adapter.query('dynamoid_tests_TestTable1', :hash_value => '1').should == { :id=> '1', :name=>"Josh" }
+      end
+
+      context 'range queries' do
+        before do
+          Dynamoid::Adapter.put_item('dynamoid_tests_TestTable3', {:id => '1', :range => 1.0})
+          Dynamoid::Adapter.put_item('dynamoid_tests_TestTable3', {:id => '1', :range => 3.0})
+        end
+
+        it 'performs query on a table with a range and selects items in a range' do
+          Dynamoid::Adapter.query('dynamoid_tests_TestTable3', :hash_value => '1', :range_value => 0.0..3.0).should =~ [{:id => '1', :range => BigDecimal.new(1)}, {:id => '1', :range => BigDecimal.new(3)}]
+        end
+
+        it 'performs query on a table with a range and selects items greater than' do
+          Dynamoid::Adapter.query('dynamoid_tests_TestTable3', :hash_value => '1', :range_greater_than => 1.0).should =~ [{:id => '1', :range => BigDecimal.new(3)}]
+        end
+
+        it 'performs query on a table with a range and selects items less than' do
+          Dynamoid::Adapter.query('dynamoid_tests_TestTable3', :hash_value => '1', :range_less_than => 2.0).should =~ [{:id => '1', :range => BigDecimal.new(1)}]
+        end
+
+        it 'performs query on a table with a range and selects items gte' do
+          Dynamoid::Adapter.query('dynamoid_tests_TestTable3', :hash_value => '1', :range_gte => 1.0).should =~ [{:id => '1', :range => BigDecimal.new(1)}, {:id => '1', :range => BigDecimal.new(3)}]
+        end
+
+        it 'performs query on a table with a range and selects items lte' do
+          Dynamoid::Adapter.query('dynamoid_tests_TestTable3', :hash_value => '1', :range_lte => 3.0).should =~ [{:id => '1', :range => BigDecimal.new(1)}, {:id => '1', :range => BigDecimal.new(3)}]
+        end
+      end
+
+      # Scan
+      it 'performs scan on a table and returns items' do
+        Dynamoid::Adapter.put_item('dynamoid_tests_TestTable1', {:id => '1', :name => 'Josh'})
+
+        Dynamoid::Adapter.scan('dynamoid_tests_TestTable1', :name => 'Josh').should == [{ :id=> '1', :name=>"Josh" }]
+      end
+
+      it 'performs scan on a table and returns items if there are multiple items but only one match' do
+        Dynamoid::Adapter.put_item('dynamoid_tests_TestTable1', {:id => '1', :name => 'Josh'})
+        Dynamoid::Adapter.put_item('dynamoid_tests_TestTable1', {:id => '2', :name => 'Justin'})
+
+        Dynamoid::Adapter.scan('dynamoid_tests_TestTable1', :name => 'Josh').should == [{ :id=> '1', :name=>"Josh" }]
+      end
+
+      it 'performs scan on a table and returns multiple items if there are multiple matches' do
+        Dynamoid::Adapter.put_item('dynamoid_tests_TestTable1', {:id => '1', :name => 'Josh'})
+        Dynamoid::Adapter.put_item('dynamoid_tests_TestTable1', {:id => '2', :name => 'Josh'})
+
+        Dynamoid::Adapter.scan('dynamoid_tests_TestTable1', :name => 'Josh').should include({:name=>"Josh", :id=>"2"}, {:name=>"Josh", :id=>"1"})
+      end
+
+      it 'performs scan on a table and returns all items if no criteria are specified' do
+        Dynamoid::Adapter.put_item('dynamoid_tests_TestTable1', {:id => '1', :name => 'Josh'})
+        Dynamoid::Adapter.put_item('dynamoid_tests_TestTable1', {:id => '2', :name => 'Josh'})
+
+        Dynamoid::Adapter.scan('dynamoid_tests_TestTable1', {}).should include({:name=>"Josh", :id=>"2"}, {:name=>"Josh", :id=>"1"})
+      end
+    end
+
+    # DescribeTable
+
+    # UpdateItem
+
+    # UpdateTable
+
+  end
+end

data/spec/dynamoid/adapter_spec.rb

@@ -0,0 +1,117 @@
+require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
+
+describe "Dynamoid::Adapter" do
+  
+  before(:all) do
+    Dynamoid::Adapter.create_table('dynamoid_tests_TestTable', :id) unless Dynamoid::Adapter.list_tables.include?('dynamoid_tests_TestTable')
+  end
+  
+  it 'extends itself automatically' do
+    lambda {Dynamoid::Adapter.list_tables}.should_not raise_error
+  end
+  
+  it 'raises nomethod if we try a method that is not on the child' do
+    lambda {Dynamoid::Adapter.foobar}.should raise_error
+  end
+  
+  context 'without partioning' do
+    before(:all) do
+      @previous_value = Dynamoid::Config.partitioning
+      Dynamoid::Config.partitioning = false
+    end
+    
+    after(:all) do
+      Dynamoid::Config.partitioning = @previous_value
+    end
+    
+    it 'writes through the adapter' do
+      Dynamoid::Adapter.expects(:put_item).with('dynamoid_tests_TestTable', {:id => '123'}).returns(true)
+
+      Dynamoid::Adapter.write('dynamoid_tests_TestTable', {:id => '123'})
+    end
+
+    it 'reads through the adapter for one ID' do
+      Dynamoid::Adapter.expects(:get_item).with('dynamoid_tests_TestTable', '123', {}).returns(true)
+
+      Dynamoid::Adapter.read('dynamoid_tests_TestTable', '123')
+    end
+
+    it 'reads through the adapter for many IDs' do
+      Dynamoid::Adapter.expects(:batch_get_item).with({'dynamoid_tests_TestTable' => ['1', '2']}).returns(true)
+
+      Dynamoid::Adapter.read('dynamoid_tests_TestTable', ['1', '2'])
+    end
+    
+    it 'reads through the adapter for one ID and a range key' do
+      Dynamoid::Adapter.expects(:get_item).with('dynamoid_tests_TestTable', '123', :range_key => 2.0).returns(true)
+
+      Dynamoid::Adapter.read('dynamoid_tests_TestTable', '123', :range_key => 2.0)      
+    end
+    
+    it 'reads through the adapter for many IDs and a range key' do
+      Dynamoid::Adapter.expects(:batch_get_item).with({'dynamoid_tests_TestTable' => [['1', 2.0], ['2', 2.0]]}).returns(true)
+
+      Dynamoid::Adapter.read('dynamoid_tests_TestTable', ['1', '2'], :range_key => 2.0)
+    end
+  end
+  
+  context 'with partitioning' do
+    before(:all) do
+      @previous_value = Dynamoid::Config.partitioning
+      Dynamoid::Config.partitioning = true
+    end
+    
+    after(:all) do
+      Dynamoid::Config.partitioning = @previous_value
+    end
+    
+    it 'writes through the adapter' do
+      Random.expects(:rand).with(Dynamoid::Config.partition_size).once.returns(0)
+      Dynamoid::Adapter.write('dynamoid_tests_TestTable', {:id => 'testid'})
+    
+      Dynamoid::Adapter.get_item('dynamoid_tests_TestTable', 'testid.0')[:id].should == 'testid.0'
+      Dynamoid::Adapter.get_item('dynamoid_tests_TestTable', 'testid.0')[:updated_at].should_not be_nil
+    end
+  
+    it 'reads through the adapter for one ID' do
+      Dynamoid::Adapter.expects(:batch_get_item).with('dynamoid_tests_TestTable' => (0...Dynamoid::Config.partition_size).collect{|n| "123.#{n}"}).returns({})
+    
+      Dynamoid::Adapter.read('dynamoid_tests_TestTable', '123')
+    end
+  
+    it 'reads through the adapter for many IDs' do
+      Dynamoid::Adapter.expects(:batch_get_item).with('dynamoid_tests_TestTable' => (0...Dynamoid::Config.partition_size).collect{|n| "1.#{n}"} + (0...Dynamoid::Config.partition_size).collect{|n| "2.#{n}"}).returns({})
+    
+      Dynamoid::Adapter.read('dynamoid_tests_TestTable', ['1', '2'])
+    end
+    
+    it 'reads through the adapter for one ID and a range key' do
+      Dynamoid::Adapter.expects(:batch_get_item).with('dynamoid_tests_TestTable' => (0...Dynamoid::Config.partition_size).collect{|n| ["123.#{n}", 2.0]}).returns({})
+    
+      Dynamoid::Adapter.read('dynamoid_tests_TestTable', '123', :range_key => 2.0)
+    end
+  
+    it 'reads through the adapter for many IDs and a range key' do
+      Dynamoid::Adapter.expects(:batch_get_item).with('dynamoid_tests_TestTable' => (0...Dynamoid::Config.partition_size).collect{|n| ["1.#{n}", 2.0]} + (0...Dynamoid::Config.partition_size).collect{|n| ["2.#{n}", 2.0]}).returns({})
+    
+      Dynamoid::Adapter.read('dynamoid_tests_TestTable', ['1', '2'], :range_key => 2.0)
+    end
+  
+    it 'returns an ID with all partitions' do
+      Dynamoid::Adapter.id_with_partitions('1').should =~ (0...Dynamoid::Config.partition_size).collect{|n| "1.#{n}"}
+    end
+    
+    it 'returns an ID and range key with all partitions' do
+      Dynamoid::Adapter.id_with_partitions([['1', 1.0]]).should =~ (0...Dynamoid::Config.partition_size).collect{|n| ["1.#{n}", 1.0]}
+    end
+  
+    it 'returns a result for one partitioned element' do
+      @time = DateTime.now
+      @array =[{:id => '1.0', :updated_at => @time - 6.hours}, {:id => '1.1', :updated_at => @time - 3.hours}, {:id => '1.2', :updated_at => @time - 1.hour}, {:id => '1.3', :updated_at => @time - 6.hours}, {:id => '2.0', :updated_at => @time}]
+    
+      Dynamoid::Adapter.result_for_partition(@array).should =~ [{:id => '1', :updated_at => @time - 1.hour}, {:id => '2', :updated_at => @time}]
+    end
+    
+  end
+
+end