You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
lua-ngcp-kamailio/tests/ngcp_pp.lua

112 lines
3.3 KiB

#!/usr/bin/env lua5.1
require('luaunit')
require 'mocks.sr'
require 'ngcp.utils'
require 'tests_v.pp_vars'
sr = srMock:new()
local mc = nil
PPFetch = {
__class__ = 'PPFetch',
_i = 1
}
function PPFetch:new()
t = {}
return setmetatable(t, { __index = PPFetch })
end
function PPFetch:val(uuid)
self._i = self._i + 1
return pp_vars[uuid][self._i-1]
end
function PPFetch:reset()
self._i = 1
end
TestNGCPPeerPrefs = {} --class
function TestNGCPPeerPrefs:setUp()
mc = lemock.controller()
self.config = mc:mock()
self.mysql = mc:mock()
self.env = mc:mock()
self.con = mc:mock()
self.cur = mc:mock()
package.loaded.luasql = nil
package.preload['luasql.mysql'] = function ()
luasql = {}
luasql.mysql = mysql
return mysql
end
require 'ngcp.pp'
self.d = NGCPPeerPrefs:new(self.config)
self.pp_vars = PPFetch:new()
end
function TestNGCPPeerPrefs:tearDown()
sr.pv.vars = {}
end
function TestNGCPPeerPrefs:test_init()
--print("TestNGCPPeerPrefs:test_init")
assertEquals(self.d.db_table, "peer_preferences")
end
function TestNGCPPeerPrefs:test_caller_load()
assertTrue(self.d.config)
self.config:getDBConnection() ;mc :returns(self.con)
self.con:execute("SELECT * FROM peer_preferences WHERE uuid = '2'") ;mc :returns(self.cur)
self.cur:fetch(mc.ANYARGS) ;mc :returns(self.pp_vars:val("p_2"))
self.cur:fetch(mc.ANYARGS) ;mc :returns(self.pp_vars:val("p_2"))
self.cur:fetch(mc.ANYARGS) ;mc :returns(nil)
self.cur:close()
self.con:close()
mc:replay()
self.d:caller_load("2")
mc:verify()
assertTrue(self.d.xavp)
assertEquals(self.d.xavp("sst_enable"),"no")
assertEquals(sr.pv.get("$xavp(domain[0]=>dummy)"), "caller")
assertEquals(self.d.xavp("dummy"),"caller")
assertEquals(sr.pv.get("$xavp(domain[0]=>sst_enable)"),"no")
assertEquals(sr.pv.get("$xavp(domain[0]=>sst_refresh_method)"), "UPDATE_FALLBACK_INVITE")
assertIsNil(self.d.xavp("error_key"))
end
function TestNGCPPeerPrefs:test_callee_load()
assertTrue(self.d.config)
self.config:getDBConnection() ;mc :returns(self.con)
self.con:execute("SELECT * FROM peer_preferences WHERE uuid = '2'") ;mc :returns(self.cur)
self.cur:fetch(mc.ANYARGS) ;mc :returns(self.pp_vars:val("p_2"))
self.cur:fetch(mc.ANYARGS) ;mc :returns(self.pp_vars:val("p_2"))
self.cur:fetch(mc.ANYARGS) ;mc :returns(nil)
self.cur:close()
self.con:close()
mc:replay()
self.d:callee_load("2")
mc:verify()
assertTrue(self.d.xavp)
assertEquals(self.d.xavp("sst_enable"),"no")
--print(table.tostring(sr.pv.vars))
assertEquals(sr.pv.get("$xavp(domain[1]=>dummy)"), "callee")
assertEquals(sr.pv.get("$xavp(domain[1]=>sst_enable)"),"no")
assertEquals(sr.pv.get("$xavp(domain[1]=>sst_refresh_method)"), "UPDATE_FALLBACK_INVITE")
assertIsNil(self.d.xavp("error_key"))
end
-- class TestNGCPPeerPrefs
---- Control test output:
lu = LuaUnit
lu:setOutputType( "TAP" )
lu:setVerbosity( 1 )
lu:run()
--EOF