superstore 1.0.0

data/lib/superstore/types/date_type.rb

@@ -0,0 +1,22 @@
+module Superstore
+  module Types
+    class DateType < BaseType
+      FORMAT = '%Y-%m-%d'
+      REGEX = /\A\d{4}-\d{2}-\d{2}\Z/
+
+      def encode(value)
+        raise ArgumentError.new("#{value.inspect} is not a Date") unless value.kind_of?(Date)
+        value.strftime(FORMAT)
+      end
+
+      def decode(str)
+        return nil if str.empty?
+        Date.parse(str)
+      end
+
+      def typecast(value)
+        value.to_date
+      end
+    end
+  end
+end

data/lib/superstore/types/float_type.rb

@@ -0,0 +1,16 @@
+module Superstore
+  module Types
+    class FloatType < BaseType
+      REGEX = /\A[-+]?\d+(\.\d+)?\Z/
+      def encode(float)
+        raise ArgumentError.new("#{float.inspect} is not a Float") unless float.kind_of?(Float)
+        float.to_s
+      end
+
+      def decode(str)
+        return nil if str.empty?
+        str.to_f
+      end
+    end
+  end
+end

data/lib/superstore/types/integer_type.rb

@@ -0,0 +1,20 @@
+module Superstore
+  module Types
+    class IntegerType < BaseType
+      REGEX = /\A[-+]?\d+\Z/
+      def encode(int)
+        raise ArgumentError.new("#{int.inspect} is not an Integer.") unless int.kind_of?(Integer)
+        int.to_s
+      end
+
+      def decode(str)
+        return nil if str.empty?
+        str.to_i
+      end
+
+      def typecast(value)
+        value.to_i
+      end
+    end
+  end
+end

data/lib/superstore/types/json_type.rb

@@ -0,0 +1,13 @@
+module Superstore
+  module Types
+    class JsonType < BaseType
+      def encode(hash)
+        ActiveSupport::JSON.encode(hash)
+      end
+
+      def decode(str)
+        ActiveSupport::JSON.decode(str)
+      end
+    end
+  end
+end

data/lib/superstore/types/string_type.rb

@@ -0,0 +1,19 @@
+module Superstore
+  module Types
+    class StringType < BaseType
+      def encode(str)
+        raise ArgumentError.new("#{str.inspect} is not a String") unless str.kind_of?(String)
+
+        unless str.encoding == Encoding::UTF_8
+          (str.frozen? ? str.dup : str).force_encoding('UTF-8')
+        else
+          str
+        end
+      end
+
+      def typecast(value)
+        value.to_s
+      end
+    end
+  end
+end

data/lib/superstore/types/time_type.rb

@@ -0,0 +1,16 @@
+module Superstore
+  module Types
+    class TimeType < BaseType
+      def encode(time)
+        raise ArgumentError.new("#{time.inspect} is not a Time") unless time.kind_of?(Time)
+        time.utc.xmlschema(6)
+      end
+
+      def decode(str)
+        Time.parse(str).in_time_zone if str
+      rescue
+
+      end
+    end
+  end
+end

data/lib/superstore/types.rb

@@ -0,0 +1,8 @@
+Superstore::Type.register(:array,          Superstore::Types::ArrayType)
+Superstore::Type.register(:boolean,        Superstore::Types::BooleanType)
+Superstore::Type.register(:date,           Superstore::Types::DateType)
+Superstore::Type.register(:float,          Superstore::Types::FloatType)
+Superstore::Type.register(:integer,        Superstore::Types::IntegerType)
+Superstore::Type.register(:json,           Superstore::Types::JsonType)
+Superstore::Type.register(:time,           Superstore::Types::TimeType)
+Superstore::Type.register(:string,         Superstore::Types::StringType)

data/lib/superstore/validations.rb

@@ -0,0 +1,44 @@
+module Superstore
+  class RecordInvalid < StandardError
+    attr_reader :record
+    def initialize(record)
+      @record = record
+      super("Invalid record: #{@record.errors.full_messages.to_sentence}")
+    end
+  end
+
+  module Validations
+    extend ActiveSupport::Concern
+    include ActiveModel::Validations
+
+    included do
+      define_callbacks :validate, scope: :name
+    end
+
+    module ClassMethods
+      def create!(attributes = {})
+        new(attributes).tap do |object|
+          object.save!
+        end
+      end
+    end
+
+    def save(options={})
+      if perform_validations(options)
+        super
+        true
+      else
+        false
+      end
+    end
+
+    def save!
+      save || raise(RecordInvalid.new(self))
+    end
+
+    protected
+      def perform_validations(options={})
+        (options[:validate] != false) ? valid? : true
+      end
+  end
+end

data/lib/superstore.rb

@@ -0,0 +1,69 @@
+require 'active_support/all'
+require 'active_model'
+require 'superstore/errors'
+
+module Superstore
+  extend ActiveSupport::Autoload
+
+  autoload :AttributeMethods
+  autoload :Base
+  autoload :BelongsTo
+  autoload :Callbacks
+  autoload :Connection
+  autoload :Core
+  autoload :Identity
+  autoload :Inspect
+  autoload :Model
+  autoload :Persistence
+  autoload :Schema
+  autoload :CassandraSchema
+  autoload :Scope
+  autoload :Scoping
+  autoload :Serialization
+  autoload :Timestamps
+  autoload :Type
+  autoload :Validations
+
+  module AttributeMethods
+    extend ActiveSupport::Autoload
+
+    eager_autoload do
+      autoload :Definition
+      autoload :Dirty
+      autoload :PrimaryKey
+      autoload :Typecasting
+    end
+  end
+
+  module Adapters
+    extend ActiveSupport::Autoload
+
+    autoload :AbstractAdapter
+    autoload :CassandraAdapter
+    autoload :HstoreAdapter
+  end
+
+  module BelongsTo
+    extend ActiveSupport::Autoload
+
+    autoload :Association
+    autoload :Builder
+    autoload :Reflection
+  end
+
+  module Types
+    extend ActiveSupport::Autoload
+
+    autoload :BaseType
+    autoload :ArrayType
+    autoload :BooleanType
+    autoload :DateType
+    autoload :FloatType
+    autoload :IntegerType
+    autoload :JsonType
+    autoload :StringType
+    autoload :TimeType
+  end
+end
+
+require 'superstore/railtie' if defined?(Rails)

data/superstore.gemspec

@@ -0,0 +1,23 @@
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = 'superstore'
+  s.version = '1.0.0'
+  s.description = 'ActiveModel for many attributes'
+  s.summary = 'Cassandra ActiveModel'
+  s.authors = ["Michael Koziarski", "gotime"]
+  s.email = 'developer@matthewhiggins.com'
+  s.homepage = 'http://github.com/data-axle/superstore'
+
+  s.required_ruby_version     = '>= 2.0.0'
+  s.required_rubygems_version = '>= 1.3.5'
+
+  s.extra_rdoc_files = ["README.md"]
+  s.files       = `git ls-files`.split("\n")
+  s.test_files  = `git ls-files -- {test}/*`.split("\n")
+  s.require_paths = ['lib']
+
+  s.add_runtime_dependency('activemodel', '>= 3.0')
+
+  s.add_development_dependency('bundler')
+end

data/test/support/cassandra.rb

@@ -0,0 +1,44 @@
+Bundler.require :cassandra
+
+Superstore::Base.config = {
+  keyspace: 'superstore_test',
+  servers: '127.0.0.1:9160',
+  consistency: 'QUORUM',
+  thrift: {
+    timeout: 5
+  }
+}
+
+begin
+  Superstore::CassandraSchema.drop_keyspace 'superstore_test'
+rescue Exception => e
+end
+
+sleep 1
+Superstore::CassandraSchema.create_keyspace 'superstore_test'
+Superstore::CassandraSchema.create_column_family 'Issues'
+
+Superstore::Base.class_eval do
+  class_attribute :created_records
+  self.created_records = []
+
+  after_create do
+    created_records << self
+  end
+
+  def self.delete_after_test
+    # created_records.reject(&:destroyed?).each(&:destroy)
+    Issue.delete_all
+    created_records.clear
+  end
+end
+
+module ActiveSupport
+  class TestCase
+    teardown do
+      if Superstore::Base.created_records.any?
+        Superstore::Base.delete_after_test
+      end
+    end
+  end
+end

data/test/support/hstore.rb

@@ -0,0 +1,40 @@
+Bundler.require :hstore
+require 'active_record'
+
+Superstore::Base.config = {'adapter' => 'hstore'}
+
+class PGInitializer
+  def self.initialize!
+    config = {
+      'adapter'   => 'postgresql',
+      'encoding'  => 'unicode',
+      'database'  => 'superstore_test',
+      'pool'      => 5,
+      'username' => 'postgres'
+    }
+
+    ActiveRecord::Base.configurations = { test: config }
+
+    ActiveRecord::Tasks::DatabaseTasks.drop config
+    ActiveRecord::Tasks::DatabaseTasks.create config
+    ActiveRecord::Base.establish_connection config
+
+    Superstore::Base.adapter.create_table('issues')
+  end
+
+  def self.table_names
+    %w(issues)
+  end
+end
+
+PGInitializer.initialize!
+
+module ActiveSupport
+  class TestCase
+    teardown do
+      PGInitializer.table_names.each do |table_name|
+        ActiveRecord::Base.connection.execute "TRUNCATE #{table_name}"
+      end
+    end
+  end
+end

data/test/support/issue.rb

@@ -0,0 +1,10 @@
+class Issue < Superstore::Base
+  string :description
+  string :title
+
+  before_create { self.description ||= 'funny' }
+
+  def self.for_key key
+    where_ids(key)
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,42 @@
+require 'bundler/setup'
+Bundler.require(:default, :test)
+
+I18n.config.enforce_available_locales = false
+
+require 'rails/test_help'
+require 'mocha/setup'
+
+# require 'support/hstore'
+require 'support/cassandra'
+require 'support/issue'
+
+def MiniTest.filter_backtrace(bt)
+  bt
+end
+
+module Superstore
+  class TestCase < ActiveSupport::TestCase
+    def temp_object(&block)
+      Class.new(Superstore::Base) do
+        self.column_family = 'Issues'
+        string :force_save
+        before_save { self.force_save = 'junk' }
+
+        def self.name
+          'Issue'
+        end
+
+        instance_eval(&block) if block_given?
+      end
+    end
+  end
+
+  module Types
+    class TestCase < Superstore::TestCase
+      attr_accessor :coder
+      setup do
+        @coder = self.class.name.sub(/Test$/, '').constantize.new
+      end
+    end
+  end
+end

data/test/unit/active_model_test.rb

@@ -0,0 +1,18 @@
+require 'test_helper'
+
+class ActiveModelTest < Superstore::TestCase
+
+  include ActiveModel::Lint::Tests
+
+  # overrides ActiveModel::Lint::Tests#test_to_param
+  def test_to_param
+  end
+
+  # overrides ActiveModel::Lint::Tests#test_to_key
+  def test_to_key
+  end
+
+  def setup
+    @model = Issue.new
+  end
+end

data/test/unit/adapters/adapter_test.rb

@@ -0,0 +1,6 @@
+require 'test_helper'
+
+class Superstore::Adapters::AdapterTest < Superstore::TestCase
+  test 'create_table' do
+  end
+end

data/test/unit/attribute_methods/definition_test.rb

@@ -0,0 +1,13 @@
+require 'test_helper'
+
+class Superstore::AttributeMethods::DefinitionTest < Superstore::TestCase
+  class TestType < Superstore::Types::BaseType
+  end
+
+  test 'typecast' do
+    definition = Superstore::AttributeMethods::Definition.new(:foo, TestType, {a: :b})
+
+    assert_equal 'foo', definition.name
+    assert_kind_of TestType, definition.coder
+  end
+end

data/test/unit/attribute_methods/dirty_test.rb

@@ -0,0 +1,72 @@
+require 'test_helper'
+
+class Superstore::AttributeMethods::DirtyTest < Superstore::TestCase
+  test 'save clears dirty' do
+    record = temp_object do
+      string :name
+    end.new name: 'foo'
+
+    assert record.changed?
+
+    record.save!
+
+    assert_equal [nil, 'foo'], record.previous_changes['name']
+    assert !record.changed?
+  end
+
+  test 'reload clears dirty' do
+    record = temp_object do
+      string :name
+    end.create! name: 'foo'
+
+    record.name = 'bar'
+    assert record.changed?
+
+    record.reload
+
+    assert !record.changed?
+  end
+
+  test 'typecast float before dirty check' do
+    record = temp_object do
+      float :price
+    end.create(price: 5.01)
+
+    record.price = '5.01'
+    assert !record.changed?
+
+    record.price = '7.12'
+    assert record.changed?
+  end
+
+  test 'typecast boolean before dirty check' do
+    record = temp_object do
+      boolean :awesome
+    end.create(awesome: false)
+
+    record.awesome = false
+    assert !record.changed?
+
+    record.awesome = true
+    assert record.changed?
+  end
+
+  test 'write_attribute' do
+    object = temp_object do
+      string :name
+    end
+
+    expected = {"name"=>[nil, "foo"]}
+
+    object.new.tap do |record|
+      record.name = 'foo'
+      assert_equal expected, record.changes
+    end
+
+    object.new.tap do |record|
+      record[:name] = 'foo'
+      # record.write_attribute(:name, 'foo')
+      assert_equal expected, record.changes
+    end
+  end
+end

data/test/unit/attribute_methods/primary_key_test.rb

@@ -0,0 +1,26 @@
+require 'test_helper'
+
+class Superstore::AttributeMethods::PrimaryKeyTest < Superstore::TestCase
+  test 'get id' do
+    model = temp_object do
+      key do
+        "foo"
+      end
+    end
+    record = model.new
+
+    assert_equal 'foo', record.id
+  end
+
+  test 'set id' do
+    issue = Issue.new id: 'foo'
+
+    assert_equal 'foo', issue.id
+  end
+
+  test 'attributes' do
+    issue = Issue.new
+
+    assert_not_nil issue.attributes['id']
+  end
+end

data/test/unit/attribute_methods/typecasting_test.rb

@@ -0,0 +1,118 @@
+require 'test_helper'
+
+class Superstore::AttributeMethods::TypecastingTest < Superstore::TestCase
+  class CustomType
+  end
+
+  class CustomCoder < Superstore::Types::BaseType
+  end
+
+  class TestIssue < Superstore::Base
+    self.column_family = 'Issues'
+
+    attribute :custom_column, type: CustomType, coder: CustomCoder
+    boolean :enabled
+    float   :rating
+    integer :price
+    json    :orders
+    string  :name
+  end
+
+  class TestChildIssue < TestIssue
+    string :description
+  end
+
+  test 'attributes not shared' do
+    assert_nothing_raised { Issue.new.description }
+    assert_raise(NoMethodError) { TestIssue.new.description }
+    assert_nothing_raised { TestChildIssue.new.description }
+  end
+
+  test 'custom attribute definer' do
+    model_attribute = TestIssue.attribute_definitions['custom_column']
+
+    assert_kind_of CustomCoder, model_attribute.coder
+    assert_equal 'custom_column', model_attribute.name
+  end
+
+  test 'typecast_attribute' do
+    assert_equal 1, TestIssue.typecast_attribute(TestIssue.new, 'price', 1)
+    assert_equal 1, TestIssue.typecast_attribute(TestIssue.new, :price, 1)
+
+    assert_raise NoMethodError do
+      TestIssue.typecast_attribute(TestIssue.new, 'wtf', 1)
+    end
+  end
+
+  test 'boolean attribute' do
+    issue = TestIssue.create! enabled: '1'
+    assert_equal true, issue.enabled
+
+    issue = TestIssue.find issue.id
+    assert_equal true, issue.enabled
+  end
+
+  test 'float attribute' do
+    issue = TestIssue.create! rating: '4.5'
+    assert_equal 4.5, issue.rating
+
+    issue = TestIssue.find issue.id
+    assert_equal(4.5, issue.rating)
+  end
+
+  test 'integer attribute' do
+    issue = TestIssue.create! price: '101'
+    assert_equal 101, issue.price
+
+    issue = TestIssue.find issue.id
+    assert_equal(101, issue.price)
+
+    issue = TestIssue.new price: ''
+    assert_nil issue.price
+  end
+
+  test 'json attribute' do
+    issue = TestIssue.create! orders: {'a' => 'b'}
+    assert_equal({'a' => 'b'}, issue.orders)
+
+    issue = TestIssue.find issue.id
+    assert_equal({'a' => 'b'}, issue.orders)
+  end
+
+  test 'string attribute' do
+    issue = TestIssue.create! name: 'hola'
+    assert_equal('hola', issue.name)
+
+    issue = TestIssue.find issue.id
+    assert_equal('hola', issue.name)
+
+    issue = TestIssue.create! name: 42
+    assert_equal '42', issue.name
+  end
+
+  test 'multiple attributes definition' do
+    class MultipleAttributesIssue < Superstore::Base
+      self.column_family = 'Issues'
+    end
+
+    assert_nothing_raised {
+      MultipleAttributesIssue.string :hello, :greetings, :bye
+    }
+    issue = MultipleAttributesIssue.new :hello => 'hey', :greetings => 'how r u', :bye => 'see ya'
+
+    assert_equal 'how r u', issue.greetings
+  end
+
+  test 'multiple attributes with options' do
+    class MultipleAttributesIssue < Superstore::Base
+      self.column_family = 'Issues'
+    end
+
+    MultipleAttributesIssue.expects(:attribute).with(:hello, { :unique => :true, :type => :string })
+    MultipleAttributesIssue.expects(:attribute).with(:world, { :unique => :true, :type => :string })
+
+    class MultipleAttributesIssue < Superstore::Base
+      string :hello, :world, :unique => :true
+    end
+  end
+end