pluginats/connect_redis_test.lua

--  Licensed to the Apache Software Foundation (ASF) under one
--  or more contributor license agreements.  See the NOTICE file
--  distributed with this work for additional information
--  regarding copyright ownership.  The ASF licenses this file
--  to you under the Apache License, Version 2.0 (the
--  "License"); you may not use this file except in compliance
--  with the License.  You may obtain a copy of the License at
--
--  http://www.apache.org/licenses/LICENSE-2.0
--
--  Unless required by applicable law or agreed to in writing, software
--  distributed under the License is distributed on an "AS IS" BASIS,
--  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
--  See the License for the specific language governing permissions and
--  limitations under the License.

_G.ts = { client_request = {}, http = {} }
_G.client = {dbone = {}, dbdefault = {}, selecteddb = 0}
_G.TS_LUA_REMAP_DID_REMAP = 1

function ts.client_request.get_url_scheme()
    return 'http'
end

function ts.client_request.get_url_host()
    return 'test.edge.com'
end

function ts.client_request.get_uri()
    return '/app1'
end
function ts.client_request.get_url_port()
    return '80'
end

function connect(path)
  return client
end

function client.select(self, number)
  if number == 1 then
    self.selecteddb = 1
  elseif number == 0 then
    self.selecteddb = 0
  end
end

function client.sadd(self, key, ...)
  db = nil
  if self.selecteddb == 1 then 
    db = self.dbone
  elseif self.selecteddb == 0 then
    db = self.dbdefault
  end
  
  if type(db[key]) ~= "table" then
    db[key] = {}
  end

  for i=1,select('#',...) do   
    local tmp = select(i,...)
    table.insert(db[key],tmp)
  end
end

function client.ping()
  return "PONG"
end

function client.smembers(self, key)
  db = nil
  if self.selecteddb == 1 then 
    db = self.dbone
  elseif self.selecteddb == 0 then
    db = self.dbdefault
  end

  return db[key]
end

function client.srandmember(self, key)
  idx = math.random(1,2)
  db = nil
  if self.selecteddb == 1 then 
    db = self.dbone
  elseif self.selecteddb == 0 then
    db = self.dbdefault
  end
  
  return db[key][idx]
end
  

describe("Unit tests - Lua", function()
  describe("Ingress Controller", function()

    setup(function()
      local match = require("luassert.match")

      package.loaded.redis = nil
      local redis = {}
      redis.connect = {}
      redis.connect = connect
      package.preload['redis'] = function () 
        return redis
      end

      client = redis.connect()

      client:select(1)
      client:sadd("E+http://test.edge.com/app1","trafficserver-test-2:appsvc1:8080")
      client:select(0)
      client:sadd("trafficserver-test-2:appsvc1:8080","172.17.0.3#8080#http","172.17.0.5#8080#http")
      --require 'pl.pretty'.dump(client)

      stub(ts, "add_package_cpath")
      stub(ts, "add_package_path")
      stub(ts, "debug")
      stub(ts, "error")
      stub(ts.client_request, "set_url_host")
      stub(ts.client_request, "set_url_port")
      stub(ts.client_request, "set_url_scheme")
      stub(ts.client_request, "set_uri")
      stub(ts.http, "skip_remapping_set")
      stub(ts.http, "set_resp")

      -- calling get_version on client request
      stub(ts.client_request, "get_version").returns("1.1")

      -- calling get_pristine_url on client request
      stub(ts.client_request, "get_pristine_url").returns("http://test.edge.com/app1")
      
      -- Used for setting the cache url
      stub(ts.http, "set_cache_url")

      _G.TS_LUA_HOOK_SEND_RESPONSE_HDR = "TS_LUA_HOOK_SEND_RESPONSE_HDR"
      stub(ts, "hook")
    end)

     -- Test - Verify caching for Http/1.1  -------------------------

     it("Test - Verify caching for Http/1.1", function()
       require("connect_redis")
       local result = do_global_read_request()

       -- Verifies the HTTP version check and host set for 1.1
       assert.stub(ts.client_request.get_version).was.called()
       assert.stub(ts.client_request.set_url_host).was.called_with("test.edge.com")

       -- Verifies that pristine URL is used to set up cache
       assert.stub(ts.http.set_cache_url).was.called_with("http://test.edge.com/app1")
     end)

    -- Test - Verify caching for Http/2.0  -------------------

     it("Test - Verify caching for Http/2.0", function()
       -- Clear cached module first
       package.loaded["connect_redis"] = nil

       -- Re-stub get_version to return "2" before requiring
       stub(ts.client_request, "get_version").returns("2.0")

       -- Reload module
       require("connect_redis")

       -- Execute target function
       do_global_read_request()

       -- Assertions
       assert.stub(ts.client_request.get_version).was.called()
       assert.stub(ts.http.set_cache_url).was.called_with("http://test.edge.com/app1")
     end)

    it("Test - Redirect to correct IP", function()
      require("connect_redis")
      local result = do_global_read_request()

      assert.stub(ts.client_request.set_url_host).was.called_with(match.is_any_of(match.is_same("172.17.0.3"),match.is_same("172.17.0.5"))) 
      assert.stub(ts.client_request.set_url_port).was.called_with("8080")
      assert.stub(ts.client_request.set_url_scheme).was.called_with("http")
    end)

    it("Test - Snippet", function()
      client:select(1)
      client:sadd("E+http://test.edge.com/app1","$trafficserver-test-3/app-ingress/411990")
      snippet = "ts.debug('Debug msg example')\nts.error('Error msg example')\n-- ts.hook(TS_LUA_HOOK_SEND_RESPONSE_HDR, function()\n--   ts.client_response.header['Location'] = 'https://test.edge.com/app2'\n-- end)\nts.http.skip_remapping_set(0)\nts.http.set_resp(301, 'Redirect')\nts.debug('Uncomment the above lines to redirect http request to https')\nts.debug('Modification for testing')\n"
      client:sadd("$trafficserver-test-3/app-ingress/411990",snippet) 
      
      --require 'pl.pretty'.dump(client)
      require "connect_redis"
      local input_args = {"snippet"}
      local result = __init__(input_args)
      result = do_global_read_request()

      assert.stub(ts.error).was.called_with("Error msg example")
      assert.stub(ts.http.skip_remapping_set).was.called_with(0)
      assert.stub(ts.http.set_resp).was.called_with(301,"Redirect")
    end)


  end)
end)