Add Message#private? and #stream?

Author: raws

lib/wonder_llama/message.rb

@@ -1,5 +1,8 @@
 module WonderLlama
   class Message
+    PRIVATE_TYPE = 'private'.freeze
+    STREAM_TYPE = 'stream'.freeze
+
     attr_reader :client, :params
 
     def initialize(client:, params:)
@@ -19,6 +22,14 @@ def id
       self[:id]
     end
 
+    def private?
+      type == PRIVATE_TYPE
+    end
+
+    def stream?
+      type == STREAM_TYPE
+    end
+
     def to
       self[:to]
     end

spec/message_spec.rb

@@ -17,6 +17,18 @@
     }
   end
 
+  describe 'PRIVATE_TYPE' do
+    subject { described_class::PRIVATE_TYPE }
+    it { is_expected.to eq('private') }
+    it { is_expected.to be_frozen }
+  end
+
+  describe 'STREAM_TYPE' do
+    subject { described_class::STREAM_TYPE }
+    it { is_expected.to eq('stream') }
+    it { is_expected.to be_frozen }
+  end
+
   describe '#[]' do
     subject { message[:foo] }
     it { is_expected.to eq('bar') }
@@ -42,6 +54,34 @@
     it { is_expected.to eq(params.transform_keys(&:to_sym)) }
   end
 
+  describe '#private?' do
+    subject { message.private? }
+
+    context 'with a private message' do
+      before { params['type'] = described_class::PRIVATE_TYPE }
+      it { is_expected.to eq(true) }
+    end
+
+    context 'with a stream message' do
+      before { params['type'] = described_class::STREAM_TYPE }
+      it { is_expected.to eq(false) }
+    end
+  end
+
+  describe '#stream?' do
+    subject { message.stream? }
+
+    context 'with a stream message' do
+      before { params['type'] = described_class::STREAM_TYPE }
+      it { is_expected.to eq(true) }
+    end
+
+    context 'with a private message' do
+      before { params['type'] = described_class::PRIVATE_TYPE }
+      it { is_expected.to eq(false) }
+    end
+  end
+
   describe '#to' do
     subject { message.to }
     it { is_expected.to eq('social') }