nay-record_with_operator 0.0.2 → 0.0.3

data/Rakefile

@@ -12,7 +12,7 @@ DESCRIPTION = "Rails plugin to set created_by, updated_by, deleted_by to ActiveR
 GITHUB_PROJECT = "record_with_operator"
 HOMEPAGE = "http://github.com/nay/#{GITHUB_PROJECT}/tree"
 BIN_FILES = %w( )
-VER = "0.0.2"
+VER = "0.0.3"
 CLEAN.include ['pkg']
 
 desc 'Default: run unit tests.'

data/lib/record_with_operator.rb

@@ -55,6 +55,37 @@ module RecordWithOperator
       end
 
       alias_method_chain :validate_find_options, :for
+
+      private
+        # define_method association, association= ...
+        def association_accessor_methods_with_operator(reflection, association_proxy_class)
+          association_accessor_methods_without_operator(reflection, association_proxy_class)
+          define_method("#{reflection.name}_with_operator") do |*params|
+            r = send("#{reflection.name}_without_operator", *params)
+            r.operator ||= self.operator if r && r.respond_to?(:operator=)
+            r
+          end
+          alias_method_chain "#{reflection.name}".to_sym, :operator
+
+          define_method("#{reflection.name}_with_operator=") do |new_value|
+            new_value.operator ||= self.operator if new_value.respond_to?(:operator=)
+            send("#{reflection.name}_without_operator=", new_value)
+          end
+          alias_method_chain "#{reflection.name}=".to_sym, :operator
+        end
+        alias_method_chain :association_accessor_methods, :operator
+
+        # define_method build_association, create_association ...
+        def association_constructor_method_with_operator(constructor, reflection, association_proxy_class)
+          association_constructor_method_without_operator(constructor, reflection, association_proxy_class)
+          define_method("#{constructor}_#{reflection.name}_with_operator") do |*params|
+            options = { :operator => self.operator }
+            params.empty? ? params[0] = options : params.first.merge!(options)
+            self.send("#{constructor}_#{reflection.name}_without_operator", *params)
+          end
+          alias_method_chain "#{constructor}_#{reflection.name}".to_sym, :operator
+        end
+        alias_method_chain :association_constructor_method, :operator
     end
 
     base.before_create :set_created_by
@@ -113,4 +144,4 @@ module RecordWithOperator
     self.updated_by = operator.id
   end
 
-end
+end

data/test/record_with_operator_belongs_to_association_test.rb

@@ -0,0 +1,67 @@
+require 'test_helper'
+
+class User < ActiveRecord::Base
+end
+
+class NoteBelongsToMemo < ActiveRecord::Base
+  set_table_name "notes"
+  belongs_to :memo
+end
+
+class Memo < ActiveRecord::Base
+  set_table_name "memos"
+end
+
+class RecordWithOperatorBelongsTo < ActiveSupport::TestCase
+  def setup
+    RecordWithOperator.config[:user_class_name] = "User"
+    @user1 = User.create!(:name => "user1")
+    @user2 = User.create!(:name => "user2")
+    @note_created_by_user1 = NoteBelongsToMemo.create!(:body => "test", :operator => @user1)
+  end
+
+  # belongs_to Association Test
+  # build_association
+  def test_build_memo_should_have_operator
+    note = NoteBelongsToMemo.find(@note_created_by_user1.id, :for => @user2)
+    memo = note.build_memo(:body => "memo")
+    assert_equal @user2, memo.operator
+  end
+
+  # create_association
+  def test_create_memo_should_have_operator_and_created_by
+    note = NoteBelongsToMemo.find(@note_created_by_user1.id, :for => @user2)
+    memo = note.create_memo(:body => "memo")
+    assert_equal false, memo.new_record?
+    assert_equal @user2, memo.operator
+    assert_equal @user2.id, memo.created_by
+  end
+
+  # association=
+  def test_memo_eql_should_have_operator_and_created_by
+    note = NoteBelongsToMemo.find(@note_created_by_user1.id, :for => @user2)
+    memo = Memo.new(:body => "memo")
+    note.memo = memo # not save
+    assert_equal @user2, memo.operator
+  end
+
+  # association
+  def test_auto_found_memo_should_have_operator
+    note = NoteBelongsToMemo.find(@note_created_by_user1.id, :for => @user2)
+    note.create_memo(:body => "memo")
+    assert_equal @user2, note.memo(true).operator
+  end
+
+  # association.nil?
+  def test_memo_nil_should_be_false_if_note_belongs_to_memo
+    note = NoteBelongsToMemo.find(@note_created_by_user1.id)
+    note.create_memo(:body => "memo")
+    assert !note.memo.nil?
+  end
+
+  # association.nil?
+  def test_memo_nil_should_be_true_if_note_is_independent
+    note = NoteBelongsToMemo.find(@note_created_by_user1.id)
+    assert note.memo.nil?
+  end
+end

data/test/record_with_operator_has_one_association_test.rb

@@ -0,0 +1,69 @@
+require 'test_helper'
+
+class User < ActiveRecord::Base
+end
+
+class NoteHasOneMemo < ActiveRecord::Base
+  set_table_name "notes"
+  has_one :memo, :class_name => "Memo", :foreign_key => "note_id"
+end
+
+class Memo < ActiveRecord::Base
+  set_table_name "memos"
+end
+
+class RecordWithOperatorHasOneAssociationTest < ActiveSupport::TestCase
+  def setup
+    RecordWithOperator.config[:user_class_name] = "User"
+    @user1 = User.create!(:name => "user1")
+    @user2 = User.create!(:name => "user2")
+    @note_created_by_user1 = NoteHasOneMemo.create!(:body => "test", :operator => @user1)
+  end
+
+  # has_one Association Test
+  # build_association
+  def test_build_memo_should_have_operator
+    note = NoteHasOneMemo.find(@note_created_by_user1.id, :for => @user2)
+    memo = note.build_memo(:body => "memo")
+    assert_equal @user2, memo.operator
+  end
+
+  # create_association
+  def test_create_memo_should_have_operator_and_created_by
+    note = NoteHasOneMemo.find(@note_created_by_user1.id, :for => @user2)
+    memo = note.create_memo(:body => "memo")
+    assert_equal false, memo.new_record?
+    assert_equal @user2, memo.operator
+    assert_equal @user2.id, memo.created_by
+  end
+
+  # association=
+  def test_memo_eql_should_have_operator_and_created_by
+    note = NoteHasOneMemo.find(@note_created_by_user1.id, :for => @user2)
+    memo = Memo.new(:body => "memo")
+    note.memo = memo
+    assert_equal false, memo.new_record?
+    assert_equal @user2, memo.operator
+    assert_equal @user2.id, memo.created_by
+  end
+
+  # association
+  def test_auto_found_memo_should_have_operator
+    note = NoteHasOneMemo.find(@note_created_by_user1.id, :for => @user2)
+    note.create_memo(:body => "memo")
+    assert_equal @user2, note.memo(true).operator
+  end
+
+  # association.nil?
+  def test_memo_nil_should_be_false_if_note_has_memo
+    note = NoteHasOneMemo.find(@note_created_by_user1.id)
+    note.create_memo(:body => "memo")
+    assert !note.memo.nil?
+  end
+
+  # association.nil?
+  def test_memo_nil_should_be_true_if_note_has_no_memo
+    note = NoteHasOneMemo.find(@note_created_by_user1.id)
+    assert note.memo.nil?
+  end
+end

data/test/schema.rb

@@ -23,6 +23,7 @@ ActiveRecord::Schema.define(:version => 1) do
   end
 
   create_table :notes, :force => true do |t|
+    t.column :memo_id, :integer
     t.column :body, :text
     t.column :created_by, :integer
     t.column :created_at, :datetime
@@ -46,4 +47,4 @@ ActiveRecord::Schema.define(:version => 1) do
   create_table :users, :force => true do |t|
     t.column :name, :string
   end
-end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: nay-record_with_operator
 version: !ruby/object:Gem::Version 
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors: 
 - Yasuko Ohba
@@ -42,6 +42,8 @@ files:
 - lib/record_with_operator.rb
 - tasks/record_with_operator_tasks.rake
 - test/database.yml
+- test/record_with_operator_belongs_to_association_test.rb
+- test/record_with_operator_has_one_association_test.rb
 - test/record_with_operator_test.rb
 - test/record_with_operator_user_class_name_test.rb
 - test/schema.rb
@@ -75,6 +77,8 @@ specification_version: 2
 summary: Rails plugin to set created_by, updated_by, deleted_by to ActiveRecord objects. Supports associations.
 test_files: 
 - test/database.yml
+- test/record_with_operator_belongs_to_association_test.rb
+- test/record_with_operator_has_one_association_test.rb
 - test/record_with_operator_test.rb
 - test/record_with_operator_user_class_name_test.rb
 - test/schema.rb

data/init.rb

@@ -1 +0,0 @@
-ActiveRecord::Base.instance_eval{include RecordWithOperator} unless ActiveRecord::Base.include?(RecordWithOperator)