modify unit test

3scale · Sep 3, 2018 · c4231f0 · c4231f0
1 parent 85a1423
commit c4231f0
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/spec/resty/limit/count-inc_spec.lua b/spec/resty/limit/count-inc_spec.lua
@@ -77,10 +77,14 @@ describe('resty.limit.count-inc', function()
 
         describe('and incr method fails', function()
           it('returns error', function()
-            lim:incoming('tmp', true)
-            ngx.shared.limiter.incr = function()
-              return nil, 'something'
+            ngx.shared.limiter.incr = function(t, key, inc, init, _)
+              local value = t:get(key) or init
+              if not value then return nil, 'not found' end
+              if inc == -1 then return nil, 'something' end
+              t:set(key, value + inc)
+              return t:get(key)
             end
+            lim:incoming('tmp', true)
             local delay, err = lim:incoming('tmp', true)
             assert.is_nil(delay)
             assert.same('something', err)