From 4453f1e7dfa657a5cfe0304003d55f3598e0ec32 Mon Sep 17 00:00:00 2001
From: David Ortiz <z.david.ortiz@gmail.com>
Date: Fri, 8 Jun 2018 15:32:49 +0200
Subject: [PATCH] spec/policy/3scale_batcher: test that it uses the
 cache_handler in the context when backend is down

---
 .../3scale_batcher/3scale_batcher_spec.lua    | 75 ++++++++++++++++++-
 1 file changed, 74 insertions(+), 1 deletion(-)

diff --git a/spec/policy/3scale_batcher/3scale_batcher_spec.lua b/spec/policy/3scale_batcher/3scale_batcher_spec.lua
index 3207cf007..31024cc74 100644
--- a/spec/policy/3scale_batcher/3scale_batcher_spec.lua
+++ b/spec/policy/3scale_batcher/3scale_batcher_spec.lua
@@ -42,11 +42,17 @@ describe('3scale batcher policy', function()
       batcher_policy.auths_cache = AuthsCache.new(lrucache.new(10), 10)
       stub(batcher_policy.reports_batcher, 'add')
 
+      stub(batcher_policy, 'backend_downtime_cache')
+
       context = {
         service = service,
         usage = usage,
-        credentials = credentials
+        credentials = credentials,
+        -- cache_handler does nothing because we just need to check if it's called
+        cache_handler = function() end
       }
+
+      stub(context, 'cache_handler')
     end)
 
     describe('when the request is cached', function()
@@ -79,5 +85,72 @@ describe('3scale batcher policy', function()
         end)
       end)
     end)
+
+    describe('when the request is not cached', function()
+      describe('and backend is available', function()
+        before_each(function()
+          local backend_client = require('apicast.backend_client')
+          stub(backend_client, 'authorize').returns({ status = 200 })
+        end)
+
+        it('updates the backend downtime cache using the handler in the context', function()
+          batcher_policy:access(context)
+
+          assert.stub(context.cache_handler).was_called()
+        end)
+      end)
+
+      describe('and backend is not available', function()
+        before_each(function()
+          local backend_client = require('apicast.backend_client')
+          stub(backend_client, 'authorize').returns({ status = 500 })
+        end)
+
+        describe('and the authorization is in the downtime cache', function()
+          describe('and it is OK', function()
+            before_each(function()
+              stub(batcher_policy.backend_downtime_cache, 'get').returns(200)
+            end)
+
+            it('adds the report to the batcher', function()
+              batcher_policy:access(context)
+
+              assert.stub(batcher_policy.reports_batcher.add).was_called_with(
+                batcher_policy.reports_batcher, transaction)
+            end)
+          end)
+
+          describe('and it is denied', function()
+            before_each(function()
+              stub(batcher_policy.backend_downtime_cache, 'get').returns(409)
+            end)
+
+            it('does not add the report to the batcher', function()
+              batcher_policy:access(context)
+
+              assert.stub(batcher_policy.reports_batcher.add).was_not_called()
+            end)
+
+            it('returns an error', function()
+              batcher_policy:access(context)
+
+              assert.is_true(ngx.status >= 400 and ngx.status < 500)
+            end)
+          end)
+        end)
+
+        describe('and the authorization is not in the downtime cache', function()
+          before_each(function()
+            stub(batcher_policy.backend_downtime_cache, 'get').returns(nil)
+          end)
+
+          it('returns an error', function()
+            batcher_policy:access(context)
+
+            assert.is_true(ngx.status >= 400 and ngx.status < 500)
+          end)
+        end)
+      end)
+    end)
   end)
 end)