Add CustomButtonEvent association to GenericObject.

https://bugzilla.redhat.com/show_bug.cgi?id=1511126
lfu · Aug 29, 2018 · 5c06f88 · 5c06f88
1 parent 0f4f7b1
commit 5c06f88
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/app/models/generic_object.rb b/app/models/generic_object.rb
@@ -6,6 +6,7 @@ class GenericObject < ApplicationRecord
 
   belongs_to :generic_object_definition
   has_one :picture, :through => :generic_object_definition
+  has_many :custom_button_events, -> { where(:type => "CustomButtonEvent") }, :class_name => "EventStream", :foreign_key => :target_id
 
   validates :name, :presence => true
 

diff --git a/spec/models/generic_object_spec.rb b/spec/models/generic_object_spec.rb
@@ -376,5 +376,13 @@
         go.custom_action_buttons
       end
     end
+
+    describe '#custom_button_events' do
+      let(:cb_event) { FactoryGirl.create(:custom_button_event, :target => go) }
+
+      it 'returns list of custom button events' do
+        expect(go.custom_button_events).to match_array([cb_event])
+      end
+    end
   end
 end