edave-enumerated_attribute 0.2.18

data/spec/active_record/association_test_classes.rb

@@ -0,0 +1,40 @@
+
+class Company < ActiveRecord::Base
+	has_one :license
+	has_many :employees
+	has_many :contract_workers
+	has_many :contractors, :through=>:contract_workers
+	enum_attr :status, %w(s_corp c_corp llc), :nil=>true
+end
+class Employee < ActiveRecord::Base
+  belongs_to :company
+	enum_attr :status, %w(full_time part_time suspended), :nil=>true
+end
+class License < ActiveRecord::Base
+  belongs_to :company
+	enum_attr :status, %w(^current expired)
+end
+class Contractor < ActiveRecord::Base
+	has_many :contract_workers
+	has_many :companies, :through=>:contract_workers
+	enum_attr :status, %w(^available unavailable)
+end
+class ContractWorker < ActiveRecord::Base
+  belongs_to :company
+	belongs_to :contractor
+	enum_attr :status, %w(unfresh ^unfresh) #i don't know what to put here
+end
+
+#polymorphic
+class Comment < ActiveRecord::Base
+	belongs_to :document, :polymorphic=>true
+	enum_attr :status, %w(^unflagged flagged)
+end
+class Article < ActiveRecord::Base
+	has_one :comment, :as=>:document
+	enum_attr :status, %w(^unreviewed accepted)
+end
+class Image < ActiveRecord::Base
+	has_one :comment, :as=>:document
+	enum_attr :status, %w(^unreviewed accepted)
+end

data/spec/active_record/associations_spec.rb

@@ -0,0 +1,130 @@
+require 'test_in_memory'
+require 'enumerated_attribute'
+require 'active_record'
+require 'association_test_classes'
+
+module TestVariables
+	def company_name; 'Company A'; end
+end
+
+describe "Polymorphic associations" do
+	include TestVariables
+	
+	it "should retrieve enum'ed attribute for articles and images of a comment" do
+		c1 = Comment.new(:comment=>'i like it')
+		c1.document = Article.create!(:name=>'Birds of a Feather', :status=>:accepted)
+		c1.save!
+		c1.status.should == :unflagged
+		
+		c2 = Comment.new(:comment=>'i hate it', :status => :flagged)
+		c2.document = Image.create!(:name=>'Martian Landscape', :status=>:unreviewed)
+		c2.save!
+		c2.status.should == :flagged
+		
+		Comment.find(c1.id).document.status.should == :accepted
+		Comment.find(c2.id).document.status.should == :unreviewed
+
+	end
+	
+	it "should retrieve enum'ed attribute for comments on articles and images" do
+		a = Article.create!(:name=>'Swimming with Whales', :status=>:accepted)
+		a.create_comment(:comment=>'i like it', :status=>:unflagged)
+		a.save!
+		
+		i = Image.create!(:name=>'Mountain Climbing', :status=>:unreviewed)
+		i.comment = Comment.create!(:comment=>'i do not like it', :status=>:flagged)
+		i.save!
+		
+		Image.find(i.id).comment.status.should == :flagged
+		Article.find(a.id).comment.status.should == :unflagged
+		
+	end
+	
+end
+
+describe "Basic Associations" do
+	include TestVariables
+
+	it "should retrieve enum'ed status for its license" do
+		c = Company.new(:status=>:llc, :name=>company_name)
+		c.save!
+		c.create_license(:status=>:current)
+		
+		r = Company.find(c.id)
+		lic = r.license
+		r.status.should == :llc
+		lic.status.should == :current
+	end
+	
+	it "should retrieve enum'ed status for multiple employees" do
+		c=Company.new(:status=>:llc, :name=>company_name)
+		c.save!
+		c.employees.create!(:status=>:full_time, :name=>'edward')
+		c.employees.create!(:status=>:suspended, :name=>'tina')
+		
+		r=Company.find(c.id)
+		r.employees.find_by_name('edward').status.should == :full_time		
+		r.employees.find_by_name('tina').status.should == :suspended		
+		Employee.delete_all
+	end
+	
+	it "should retrieve enum'ed status for multiple contractors" do
+		c=Company.new(:status=>:llc, :name=>company_name)
+		c.save!
+		c1 = Contractor.new(:status=>:available, :name=>'john')
+		c2 = Contractor.new(:status=>:unavailable, :name=>'sally')
+		c.contractors << c1
+		c.contractors << c2
+		
+		r=Company.find(c.id)
+		r.contractors.find_by_name('sally').status.should == :unavailable
+		r.contractors.find_by_name('john').status.should == :available
+		Contractor.delete_all
+		ContractWorker.delete_all
+	end
+end
+
+describe "License" do
+	include TestVariables
+  it "should retrieve the status for the company" do
+		Company.delete_all
+		License.delete_all
+		lic = License.create!(:status=>:expired)
+		lic.company = Company.create!(:name=>company_name, :status=>:llc)
+		lic.save!
+		
+		License.find(lic.id).company.status.should == :llc
+	end
+end
+
+describe "Contractor" do
+	include TestVariables
+	it "should retrieve enum'ed status from multiple companies" do
+		Company.delete_all
+		Contractor.delete_all
+		ContractWorker.delete_all
+		c = Contractor.create!(:name=>'jack', :status=>:available)
+		
+		c1 = Company.create!(:name=>company_name, :status=>:s_corp)
+		c2 = Company.create!(:name=>'other company', :status=>:llc)
+		c.companies << c1 << c2
+		
+		j = Contractor.find(c.id)
+		j.companies.find_by_name('other company').status.should == :llc
+		j.companies.find_by_name(company_name).status.should == :s_corp		
+	end
+end
+
+describe "Employee" do
+	include TestVariables
+	it "should retrieve enum'ed status from the company" do
+		Company.delete_all
+		Employee.delete_all
+		e = Employee.create!(:name=>'juanita', :status=>:part_time)
+		e.company = Company.create!(:name=>company_name, :status=>:c_corp)
+		e.save!
+		
+		emp = Employee.find(e.id)
+		emp.company.status.should == :c_corp
+	end
+end

data/spec/active_record/cfg.rb

@@ -0,0 +1,6 @@
+#irb -r cfg ==> to work configure for working in IRB
+$:.unshift '../../lib'
+require 'test_in_memory'
+require 'enumerated_attribute'
+require 'race_car'
+require 'sti_classes'

data/spec/active_record/df.rb

@@ -0,0 +1,12 @@
+require 'cfg'
+
+r=RaceCar.new
+r.gear = :second
+r.name = 'special'
+r.save!
+
+#RaceCar.find_by_gear_and_name(:second, 'special')
+car = RaceCar.find_or_create_by_name_and_gear('special', :second)
+debugger
+car0 = RaceCar.find_or_create_by_name_and_gear('special', :second)
+car0

data/spec/active_record/inheritance_classes.rb

@@ -0,0 +1,19 @@
+require 'race_car'
+
+class SubRaceCar < RaceCar
+	set_table_name "race_cars"
+
+	enum_attr :extra, %w(^one two three four)
+end
+
+class SubRaceCar2 < RaceCar
+	set_table_name "race_cars"
+	
+end
+
+class SubRaceCar3 < RaceCar
+	set_table_name "race_cars"
+	
+	enum_attr :gear, %w(reverse neutral ^first second third over_drive)
+end
+

data/spec/active_record/inheritance_spec.rb

@@ -0,0 +1,60 @@
+require 'test_in_memory'
+require 'enumerated_attribute'
+require 'active_record'
+require 'active_record/inheritance_classes'
+
+def one_to_four; [:one, :two, :three, :four]; end
+def gear1; [:reverse, :neutral, :first, :second, :over_drive]; end
+def gear2; [:reverse, :neutral, :first, :second, :third, :over_drive]; end
+def choke; [:none, :medium, :full]; end
+
+describe "SubRaceCar" do
+	it "should have :gear init to :neutral" do
+		o=SubRaceCar.new
+		o.gear.should == :neutral
+		o.enums(:gear).should == gear1
+		o.gear_previous.should == :reverse
+	end
+	it "should have :choke init to :none" do
+		o=SubRaceCar.new
+		o.choke.should == :none
+		o.enums(:choke).should == choke
+		o.choke_previous.should == :full
+	end
+	it "should have :extra init to :one" do
+		o=SubRaceCar.new
+		o.extra.should == :one
+		o.enums(:extra).should == one_to_four
+		o.extra_previous.should == :four
+	end
+end
+
+describe "SubRaceCar2" do
+	it "should have :gear init to :neutral" do
+		o=SubRaceCar2.new
+		o.gear.should == :neutral
+		o.enums(:gear).should == gear1
+		o.gear_previous.should == :reverse
+	end
+	it "should have :choke init to :none" do
+		o=SubRaceCar2.new
+		o.choke.should == :none
+		o.enums(:choke).should == choke
+		o.choke_previous.should == :full
+	end
+end
+
+describe "SubRaceCar3" do
+	it "should have overridden :gear init to :first" do
+		o=SubRaceCar3.new
+		o.gear.should == :first
+		o.enums(:gear).should == gear2
+		o.gear_previous.should == :neutral
+	end
+	it "should have :choke init to :none" do
+		o=SubRaceCar3.new
+		o.choke.should == :none
+		o.enums(:choke).should == choke
+		o.choke_previous.should == :full
+	end
+end

data/spec/active_record/race_car.rb

@@ -0,0 +1,15 @@
+
+class RaceCar	< ActiveRecord::Base
+	enum_attr :gear, %w(reverse ^neutral first second over_drive)
+	enum_attr :choke, %w(^none medium full)
+end
+
+#gear = enumerated column attribute
+#choke = enumerated non-column attribute
+#lights = non-enumerated column attribute
+
+=begin
+	t.string :name
+	t.string :gear
+	t.string :lights
+=end

data/spec/active_record/sti_classes.rb

@@ -0,0 +1,10 @@
+
+class StiParent < ActiveRecord::Base
+  enum_attr :parent_enum, %w(p1 p2 p3)
+
+end
+
+class StiSub < StiParent
+  enum_attr :sub_enum, %w(s1 s2 s3)
+end
+

data/spec/active_record/sti_spec.rb

@@ -0,0 +1,41 @@
+require 'test_in_memory'
+require 'enumerated_attribute'
+require 'active_record'
+require 'active_record/sti_classes'
+
+
+#contributed by Rob Chekaluk
+
+describe "enum_attr for STI subclass" do
+	it "should assign, save and retrieve non-enumerated attribute" do
+		s = StiSub.new()
+		s.sub_nonenum = "value2"
+    s.save
+    s0 = StiSub.find(s.id)
+    s0.sub_nonenum.should == "value2"
+	end
+	it "should assign, save and retrieve subclass enum" do
+		s = StiSub.new()
+		s.sub_enum = :s3
+		s.sub_enum.should == :s3
+    s.save
+    s0 = StiSub.find(s.id)
+    s0.sub_enum.should == :s3
+	end
+	it "should assign, save and retrieve inherited enum" do
+		s = StiSub.new()
+    s.parent_enum = :p2
+    s.parent_enum.should == :p2
+    s.save
+    s0 = StiSub.find(s.id)
+    s0.parent_enum.should == :p2
+	end
+  it "should use predicate methods to access enumerated attributes" do
+    s=StiSub.create(:parent_enum=>:p2, :sub_enum=>:s1)
+    s.parent_enum_is_p2?.should be_true
+    s.parent_enum_is_not_p2?.should be_false
+    s.sub_enum_is_s2?.should be_false
+    s.sub_enum_is_not_s2?.should be_true
+  end
+end
+

data/spec/active_record/test_in_memory.rb

@@ -0,0 +1,71 @@
+require 'rubygems'
+
+gem 'activerecord', ENV['AR_VERSION'] ? "=#{ENV['AR_VERSION']}" : '>=2.1.0'
+require 'active_record'
+require 'active_support/core_ext/logger' rescue nil  # rails3
+
+require 'enumerated_attribute'
+
+ActiveRecord::Base.establish_connection({'adapter' => 'sqlite3', 'database' => ':memory:'})
+ActiveRecord::Base.logger = Logger.new("#{File.dirname(__FILE__)}/active_record.log")
+
+connection = ActiveRecord::Base.connection
+  connection.create_table(:race_cars, :force=>true) do |t|
+	t.string :name
+	t.enum :gear
+	t.enum :lights
+  t.timestamps
+end
+connection.create_table(:bicycles, :force=>true) do |t|
+	t.string :name
+	t.enum :speed
+	t.enum :gear
+end
+	
+#basic_associations
+connection.create_table(:companies, :force=>true) do |t|
+	t.string :name
+	t.string :status
+end
+connection.create_table(:contract_workers, :force=>true) do |t|
+	t.references :company
+	t.references :contractor
+	t.string :status
+end
+connection.create_table(:licenses, :force=>true) do |t|
+	t.references :company
+	t.string :status
+end
+connection.create_table(:contractors, :force=>true) do |t|
+	t.string :name
+	t.string :status
+end
+connection.create_table(:employees, :force=>true) do |t|
+	t.references :company
+	t.string :name
+	t.string :status
+end
+
+#polymorphic_associations
+connection.create_table(:comments, :force=>true) do |t|
+	t.references :document, :polymorphic=>true
+	t.text :comment
+	t.string :status
+end
+connection.create_table(:articles, :force=>true) do |t|
+	t.string :name
+	t.string :status
+end
+connection.create_table(:images, :force=>true) do |t|
+	t.string :name
+	t.string :status
+end
+
+#single table inheritance
+connection.create_table(:sti_parents, :force=>true) do |t|
+  t.string :type
+  t.enum :parent_enum
+  t.string :sub_nonenum
+  t.enum :sub_enum
+  t.timestamps
+end

data/spec/car.rb

@@ -0,0 +1,67 @@
+require 'enumerated_attribute'
+
+#used to test that method_missing chaining plays nice in inheritance situations
+
+class Vehicle  
+  def vehicle_method_missing_called?; @vehicle_method_missing_called; end
+	def self.vehicle_new_called?; @@vehicle_new_called; end
+	def self.reset; @@vehicle_new_called = false; end
+	
+	@@vehicle_new_called = false  
+  def initialize
+    @vehicle_method_missing_called = false
+		super
+  end
+  
+  def method_missing(methId, *args)
+    @vehicle_method_missing_called = true
+    #end here
+  end
+	
+	def self.new
+		@@vehicle_new_called = true
+		super
+	end
+  
+end
+
+class CarWithMethods < Vehicle  
+  def car_method_missing_called?; @car_method_missing_called; end
+	def self.car_new_called?; @@car_new_called; end
+	def self.reset; @@car_new_called = false; super; end
+
+	def self.new 
+		@@car_new_called = true
+		super
+	end
+	
+  def method_missing(methId, *args)
+    @car_method_missing_called = true
+    super
+  end		
+
+  enum_attr :gear, %w(reverse ^neutral drive)  
+
+	@@car_new_called = false  
+  def initialize
+    @car_method_missing_called = false
+    super
+  end
+  
+end
+
+class CarWithoutMethods < Vehicle
+  def car_method_missing_called?; @car_method_missing_called; end
+	def self.car_new_called?; @@car_new_called; end
+	def self.reset; @@car_new_called = false; super; end
+  
+  enum_attr :gear, %w(reverse ^neutral drive)  
+
+	@@car_new_called = false  
+  def initialize
+    @car_method_missing_called = false
+    super
+  end
+  
+end
+