archiver_spec.rb in alephant-publisher-queue-1.4.0

- old
+ new
@@ -1,33 +1,69 @@
 require 'spec_helper'
 
 describe Alephant::Publisher::Queue::SQSHelper::Archiver do
-  describe "#see" do
-    it "calls cache put with the correct params" do
-      time_now = DateTime.parse("Feb 24 1981")
-      allow(DateTime).to receive(:now).and_return(time_now)
+  let (:cache) { instance_double("Alephant::Cache", :put => nil) }
+  let (:queue) { instance_double("AWS::SQS::Queue", :url => nil) }
+  let (:message) do
+    instance_double(
+      "AWS::SQS::ReceivedMessage",
+      :id    => "id",
+      :body  => "bar_baz",
+      :md5   => "qux",
+      :queue => queue
+    )
+  end
 
-      q = double("queue").as_null_object
-      c = double("cache").as_null_object
+  let (:opts) do
+    {
+      :log_archive_message => true,
+      :async_store         => false
+    }
+  end
 
-      expect(q).to receive(:url).and_return('url')
+  let (:subject) { described_class.new(cache, opts) }
 
-      m = Struct.new(:id, :body, :md5, :queue).new('id', 'body', 'md5', q)
+  describe "#see" do
+    let (:time_now) { DateTime.parse("Feb 24 1981") }
 
-      expect(c).to receive(:put).with(
-        "archive/#{time_now.strftime('%d-%m-%Y_%H')}/id",
-        "body",
-        {
-          :id        => "id",
-          :md5       => "md5",
-          :logged_at => time_now.to_s,
-          :queue     => "url"
-        }
-      )
+    context "calls cache put with the correct params" do
+      before(:each) do
+        allow(DateTime).to receive(:now).and_return(time_now)
+      end
 
-      instance = Alephant::Publisher::Queue::SQSHelper::Archiver.new(c, false)
+      specify do
+        expect(cache).to receive(:put).with(
+          "archive/#{time_now.strftime('%d-%m-%Y_%H')}/id",
+          message.body,
+          {
+            :id        => message.id,
+            :md5       => message.md5,
+            :logged_at => time_now.to_s,
+            :queue     => message.queue.url
+          }
+        )
+        subject.see(message)
+      end
+    end
+  end
 
-      instance.see(m)
+  describe "logging archive message" do
+    context "enabled" do
+      specify do
+        subject.see(message)
+        expect(message).to have_received(:body).twice
+      end
+    end
+
+    context "disabled" do
+      before(:each) do
+        opts[:log_archive_message] = false
+      end
+
+      specify do
+        subject.see(message)
+        expect(message).to have_received(:body).once
+      end
     end
   end
 end
 