active-asari 0.0.2

data/.gitignore

@@ -0,0 +1,20 @@
+*.gem
+*.rbc
+.bundle
+.config
+coverage
+InstalledFiles
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+
+# YARD artifacts
+.yardoc
+_yardoc
+doc/
+*.swp
+.DS_Store

data/Gemfile

@@ -0,0 +1,15 @@
+source 'https://rubygems.org'
+
+ruby '1.9.3'
+
+gem 'asari', :git => 'git://github.com/lgleasain/asari.git'
+gem 'aws-sdk'
+
+group :test, :development do
+  gem 'rspec'
+  gem 'factory_girl'
+  gem 'jeweler'
+  gem 'simplecov'
+  gem 'activerecord', '~>3.2.13'
+  gem 'sqlite3'
+end

data/Gemfile.lock

@@ -0,0 +1,103 @@
+GIT
+  remote: git://github.com/lgleasain/asari.git
+  revision: f17d8baf3707a36982566db5fc2f2fd792c2f023
+  specs:
+    asari (0.8.0)
+      httparty
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    activemodel (3.2.13)
+      activesupport (= 3.2.13)
+      builder (~> 3.0.0)
+    activerecord (3.2.13)
+      activemodel (= 3.2.13)
+      activesupport (= 3.2.13)
+      arel (~> 3.0.2)
+      tzinfo (~> 0.3.29)
+    activesupport (3.2.13)
+      i18n (= 0.6.1)
+      multi_json (~> 1.0)
+    addressable (2.3.5)
+    arel (3.0.2)
+    aws-sdk (1.12.0)
+      json (~> 1.4)
+      nokogiri (< 1.6.0)
+      uuidtools (~> 2.1)
+    builder (3.0.4)
+    diff-lcs (1.2.4)
+    factory_girl (4.2.0)
+      activesupport (>= 3.0.0)
+    faraday (0.8.7)
+      multipart-post (~> 1.1)
+    git (1.2.5)
+    github_api (0.10.1)
+      addressable
+      faraday (~> 0.8.1)
+      hashie (>= 1.2)
+      multi_json (~> 1.4)
+      nokogiri (~> 1.5.2)
+      oauth2
+    hashie (2.0.5)
+    highline (1.6.19)
+    httparty (0.11.0)
+      multi_json (~> 1.0)
+      multi_xml (>= 0.5.2)
+    httpauth (0.2.0)
+    i18n (0.6.1)
+    jeweler (1.8.6)
+      builder
+      bundler (~> 1.0)
+      git (>= 1.2.5)
+      github_api (= 0.10.1)
+      highline (>= 1.6.15)
+      nokogiri (= 1.5.10)
+      rake
+      rdoc
+    json (1.8.0)
+    jwt (0.1.8)
+      multi_json (>= 1.5)
+    multi_json (1.7.7)
+    multi_xml (0.5.4)
+    multipart-post (1.2.0)
+    nokogiri (1.5.10)
+    oauth2 (0.9.2)
+      faraday (~> 0.8)
+      httpauth (~> 0.2)
+      jwt (~> 0.1.4)
+      multi_json (~> 1.0)
+      multi_xml (~> 0.5)
+      rack (~> 1.2)
+    rack (1.5.2)
+    rake (10.1.0)
+    rdoc (4.0.1)
+      json (~> 1.4)
+    rspec (2.14.1)
+      rspec-core (~> 2.14.0)
+      rspec-expectations (~> 2.14.0)
+      rspec-mocks (~> 2.14.0)
+    rspec-core (2.14.3)
+    rspec-expectations (2.14.0)
+      diff-lcs (>= 1.1.3, < 2.0)
+    rspec-mocks (2.14.1)
+    simplecov (0.7.1)
+      multi_json (~> 1.0)
+      simplecov-html (~> 0.7.1)
+    simplecov-html (0.7.1)
+    sqlite3 (1.3.7)
+    tzinfo (0.3.37)
+    uuidtools (2.1.4)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  activerecord (~> 3.2.13)
+  asari!
+  aws-sdk
+  factory_girl
+  jeweler
+  rspec
+  simplecov
+  sqlite3

data/LICENSE

@@ -0,0 +1,20 @@
+The MIT License (MIT)
+
+Copyright (c) 2013 lgleasain
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of
+this software and associated documentation files (the "Software"), to deal in
+the Software without restriction, including without limitation the rights to
+use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+the Software, and to permit persons to whom the Software is furnished to do so,
+subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,102 @@
+# ActiveAsari
+
+## Description
+
+Full Active Record Integration with Asari along with migration-like indexing and domain creation.  This makes using Amazon Cloud Search a lot easier in your rails application.
+
+#### Why ActiveAsari?
+
+Active Asari builds functionality on Asar which gives you a powerfull way to create and access Amazon Cloud Search instances and data.
+
+## Usage
+
+#### Configuration
+
+Include the gem in your project.
+
+  gem 'active_asari'
+
+In your application.rb file or in the main part of your rack app.  NOTE:  The ActiveAsari.configure method needs a directory to be passed to it.  It will look for the configuration files and load them from this location.
+
+    AWS.config({:access_key_id => 'your Amazon key id', :secret_access_key => 'your Amazon secret key'})
+    require 'active_asari'
+    ACTIVE_ASARI_CONFIG, ACTIVE_ASARI_ENV = ActiveAsari.configure(File.dirname(__FILE__))
+    require 'active_asari/active_record'
+  
+There are two configuration files that active_asari looks for.  First,  active_asari_env.yml.  Right now this file is used to configure access permissions for your domains and the prefix to use for them.  Prefixing was added so that you can have domains for multiple environments under one amazon account without having them stomp on each other.  Here is an example of a active_asari_env.yml file.
+
+    development:
+      domain_prefix: dev
+      access_permissions:
+      - ip_address: 0.0.0.0/0  
+      - ip_address: 25.44.23.25/32
+    staging:
+      domain_prefix: staging
+      access_permissions:
+      - ip_address: 192.168.66.23/32  
+      - ip_address: 28.44.23.25/32
+
+When you are in a test environment active_asari overlays are disabled.  Right now the recommended way of testing asari calls is via mocking.
+
+#### Your Domain
+
+Domains are collections of data that you can store.  ActiveAsari has taken the approach of associating a activerecord table with a domain.  So if you have a talble called events then you might create a domain of the same name.  Domain objects are specified in the active_asari_config.yml file.  Below is an example of a file.
+
+    TestModel:
+      name: 
+        index_field_type: text
+        search_enabled: true
+      amount: 
+        index_field_type: uint
+        search_enabled: true
+      last_updated:
+        index_field_type: uint
+        search_enabled: false
+      bee_larvae_type:
+        index_field_type: literal
+    HoneyBadger:
+      name:
+        index_field_type: text 
+
+The domain is the top most element in the structure.  The example above specifies two domains.  The first one has indexes for name,  amount,  last_updated and bee_larvae_type.  All active_asari fields are result enabled to allow you to retrive the result values from Amazon Cloud search.  The index_field_type specifies the Amazon Cloud Search unit type.  search_enabled specifies your ability to search on a field.  text fields are always searchable,  uint are configurable but default to searchable and literals need to be enabled if you want to be able to search them.  
+
+The ActiveAsari::Migrations class has methods to 'migrate' your cloud search instance to have domains with indexes specified in your config file.  We have a todo to add rake tasks to the gem.  In the meantime you can all migrate_all to create everything in your configuration file or migrate_domain to migrate one domain.  Both of these commands also apply to access policies you specified in your env file.  When the domain is created,  the prefix you specified is added to the domain and it is uncased similar to rails but instead of using underscores it uses -.  For example HoneyBadger in the development environment with a prefix of dev would create a domain called dev-honey-badger.
+
+#### Your Models
+
+Any model can be associated with a ActiveAsari Object.  This is done by adding the following three lines to the beginning of your model class.
+
+    include ActiveAsari::ActiveRecord
+    include Asari::ActiveRecord if !env_test?
+    active_asari_index 'YourDomain'
+
+You would substitute the appropriate domain specified in your config file.  The !env_test option was the only way to not have ActiveAsari not interfere with tests.  We would welcome pull requests to make this cleaner.
+
+Once you have this in your model,  and have run your migrations all update,  create and deletes to the model in a non_test environment will automatically be performed in AmazonCloud search keeping the two environments in sync.
+
+#### Searching 
+
+Searching is done via ActiveAsari.active_asari_search.  IE:
+
+    ActiveAsari.active_asari_search 'HoneyBadger', 'name:beavis',  :query_type => :boolean
+
+The query_type allows you to specify if you want to do a boolean or regular query.  All other options are passed directly to asari,  so see the asari gem for documentation on how to use it.  Results are automatically returned in a hash of objects indexed by the document_id.  The object contains a raw_result accessor along with accessors for all returned fields in the hash.  So...
+
+    my_result = ActiveAsari.active_asari_search 'HoneyBadger', 'beavis'
+    my_result['6'].name.should include 'beavis'
+
+Search parameters are passed directly to Amazon Cloud Search.  See it's documentation for otpions,  syntax etc..
+
+
+## Contributions 
+
+If ActiveAsari interests you,  please fork the repo and submit a pull request.  
+
+### Contributors 
+
+* [Lance Gleason](https://github.com/lgleasain "lgleasain on Github")
+
+## License 
+
+MIT,  see LICENSE for more details.
+

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/active-asari.gemspec

@@ -0,0 +1,28 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+
+Gem::Specification.new do |s|
+  s.name        = "active-asari"
+  s.version     = '0.0.2'
+  s.authors     = ["Lance Gleason"]
+  s.email       = ["lgleason@polyglotprogramminginc.com"]
+  s.homepage    = "http://github.com/playon/active-asari"
+  s.summary     = %q{ActiveAsari is a Ruby interface for AWS CloudSearch.}
+  s.description = %q{ActiveAsari is a Ruby interface for AWS CloudSearch.  It uses Asari as a foundation and make it work more like Active Record.}
+
+  s.rubyforge_project = "asari"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  # specify any dependencies here; for example:
+  # s.add_development_dependency "rspec"
+  # s.add_runtime_dependency "rest-client"
+
+  s.add_runtime_dependency "asari"
+  s.add_runtime_dependency 'aws-sdk'
+
+  s.add_development_dependency "rspec"
+end

data/circle.yml

@@ -0,0 +1,3 @@
+database:
+  override:
+    - echo "skipping database setup"    

data/lib/active_asari.rb

@@ -0,0 +1,7 @@
+require 'asari'
+require 'asari/active_record'
+require 'active_asari/migrations'
+require 'active_asari/active_record'
+require 'active_asari/common'
+require 'active_asari/result_object'
+

data/lib/active_asari/active_record.rb

@@ -0,0 +1,21 @@
+module ActiveAsari
+  module ActiveRecord
+
+    alias_attribute :active_asari_id, :id
+
+    def self.included(base)
+      base.extend(ClassMethods)
+    end
+
+    module ClassMethods
+      def env_test?
+        (ENV['RAILS_ENV'] == 'test' or ENV['RACK_ENV'] == 'test')
+      end
+
+      def active_asari_index(class_name)
+        active_asari_index_array = ACTIVE_ASARI_CONFIG[class_name].symbolize_keys.keys.concat [:active_asari_id] 
+        asari_index ActiveAsari.asari_domain_name(class_name),  active_asari_index_array if !env_test?
+      end
+    end
+  end
+end

data/lib/active_asari/common.rb

@@ -0,0 +1,50 @@
+module ActiveAsari
+
+  def self.amazon_safe_domain_name(domain)
+    environment = ENV['RAILS_ENV'] ? ENV['RAILS_ENV'] : ENV['RACK_ENV'] 
+    "#{ACTIVE_ASARI_ENV[environment]['domain_prefix']}-#{domain.underscore.sub(/_/, '-')}" 
+  end
+
+  def self.aws_client
+    AWS::CloudSearch::Client.new
+  end
+
+  def self.asari_domain_name(domain)
+    $domains = {} if !$domains
+    amazon_domain = amazon_safe_domain_name domain
+    if !$domains[amazon_domain]
+      domain_data = aws_client.describe_domains[:domain_status_list].select { |domain_data|  
+        domain_data[:domain_name] == amazon_domain}
+      $domains[amazon_domain] = domain_data.first[:search_service][:endpoint].split('.').first[7..-1]
+    end
+    $domains[amazon_domain] 
+  end
+
+  def self.active_asari_raw_search(domain, query, search_options = {})
+    asari = Asari.new asari_domain_name(domain)
+    fields = ACTIVE_ASARI_CONFIG[domain].map {|field| field.first.to_sym}
+    fields = fields.concat([:active_asari_id])
+    search_options[:return_fields] = fields
+    asari.search query, search_options
+  end
+
+  def self.objectify_results(hash_results)
+    results = {}
+    hash_results.each do |key, value|
+      results[key] = ResultObject.new
+      results[key].raw_result = value
+    end
+    results
+  end
+
+  def self.active_asari_search(domain, query, search_options = {})
+    raw_result = active_asari_raw_search domain, query, search_options
+    objectify_results raw_result
+  end
+
+  def self.configure(yaml_file_dir)
+    active_asari_config = YAML.load_file(File.expand_path(yaml_file_dir) + '/active_asari_config.yml')
+    active_asari_env = YAML.load_file(File.expand_path(yaml_file_dir) + '/active_asari_env.yml') 
+    return active_asari_config, active_asari_env
+  end
+end 

data/lib/active_asari/migrations.rb

@@ -0,0 +1,61 @@
+module ActiveAsari
+  class Migrations
+
+    attr_accessor :connection
+
+    def initialize 
+      self.connection = AWS::CloudSearch::Client.new
+    end
+
+    def migrate_all
+      ACTIVE_ASARI_CONFIG.keys.each do |domain|
+        migrate_domain domain
+      end
+    end
+
+    def migrate_domain(domain)
+      create_domain domain
+      ACTIVE_ASARI_CONFIG[domain].each do |field|
+        create_index_field domain, field.first => field.last
+      end
+      create_index_field domain, 'active_asari_id' => {'index_field_type' => 'uint'}
+      connection.index_documents :domain_name => ActiveAsari.amazon_safe_domain_name(domain)
+      update_service_access_policies ActiveAsari.amazon_safe_domain_name(domain)
+    end
+
+    def update_service_access_policies(domain)
+      policy_array = []
+      asari_env = ENV['RAILS_ENV'] ? ENV['RAILS_ENV'] : ENV['RACK_ENV'] 
+      ACTIVE_ASARI_ENV[asari_env]['access_permissions'].each do |permission|
+        policy_array << {:Effect => :Allow, :Action => '*', :Resource => '*', :Condition => {:IpAddress => {'aws:SourceIp' => [permission['ip_address']]}}} 
+      end
+      access_policies = {:Statement => policy_array}
+      connection.update_service_access_policies :domain_name => domain, :access_policies => access_policies.to_json
+    end
+
+    def create_index_field(domain, field)
+      index_field_name = field.keys.first
+      index_field_type = field[index_field_name]['index_field_type']
+      search_enabled = field[index_field_name]['search_enabled']
+      if search_enabled == nil or search_enabled.blank?
+        search_enabled = false
+      end
+      
+
+      request = {:domain_name => ActiveAsari.amazon_safe_domain_name(domain), :index_field => {:index_field_name => index_field_name,
+      :index_field_type => index_field_type}}
+      case index_field_type 
+      when 'literal'
+        request[:index_field][:literal_options] = {:search_enabled => search_enabled, :result_enabled => true}
+      when 'text'
+        request[:index_field][:text_options] = {:result_enabled => true}
+      end
+      connection.define_index_field request
+    end
+
+    def create_domain(domain)
+      connection.create_domain :domain_name => ActiveAsari.amazon_safe_domain_name(domain)        
+    end
+
+  end
+end

data/lib/active_asari/result_object.rb

@@ -0,0 +1,21 @@
+class ActiveAsari::ResultObject
+  attr_accessor :raw_result
+
+  def method_missing(method_sym, *arguments, &block) 
+    base_method = method_sym.to_s.end_with?('_array') ? method_sym.to_s.chomp('_array') : method_sym.to_s
+    if raw_result.has_key? base_method
+      method_sym.to_s.end_with?('_array') ? raw_result[base_method] : raw_result[base_method].first
+    else
+      super
+    end    
+  end 
+
+  def respond_to?(method_sym, include_private = false)
+    base_method = method_sym.to_s.end_with?('_array') ? method_sym.to_s.chomp('_array') : method_sym.to_s
+    if raw_result.has_key? base_method
+      true
+    else
+      super
+    end    
+  end
+end

data/spec/active_asari_config.yml

@@ -0,0 +1,15 @@
+TestModel:
+  name: 
+    index_field_type: text
+    search_enabled: true
+  amount: 
+    index_field_type: uint
+    search_enabled: true
+  last_updated:
+    index_field_type: uint
+    search_enabled: false
+  bee_larvae_type:
+    index_field_type: literal
+HoneyBadger:
+  name:
+    index_field_type: text

data/spec/active_asari_env.yml

@@ -0,0 +1,14 @@
+test:
+  domain_prefix: test
+  access_key: my_great_key
+  secrite_access_key: not_a_big_secret
+  access_permissions:
+  - ip_address: 192.168.66.23/32  
+  - ip_address: 23.44.23.25/32
+development:
+  domain_prefix: test
+  access_key: my_great_key
+  secrite_access_key: not_a_big_secret
+  access_permissions:
+  - ip_address: 192.168.66.23/32  
+  - ip_address: 23.44.23.25/32

data/spec/database.yml

@@ -0,0 +1,5 @@
+test:
+  adapter: sqlite3
+  database: db/test.sqlite3
+  pool: 5
+

data/spec/lib/active_asari_spec.rb

@@ -0,0 +1,123 @@
+require 'lib/active_asari_spec_data'
+include ActiveAsariSpecData
+require 'spec_helper'
+
+describe 'active_record' do
+
+  context 'asari index' do 
+    it 'should call asari index with the correct parameters' do
+      TestModel.should_receive(:asari_index).with('test-test-model-7yopqryvjnumbe547ha7xhmjwi', [:name, :amount, :last_updated, :bee_larvae_type, :active_asari_id])
+      ENV['RAILS_ENV'] = 'development'
+      ENV['RACK_ENV'] = 'development' 
+      ActiveAsari.should_receive(:asari_domain_name).and_return 'test-test-model-7yopqryvjnumbe547ha7xhmjwi'
+      TestModel.send(:active_asari_index, 'TestModel')
+    end 
+  end
+
+  context 'models' do
+    before :each do
+      asari_instance = double 'asari instance'
+      asari_instance.should_receive(:update_item).with(1, {:name => 'test', :amount => 2, :last_updated => nil, :bee_larvae_type => nil})
+      asari_instance.should_receive(:add_item).with(1, {:name => 'test', :amount => 2, :last_updated => '', :bee_larvae_type => ''})
+      TestModel.class_variable_set(:@@asari_when, nil)
+      TestModel.class_variable_set(:@@asari_fields, [:name, :amount, :last_updated, :bee_larvae_type])
+      TestModel.class_variable_set(:@@asari_instance, asari_instance)
+      CreateTestModel.up 
+      @model = TestModel.create :name => 'test', :amount => 2
+      @model.save
+    end
+
+    it 'should add new records to cloud search and alias object id and active_asari_id' do
+      @model.id.should eq @model.active_asari_id
+    end
+
+    after :each do 
+      CreateTestModel.down
+    end
+  end
+
+  context 'asari_domain_name' do
+    it 'should get the correct domain name and only call Amazon once' do
+      aws_client = double 'AWS Client'
+      aws_client.should_receive(:describe_domains).once.and_return DESCRIBE_DOMAINS_RESPONSE 
+      ActiveAsari.should_receive(:aws_client).once.and_return aws_client
+      ActiveAsari.asari_domain_name('lance-event').should eq 'test-lance-event-7yopqryvjnumbe547ha7xhmjwi'
+      ActiveAsari.asari_domain_name('lance-event').should eq 'test-lance-event-7yopqryvjnumbe547ha7xhmjwi'
+    end
+  end
+
+  context 'active_asari_raw_search' do
+    let(:asari) {double 'Asari'}
+
+    before :each do 
+      ActiveAsari.should_receive(:asari_domain_name).with('TestModel').and_return('test-model-666')
+    end
+
+    it 'should search for all available fields for a item' do
+      asari.should_receive(:search).with('foo', :return_fields => [:name, :amount, :last_updated, :bee_larvae_type, :active_asari_id]).and_return(
+        {'33' => {'name' => ['beavis'], 'amount' => ['22'], 'last_updated' => ['4543457887875']}}) 
+        Asari.should_receive(:new).with('test-model-666').and_return asari
+        ActiveAsari.active_asari_raw_search 'TestModel', 'foo'
+    end
+
+    it 'should search for all available fields for a item with a binary search' do
+      asari.should_receive(:search).with('foo', :return_fields => [:name, :amount, :last_updated, :bee_larvae_type, :active_asari_id], :query_type => :boolean).and_return(
+        {'33' => {'name' => ['beavis'], 'amount' => ['22'], 'last_updated' => ['4543457887875']}}) 
+        Asari.should_receive(:new).with('test-model-666').and_return asari
+        ActiveAsari.active_asari_raw_search 'TestModel', 'foo', :query_type => :boolean
+    end
+  end
+
+  context 'active_asari_search' do
+    let(:raw_result) {{'33' => {'name' => ['beavis'], 'amount' => ['22'], 'last_updated' => ['4543457887875']}}}
+    it 'should call out to do a raw search and then objectify the results' do
+      ActiveAsari.should_receive(:active_asari_raw_search).with('TestModel', 'foo', {}).and_return(raw_result) 
+      ActiveAsari.should_receive(:objectify_results).with(raw_result).and_return({'33' => 'stuff'})
+      ActiveAsari.active_asari_search 'TestModel', 'foo'
+    end
+     
+    it 'should pass on the parameters for a boolean search' do 
+      ActiveAsari.should_receive(:active_asari_raw_search).with('TestModel', 'foo', :query_type => :boolean).and_return(raw_result) 
+      ActiveAsari.should_receive(:objectify_results).with(raw_result).and_return({'33' => 'stuff'})
+      ActiveAsari.active_asari_search 'TestModel', 'foo', :query_type => :boolean
+    end
+  end
+
+  context 'objectify results' do
+    let(:hash_results) {{'33' => {'name' => ['beavis'], 'amount' => ['22', '33'], 'last_updated' => ['4543457887875']},
+      '34' => {'name' => ['butthead'], 'amount' => ['666'], 'last_updated' => ['454333457887875']}}}
+    let(:objectified_results) {ActiveAsari.objectify_results(hash_results)}
+
+    it 'should create an object with a reference to the raw results' do
+      objectified_results['33'].raw_result.should eq hash_results['33']
+      objectified_results['34'].raw_result.should eq hash_results['34']
+    end
+
+    context 'method_missing' do
+      it 'should create methods on the fly for hash items in an object' do
+        objectified_results['33'].name.should eq 'beavis'
+        objectified_results['34'].name.should eq 'butthead'
+      end
+
+      it 'should return a array of items is an array accessor is used' do
+        objectified_results['33'].amount_array.should eq ['22', '33']
+        objectified_results['34'].amount_array.should eq ['666']
+      end
+
+      it 'should raise an error if a invalid parameter is specified' do
+        expect {objectified_results['33'].foo}.to raise_error(NoMethodError)
+      end
+    end
+
+    context 'respond_to?' do
+      it 'should return true if the method exists' do
+        objectified_results['33'].respond_to?(:amount_array).should eq true
+        objectified_results['33'].respond_to?(:name).should eq true
+      end
+
+      it 'should return true if the method exists' do
+        objectified_results['33'].respond_to?(:cornholio).should eq false
+      end
+    end
+  end 
+end

data/spec/lib/active_asari_spec_data.rb

@@ -0,0 +1,6 @@
+module ActiveAsariSpecData
+  DESCRIBE_DOMAINS_TEST_MODEL_RESPONSE = {:domain_status_list=>[{:search_partition_count=>1, :search_service=>{:arn=>"arn:aws:cs:us-east-1:658167492042:search/test-model", :endpoint=>"search-test-model-7yopqryvjnumbe547ha7xhmjwi.us-east-1.cloudsearch.amazonaws.com"}, :num_searchable_docs=>0, :search_instance_type=>"search.m1.small", :created=>true, :domain_id=>"658167492042/test-model", :processing=>false, :search_instance_count=>1, :domain_name=>"test-model", :requires_index_documents=>false, :deleted=>false, :doc_service=>{:arn=>"arn:aws:cs:us-east-1:658167492042:doc/test-model", :endpoint=>"doc-test-model-7yopqryvjnumbe547ha7xhmjwi.us-east-1.cloudsearch.amazonaws.com"}}, {:search_partition_count=>1, :search_service=>{:arn=>"arn:aws:cs:us-east-1:658167492042:search/lance", :endpoint=>"search-lance-lagja54pf5qhpzayza4awcw7wu.us-east-1.cloudsearch.amazonaws.com"}, :num_searchable_docs=>28005, :search_instance_type=>"search.m1.small", :created=>true, :domain_id=>"658167492042/lance", :processing=>false, :search_instance_count=>1, :domain_name=>"lance", :requires_index_documents=>true, :deleted=>false, :doc_service=>{:arn=>"arn:aws:cs:us-east-1:658167492042:doc/lance", :endpoint=>"doc-lance-lagja54pf5qhpzayza4awcw7wu.us-east-1.cloudsearch.amazonaws.com"}}, {:search_partition_count=>1, :search_service=>{:arn=>"arn:aws:cs:us-east-1:658167492042:search/beavis", :endpoint=>"search-beavis-bg3kl446fx2bm5s4in2rltsoui.us-east-1.cloudsearch.amazonaws.com"}, :num_searchable_docs=>0, :search_instance_type=>"search.m1.small", :created=>true, :domain_id=>"658167492042/beavis", :processing=>false, :search_instance_count=>1, :domain_name=>"beavis", :requires_index_documents=>false, :deleted=>false, :doc_service=>{:arn=>"arn:aws:cs:us-east-1:658167492042:doc/beavis", :endpoint=>"doc-beavis-bg3kl446fx2bm5s4in2rltsoui.us-east-1.cloudsearch.amazonaws.com"}}], :response_metadata=>{:request_id=>"2b782398-fb82-11e2-a681-9b8b1f7eca45"}}
+
+  DESCRIBE_DOMAINS_RESPONSE = {:domain_status_list=>[{:search_partition_count=>1, :search_service=>{:arn=>"arn:aws:cs:us-east-1:658167492042:search/test-lance-event", :endpoint=>"search-test-lance-event-7yopqryvjnumbe547ha7xhmjwi.us-east-1.cloudsearch.amazonaws.com"}, :num_searchable_docs=>0, :search_instance_type=>"search.m1.small", :created=>true, :domain_id=>"658167492042/test-lance-event", :processing=>false, :search_instance_count=>1, :domain_name=>"test-lance-event", :requires_index_documents=>false, :deleted=>false, :doc_service=>{:arn=>"arn:aws:cs:us-east-1:658167492042:doc/test-lance-event", :endpoint=>"doc-test-lance-event-7yopqryvjnumbe547ha7xhmjwi.us-east-1.cloudsearch.amazonaws.com"}}, {:search_partition_count=>1, :search_service=>{:arn=>"arn:aws:cs:us-east-1:658167492042:search/lance", :endpoint=>"search-test-lance-lagja54pf5qhpzayza4awcw7wu.us-east-1.cloudsearch.amazonaws.com"}, :num_searchable_docs=>28005, :search_instance_type=>"search.m1.small", :created=>true, :domain_id=>"658167492042/test-lance", :processing=>false, :search_instance_count=>1, :domain_name=>"test-lance", :requires_index_documents=>true, :deleted=>false, :doc_service=>{:arn=>"arn:aws:cs:us-east-1:658167492042:doc/lance", :endpoint=>"doc-test-lance-lagja54pf5qhpzayza4awcw7wu.us-east-1.cloudsearch.amazonaws.com"}}, {:search_partition_count=>1, :search_service=>{:arn=>"arn:aws:cs:us-east-1:658167492042:search/beavis", :endpoint=>"search-beavis-bg3kl446fx2bm5s4in2rltsoui.us-east-1.cloudsearch.amazonaws.com"}, :num_searchable_docs=>0, :search_instance_type=>"search.m1.small", :created=>true, :domain_id=>"658167492042/beavis", :processing=>false, :search_instance_count=>1, :domain_name=>"test-beavis", :requires_index_documents=>false, :deleted=>false, :doc_service=>{:arn=>"arn:aws:cs:us-east-1:658167492042:doc/beavis", :endpoint=>"doc-beavis-bg3kl446fx2bm5s4in2rltsoui.us-east-1.cloudsearch.amazonaws.com"}}], :response_metadata=>{:request_id=>"2b782398-fb82-11e2-a681-9b8b1f7eca45"}}
+
+end

data/spec/lib/migrations_spec.rb

@@ -0,0 +1,107 @@
+require 'spec_helper'
+require 'lib/migrations_spec_data'
+include MigrationsSpecData
+
+describe 'migrations' do
+
+  let(:migrations) {ActiveAsari::Migrations.new}
+
+  before :each do
+  end
+
+  context 'index_field' do
+    context 'literal fields' do
+      before :each do
+        expected_index_field_request = {:domain_name => 'test-beavis-butthead', :index_field => 
+          {:index_field_name => 'band_name', :index_field_type => 'literal', :literal_options =>
+            {:search_enabled => false, :result_enabled => true}}}
+            migrations.connection.should_receive(:define_index_field).with(expected_index_field_request).and_return CREATE_LITERAL_INDEX_RESPONSE     
+      end
+
+      it 'should add a index to the domain' do
+        migrations.create_index_field('BeavisButthead', 'band_name' => { 'index_field_type' => 'literal', 'search_enabled' => false})
+      end
+
+      it 'should default search enabled to false if it is not specified' do
+        migrations.create_index_field('BeavisButthead', 'band_name' => { 'index_field_type' => 'literal'})
+      end
+
+      it 'should default search enabled to false if it is not specified as a blank string' do
+        migrations.create_index_field('BeavisButthead', 'band_name' => { 'index_field_type' => 'literal', 'search_enabled' => ''})
+      end
+
+    end
+
+    it 'should set search enabled to true if it is a string that evaluates to true' do
+      expected_index_field_request = {:domain_name => 'test-beavis-butthead', :index_field => 
+        {:index_field_name => 'band_name', :index_field_type => 'literal', :literal_options =>
+          {:search_enabled => true, :result_enabled => true}}}
+          migrations.connection.should_receive(:define_index_field).with(expected_index_field_request).and_return CREATE_LITERAL_INDEX_RESPONSE     
+          migrations.create_index_field('BeavisButthead', 'band_name' => { 'index_field_type' => 'literal', 'search_enabled' => true})
+    end
+
+    it 'should add a text index to the domain' do
+      expected_index_field_request = {:domain_name => 'test-beavis', :index_field => 
+        {:index_field_name => 'tv_location', :index_field_type => 'text', :text_options =>
+          {:result_enabled => true}}}
+          migrations.connection.should_receive(:define_index_field).with(expected_index_field_request).and_return CREATE_TEXT_INDEX_RESPONSE 
+          migrations.create_index_field('beavis', 'tv_location' => { 'index_field_type' => 'text'})
+    end
+
+    it 'should add a uint index to the domain' do
+      expected_index_field_request = {:domain_name => 'test-beavis', :index_field => 
+        {:index_field_name => 'num_tvs', :index_field_type => 'uint'}}
+        migrations.connection.should_receive(:define_index_field).with(expected_index_field_request).and_return CREATE_UINT_INDEX_RESPONSE 
+        migrations.create_index_field('beavis', 'num_tvs' => { 'index_field_type' => 'uint'})
+    end
+  end
+
+  context 'update_service_access_policies' do
+    it 'should allow all access for ip addresses specified in the configuration file' do
+      access_policies = "{\"Statement\":[{\"Effect\":\"Allow\",\"Action\":\"*\",\"Resource\":\"*\",\"Condition\":{\"IpAddress\":{\"aws:SourceIp\":[\"192.168.66.23/32\"]}}},{\"Effect\":\"Allow\",\"Action\":\"*\",\"Resource\":\"*\",\"Condition\":{\"IpAddress\":{\"aws:SourceIp\":[\"23.44.23.25/32\"]}}}]}"
+      migrations.connection.should_receive(:update_service_access_policies).with(:domain_name => 'beavis',
+                                                                                 :access_policies => access_policies)
+      ENV['RAILS_ENV'] = nil
+      ENV['RACK_ENV'] = 'test'
+      migrations.update_service_access_policies 'beavis'
+    end
+  end
+
+  context 'domain' do
+
+    it 'should create a domain if one doesnt exist' do
+      migrations.connection.should_receive(:create_domain).with({:domain_name => 'test-beavis-butthead'}).and_return CREATE_DOMAIN_RESPONSE    
+      migrations.create_domain 'BeavisButthead'
+    end
+
+    it 'should create indexes for all items in the domain and create the domain' do
+      migrations.should_receive(:create_domain).once.with 'TestModel'
+      migrations.should_receive(:create_index_field).once.with('TestModel', 
+                                                               'name' => { 'index_field_type' => 'text', 
+                                                                 'search_enabled' => true})
+      migrations.should_receive(:create_index_field).once.with('TestModel', 
+                                                               'amount' => { 'index_field_type' => 'uint', 
+                                                                 'search_enabled' => true})
+      migrations.should_receive(:create_index_field).once.with('TestModel', 
+                                                               'last_updated' => { 'index_field_type' => 'uint', 
+                                                                 'search_enabled' => false})
+      migrations.should_receive(:create_index_field).once.with('TestModel', 
+                                                               'bee_larvae_type' => { 'index_field_type' => 'literal'})
+      migrations.should_receive(:create_index_field).once.with('TestModel', 
+                                                               'active_asari_id' => { 'index_field_type' => 'uint'})
+      ActiveAsari.should_receive(:amazon_safe_domain_name).twice.with('TestModel').and_return 'test-model-666'
+      migrations.should_receive(:update_service_access_policies).once.with('test-model-666')
+      migrations.connection.should_receive(:index_documents).with(:domain_name => 'test-model-666')
+      migrations.migrate_domain 'TestModel'
+    end
+  end
+
+  context 'migrate_all' do
+    it 'should migrate all of the domains' do
+      migrations.should_receive(:migrate_domain).once.with 'TestModel'
+      migrations.should_receive(:migrate_domain).once.with 'HoneyBadger'
+      migrations.migrate_all
+    end
+  end
+
+end

data/spec/lib/migrations_spec_data.rb

@@ -0,0 +1,9 @@
+module MigrationsSpecData
+
+  CREATE_DOMAIN_RESPONSE = {:domain_status=>{:search_partition_count=>0, :search_service=>{:arn=>"arn:aws:cs:us-east-1:888167492042:search/beavis"}, :num_searchable_docs=>0, :created=>true, :domain_id=>"888167492042/beavis", :processing=>false, :search_instance_count=>0, :domain_name=>"beavis", :requires_index_documents=>false, :deleted=>false, :doc_service=>{:arn=>"arn:aws:cs:us-east-1:888167492042:doc/beavis"}}, :response_metadata=>{:request_id=>"88e3adcb-f999-11e2-ba8b-ab9a7c0903a8"}}
+  CREATE_LITERAL_INDEX_RESPONSE = {:index_field=>{:status=>{:creation_date=>'2013-07-30 20:47:55 UTC', :pending_deletion=>"false", :update_version=>20, :state=>"RequiresIndexDocuments", :update_date=>'2013-07-30 20:47:55 UTC'}, :options=>{:source_attributes=>[], :literal_options=>{:search_enabled=>false}, :index_field_type=>"literal", :index_field_name=>"test"}}, :response_metadata=>{:request_id=>"8885505e-f959-11e2-b89b-2d5c6f978750"}}
+  CREATE_TEXT_INDEX_RESPONSE = {:index_field=>{:status=>{:creation_date=>'2013-07-30 20:47:55 UTC', :pending_deletion=>"false", :update_version=>20, :state=>"RequiresIndexDocuments", :update_date=>'2013-07-30 20:47:55 UTC'}, :options=>{:source_attributes=>[], :text_options=>{:result_enabled=>true}, :index_field_type=>"text", :index_field_name=>"test"}}, :response_metadata=>{:request_id=>"8885505e-f959-11e2-b89b-2d5c6f978750"}}
+  CREATE_UINT_INDEX_RESPONSE = {:index_field=>{:status=>{:creation_date=>'2013-07-30 20:47:55 UTC', :pending_deletion=>"false", :update_version=>20, :state=>"RequiresIndexDocuments", :update_date=>'2013-07-30 20:47:55 UTC'}, :options=>{:source_attributes=>[], :index_field_type=>"uint", :index_field_name=>"num_tvs"}}, :response_metadata=>{:request_id=>"8885505e-f959-11e2-b89b-2d5c6f978750"}}
+end
+
+

data/spec/model/create_test_model.rb

@@ -0,0 +1,14 @@
+class CreateTestModel < ActiveRecord::Migration
+  def self.up
+    create_table :test_models do |t|
+      t.string :name
+      t.integer :amount
+      t.datetime :last_updated
+      t.string :bee_larvae_type
+    end
+  end
+
+  def self.down
+    drop_table :test_models
+  end
+end

data/spec/model/test_model.rb

@@ -0,0 +1,15 @@
+
+class AWS::CloudSearch::Client
+  def initialize
+  end
+
+  def describe_domains
+    ActiveAsariSpecData::DESCRIBE_DOMAINS_TEST_MODEL_RESPONSE
+  end
+end
+
+class TestModel < ActiveRecord::Base
+  include Asari::ActiveRecord
+  include ActiveAsari::ActiveRecord
+  active_asari_index 'TestModel'
+end

data/spec/spec_helper.rb

@@ -0,0 +1,22 @@
+ENV['RACK_ENV'] = 'test'
+require 'simplecov'
+SimpleCov.start
+require 'active_record'
+require 'rspec'
+require 'factory_girl'
+require 'rspec/autorun'
+db_config = YAML.load_file(File.expand_path(File.dirname(__FILE__)) + '/database.yml')
+ActiveRecord::Base.establish_connection db_config['test']
+
+ActiveRecord::Base.connection
+
+ACTIVE_ASARI_SEARCH_DOMAIN = 'my_great_domain'
+Dir[File.dirname(__FILE__) + "/../lib/active_asari/*.rb"].each {|file| require file }
+require 'active_asari'
+ACTIVE_ASARI_CONFIG, ACTIVE_ASARI_ENV = ActiveAsari.configure(File.dirname(__FILE__))
+require 'active_asari/active_record'
+require 'asari'
+require 'aws'
+require 'model/test_model'
+require 'model/create_test_model'
+

metadata

@@ -0,0 +1,128 @@
+--- !ruby/object:Gem::Specification
+name: active-asari
+version: !ruby/object:Gem::Version
+  version: 0.0.2
+  prerelease: 
+platform: ruby
+authors:
+- Lance Gleason
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-10-09 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: asari
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: ActiveAsari is a Ruby interface for AWS CloudSearch.  It uses Asari as
+  a foundation and make it work more like Active Record.
+email:
+- lgleason@polyglotprogramminginc.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- Gemfile.lock
+- LICENSE
+- README.md
+- Rakefile
+- active-asari.gemspec
+- circle.yml
+- db/test.sqlite3
+- lib/active_asari.rb
+- lib/active_asari/active_record.rb
+- lib/active_asari/common.rb
+- lib/active_asari/migrations.rb
+- lib/active_asari/result_object.rb
+- spec/active_asari_config.yml
+- spec/active_asari_env.yml
+- spec/database.yml
+- spec/lib/active_asari_spec.rb
+- spec/lib/active_asari_spec_data.rb
+- spec/lib/migrations_spec.rb
+- spec/lib/migrations_spec_data.rb
+- spec/model/create_test_model.rb
+- spec/model/test_model.rb
+- spec/spec_helper.rb
+homepage: http://github.com/playon/active-asari
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: asari
+rubygems_version: 1.8.25
+signing_key: 
+specification_version: 3
+summary: ActiveAsari is a Ruby interface for AWS CloudSearch.
+test_files:
+- spec/active_asari_config.yml
+- spec/active_asari_env.yml
+- spec/database.yml
+- spec/lib/active_asari_spec.rb
+- spec/lib/active_asari_spec_data.rb
+- spec/lib/migrations_spec.rb
+- spec/lib/migrations_spec_data.rb
+- spec/model/create_test_model.rb
+- spec/model/test_model.rb
+- spec/spec_helper.rb