blob: 1f10763f7e694764587ac99201520417a5468619 [file] [log] [blame]
--
-- 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.
--
local fakengx = require 'fakengx'
local fakeredis = require 'fakeredis'
local redis = require 'lib/redis'
local routing = require 'routing'
local cjson = require 'cjson'
describe('Testing routing module', function()
before_each(function()
_G.ngx = fakengx.new()
ngx.var.gatewayPath = ''
red = fakeredis.new()
operations = {
GET = {
backendUrl = 'https://httpbin.org',
backendMethod = 'GET'
}
}
keys = {'resources:guest:bp1/test/hello', 'resources:guest:bp2/hello', 'resources:guest:bp3/testing/test/hi',
'resources:guest:bp4/another/hello', 'resources:guest:nobp', 'resources:guest:noresource/'}
local field = 'resources'
for _, key in pairs(keys) do
red:hset(key, field, redis.generateResourceObj(operations, nil))
red:sadd('resources:guest:__index__', key)
end
end)
it('should find the correct redis key', function()
local expected = 'resources:guest:bp1/test/hello'
local tenant = 'guest'
local path = 'bp1/test/hello'
local ds = require 'lib/dataStore'
local actual = routing.findResource(ds.initWithDriver(red), tenant, path)
assert.are.same(expected, actual)
expected = 'bp1/test/hello'
actual = ngx.var.gatewayPath
assert.are.same(expected, actual)
end)
it('should return nil if redis key doesn\'t exist', function()
local expected = nil
local tenant = 'guest'
local path = 'bp1/bad/path'
local ds = require 'lib/dataStore'
local actual = routing.findResource(ds.initWithDriver(red), tenant, path)
assert.are.same(expected, actual)
end)
it('should find correct key when basePath is "/"', function()
local expected = 'resources:guest:nobp'
local tenant = 'guest'
local path = 'nobp'
local ds = require 'lib/dataStore'
local actual = routing.findResource(ds.initWithDriver(red), tenant, path)
assert.are.same(expected, actual)
expected = 'nobp'
actual = ngx.var.gatewayPath
assert.are.same(expected, actual)
end)
it('should find correct key when resourcePath is "/"', function()
local expected = 'resources:guest:noresource/'
local tenant = 'guest'
local path = 'noresource/'
local ds = require 'lib/dataStore'
local actual = routing.findResource(ds.initWithDriver(red), tenant, path)
assert.are.same(expected, actual)
expected = 'noresource/'
actual = ngx.var.gatewayPath
assert.are.same(expected, actual)
end)
it('should match the correct path parameters', function()
local key = 'resources:guest:bp5/{pathVar}'
local redisKey = 'resources:guest:bp5/test'
local actual = routing.pathParamMatch(key, redisKey)
local expected = true
assert.are.same(expected, actual)
expected = 'test'
actual = ngx.ctx.pathVar
assert.are.same(expected, actual)
end)
it('should return false if there isn\'t a path parameter match', function()
local key = 'resources:guest:bp6/{pathVar}/hey'
local redisKey = 'resources:guest:bp6/test/hi'
local actual = routing.pathParamMatch(key, redisKey)
local expected = false
assert.are.same(expected, actual)
expected = nil
actual = ngx.ctx.pathVar
assert.are.same(expected, actual)
end)
it('should match multiple path parameters', function()
local key = 'resources:guest:base/{var1}/hello/{var2}'
local redisKey = 'resources:guest:base/test/hello/testing'
local actual = routing.pathParamMatch(key, redisKey)
local expected = true
assert.are.same(expected, actual)
expected = 'test'
actual = ngx.ctx.var1
assert.are.same(expected, actual)
expected = 'testing'
actual = ngx.ctx.var2
assert.are.same(expected, actual)
end)
end)
describe('Testing routing with snapshotting', function()
it('Gets a redis key for a tenant with the correct snapshot', function()
local red = fakeredis.new()
red:set('snapshots:tenant:test', 'abcd1234')
red:sadd('snapshots:abcd1234:resources:test:__index__', 'snapshots:abcd1234:resources:test:v1/test')
red:hset('snapshots:abcd1234:resources:test:v1/test', 'resources', '{"operations":{}}')
local ds = require 'lib/dataStore'
local dataStore = ds.initWithDriver(red)
dataStore:setSnapshotId('test')
local result = dataStore:getAllResources('test')[1]
assert.are.same(result, 'resources:test:v1/test')
local routing = require 'routing'
local result = routing.findResource(dataStore, dataStore:getAllResources('test'), 'test', 'v1/test')
end)
end)