fix tests

gateway/src/apicast/balancer.lua

@@ -22,7 +22,7 @@ function _M.call(_, _, balancer)
 
   local peer, err = balancer:select_peer(peers)
 
-  local address, port, ok
+  local address, port
 
   if not peer then
     ngx.log(ngx.ERR, 'could not select peer: ', err)
@@ -40,7 +40,8 @@ function _M.call(_, _, balancer)
     port = get_default_port(ngx.var.proxy_pass)
   end
 
-  ok, err = balancer.balancer.set_current_peer(address, port)
+  local _
+  _, err = balancer.balancer.set_current_peer(address, port)
 
   ngx.log(ngx.INFO, 'balancer set peer ', address, ':', port)
 

spec/balancer_spec.lua

@@ -0,0 +1,22 @@
+local apicast_balancer = require 'apicast.balancer'
+local resty_balancer = require 'resty.balancer'
+
+describe('apicast.balancer', function()
+
+  describe('.call', function()
+    local b = resty_balancer.new(function(peers) return peers[1] end)
+    b.peers = function() return { { '127.0.0.2' } } end
+    b.balancer = { }
+
+    ngx.var = {
+      proxy_host = 'upstream',
+      proxy_pass = 'https://example.com'
+    }
+
+    it('returns peers from servers', function()
+      b.balancer.set_current_peer = spy.new(function() return true end)
+      apicast_balancer:call({},b)
+      assert.spy(b.balancer.set_current_peer).was.called_with('127.0.0.2', 443)
+    end)
+  end)
+end)

spec/resty/balancer_spec.lua

@@ -52,19 +52,6 @@ describe('resty.balancer', function()
       assert.truthy(ok)
       assert.spy(b.balancer.set_current_peer).was.called_with('127.0.0.2', 8091)
     end)
-
-    it('works with default port', function()
-      local peers = {
-        { '127.0.0.2' }
-      }
-      b.balancer.set_current_peer = spy.new(function() return true end)
-
-      local ok, err = b:set_peer(peers, 443)
-
-      assert.falsy(err)
-      assert.truthy(ok)
-      assert.spy(b.balancer.set_current_peer).was.called_with('127.0.0.2', 443)
-    end)
   end)
 
   describe('round-robin balancer', function()