Allow specifying companies in messages

Author: benedikt

lib/userlist/push/message.rb

@@ -4,6 +4,7 @@ class Message < Resource
       include Operations::Push
 
       has_one :user, type: 'Userlist::Push::User'
+      has_one :company, type: 'Userlist::Push::Company'
 
       def initialize(payload = {}, config = Userlist.config)
         raise Userlist::ArgumentError, 'Missing required payload' unless payload

spec/userlist/push/message_spec.rb

@@ -5,6 +5,7 @@
     {
       template: 'template-identifier',
       user: 'user-identifier',
+      company: 'company-identifier',
       properties: {
         value: '$100.00'
       }
@@ -39,4 +40,23 @@
       expect(subject.push?).to be_truthy
     end
   end
+
+  context 'when a company hash is given' do
+    let(:payload) do
+      super().merge(
+        company: {
+          identifier: 'company-identifier',
+          name: 'Example Company'
+        }
+      )
+    end
+
+    it 'should convert it into a company object' do
+      expect(subject.company).to be_kind_of(Userlist::Push::Company)
+    end
+
+    it 'should be pushable' do
+      expect(subject.push?).to be_truthy
+    end
+  end
 end