--
-- Copyright 2013-2020 SipWise Team <development@sipwise.com>
--
-- This program is free software: you can redistribute it and/or modify
-- it under the terms of the GNU General Public License as published by
-- the Free Software Foundation, either version 3 of the License, or
-- (at your option) any later version.
--
-- This package is distributed in the hope that it will be useful,
-- but WITHOUT ANY WARRANTY; without even the implied warranty of
-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-- GNU General Public License for more details.
--
-- You should have received a copy of the GNU General Public License
-- along with this program. If not, see <http://www.gnu.org/licenses/>.
-- .
-- On Debian systems, the complete text of the GNU General
-- Public License version 3 can be found in "/usr/share/common-licenses/GPL-3".
--
local lu = require('luaunit')
local lemock = require('lemock')
local PProfFetch = require 'tests_v.pprof_vars'

local ksrMock = require 'mocks.ksr'
KSR = ksrMock:new()

local mc,env,con
local pprof_vars = PProfFetch:new()

package.loaded.luasql = nil
package.preload['luasql.mysql'] = function ()
    local luasql = {}
    luasql.mysql = function ()
        return env
    end
end
local NGCPConfig = require 'ngcp.config'
local NGCPProfilePrefs = require 'ngcp.pprof'
-- luacheck: ignore TestNGCPProfilePrefs
TestNGCPProfilePrefs = {} --class

    function TestNGCPProfilePrefs:setUp()
        mc = lemock.controller()
        env = mc:mock()
        con = mc:mock()
        self.cur = mc:mock()

        package.loaded.luasql = nil
        package.preload['luasql.mysql'] = function ()
            local luasql = {}
            luasql.mysql = function ()
                return env
            end
        end

        self.config = NGCPConfig:new()
        self.config.env = env
        self.config.getDBConnection = function ()
            return con
        end

        self.d = NGCPProfilePrefs:new(self.config)
        pprof_vars:reset()
    end

    function TestNGCPProfilePrefs:tearDown()
        KSR.pv.vars = {}
    end

    function TestNGCPProfilePrefs:test_init()
        --print("TestNGCPProfilePrefs:test_init")
        lu.assertEquals(self.d.db_table, "prof_preferences")
    end

    function TestNGCPProfilePrefs:test_caller_load_empty()
        lu.assertEvalToTrue(self.d.config)
        lu.assertEquals(self.d:caller_load(), {})
    end

    function TestNGCPProfilePrefs:test_callee_load_empty()
        lu.assertEvalToTrue(self.d.config)
        lu.assertEquals(self.d:callee_load(), {})
    end

    function TestNGCPProfilePrefs:test_caller_load()
        lu.assertEvalToTrue(self.d.config)
        con:execute("SELECT prefs.* FROM provisioning.voip_subscribers as usr LEFT JOIN prof_preferences AS prefs ON usr.profile_id = prefs.uuid WHERE usr.uuid = 'ah736f72-21d1-4ea6-a3ea-4d7f56b3887c'")  ;mc :returns(self.cur)
        self.cur:fetch(mc.ANYARGS)    ;mc :returns(pprof_vars:val("prof_2"))
        self.cur:fetch(mc.ANYARGS)    ;mc :returns(pprof_vars:val("prof_2"))
        self.cur:fetch(mc.ANYARGS)    ;mc :returns(pprof_vars:val("prof_2"))
        self.cur:fetch(mc.ANYARGS)    ;mc :returns(nil)
        self.cur:close()

        mc:replay()
        local keys = self.d:caller_load("ah736f72-21d1-4ea6-a3ea-4d7f56b3887c")
        mc:verify()

        lu.assertItemsEquals(keys, {"sst_enable", "sst_refresh_method", "outbound_from_user"})
        lu.assertEquals(KSR.pv.get("$xavp(caller_prof_prefs=>dummy)"), "caller")
        lu.assertEquals(KSR.pv.get("$xavp(caller_prof_prefs=>sst_enable)"),"yes")
        lu.assertEquals(KSR.pv.get("$xavp(caller_prof_prefs=>sst_refresh_method)"), "UPDATE_FALLBACK_INVITE")
        lu.assertEquals(KSR.pv.get("$xavp(caller_prof_prefs=>outbound_from_user)"), "upn")
    end

    function TestNGCPProfilePrefs:test_callee_load()
        lu.assertEvalToTrue(self.d.config)
        con:execute("SELECT prefs.* FROM provisioning.voip_subscribers as usr LEFT JOIN prof_preferences AS prefs ON usr.profile_id = prefs.uuid WHERE usr.uuid = 'ah736f72-21d1-4ea6-a3ea-4d7f56b3887c'")  ;mc :returns(self.cur)
        self.cur:fetch(mc.ANYARGS)    ;mc :returns(pprof_vars:val("prof_2"))
        self.cur:fetch(mc.ANYARGS)    ;mc :returns(pprof_vars:val("prof_2"))
        self.cur:fetch(mc.ANYARGS)    ;mc :returns(pprof_vars:val("prof_2"))
        self.cur:fetch(mc.ANYARGS)    ;mc :returns(nil)
        self.cur:close()

        mc:replay()
        local keys = self.d:callee_load("ah736f72-21d1-4ea6-a3ea-4d7f56b3887c")
        mc:verify()

        lu.assertItemsEquals(keys, {"sst_enable", "sst_refresh_method", "outbound_from_user"})
        lu.assertEquals(KSR.pv.get("$xavp(callee_prof_prefs=>dummy)"), "callee")
        lu.assertEquals(KSR.pv.get("$xavp(callee_prof_prefs=>sst_enable)"),"yes")
        lu.assertEquals(KSR.pv.get("$xavp(callee_prof_prefs=>sst_refresh_method)"), "UPDATE_FALLBACK_INVITE")
        lu.assertEquals(KSR.pv.get("$xavp(callee_prof_prefs=>outbound_from_user)"), "upn")
    end

    function TestNGCPProfilePrefs:test_clean()
        local xavp = NGCPProfilePrefs:xavp('callee')
        xavp("testid",1)
        xavp("foo","foo")
        lu.assertEquals(KSR.pv.get("$xavp(callee_prof_prefs=>testid)"),1)
        lu.assertEquals(KSR.pv.get("$xavp(callee_prof_prefs=>foo)"),"foo")
        lu.assertEquals(KSR.pv.get("$xavp(caller_prof_prefs=>dummy)"),"caller")
        lu.assertEquals(KSR.pv.get("$xavp(callee_prof_prefs=>dummy)"),"callee")
        self.d:clean()
        lu.assertEquals(KSR.pv.get("$xavp(caller_prof_prefs=>dummy)"),"caller")
        lu.assertEquals(KSR.pv.get("$xavp(callee_prof_prefs=>dummy)"),"callee")
        lu.assertNil(KSR.pv.get("$xavp(prof)"))
    end

    function TestNGCPProfilePrefs:test_callee_clean()
        local callee_xavp = NGCPProfilePrefs:xavp('callee')
        callee_xavp("testid",1)
        callee_xavp("foo","foo")
        local caller_xavp = NGCPProfilePrefs:xavp('caller')
        caller_xavp("other",1)
        caller_xavp("otherfoo","foo")
        lu.assertEquals(KSR.pv.get("$xavp(callee_prof_prefs=>testid)"),1)
        lu.assertEquals(KSR.pv.get("$xavp(callee_prof_prefs=>foo)"),"foo")
        lu.assertEquals(KSR.pv.get("$xavp(caller_prof_prefs=>dummy)"),"caller")
        lu.assertEquals(KSR.pv.get("$xavp(caller_prof_prefs=>other)"),1)
        lu.assertEquals(KSR.pv.get("$xavp(caller_prof_prefs=>otherfoo)"),"foo")
        lu.assertEquals(KSR.pv.get("$xavp(callee_prof_prefs=>dummy)"),"callee")
        self.d:clean('callee')
        lu.assertEquals(KSR.pv.get("$xavp(caller_prof_prefs=>dummy)"),'caller')
        lu.assertNil(KSR.pv.get("$xavp(callee_prof_prefs=>testid)"))
        lu.assertNil(KSR.pv.get("$xavp(callee_prof_prefs=>foo)"))
        lu.assertEquals(KSR.pv.get("$xavp(caller_prof_prefs=>other)"),1)
        lu.assertEquals(KSR.pv.get("$xavp(caller_prof_prefs=>otherfoo)"),"foo")
        lu.assertEquals(KSR.pv.get("$xavp(callee_prof_prefs=>dummy)"),"callee")
    end

    function TestNGCPProfilePrefs:test_caller_clean()
        local callee_xavp = NGCPProfilePrefs:xavp('callee')
        callee_xavp("testid",1)
        callee_xavp("foo","foo")
        local caller_xavp = NGCPProfilePrefs:xavp('caller')
        caller_xavp("other",1)
        caller_xavp("otherfoo","foo")
        lu.assertEquals(KSR.pv.get("$xavp(callee_prof_prefs=>testid)"),1)
        lu.assertEquals(KSR.pv.get("$xavp(callee_prof_prefs=>foo)"),"foo")
        lu.assertEquals(KSR.pv.get("$xavp(caller_prof_prefs=>dummy)"),"caller")
        lu.assertEquals(KSR.pv.get("$xavp(caller_prof_prefs=>other)"),1)
        lu.assertEquals(KSR.pv.get("$xavp(caller_prof_prefs=>otherfoo)"),"foo")
        lu.assertEquals(KSR.pv.get("$xavp(callee_prof_prefs=>dummy)"),"callee")
        self.d:clean('caller')
        lu.assertEquals(KSR.pv.get("$xavp(caller_prof_prefs=>dummy)"),"caller")
        lu.assertNil(KSR.pv.get("$xavp(caller_prof_prefs=>other)"))
        lu.assertNil(KSR.pv.get("$xavp(caller_prof_prefs=>otherfoo)"))
        lu.assertEquals(KSR.pv.get("$xavp(callee_prof_prefs=>testid)"),1)
        lu.assertEquals(KSR.pv.get("$xavp(callee_prof_prefs=>foo)"),"foo")
        lu.assertEquals(KSR.pv.get("$xavp(callee_prof_prefs=>dummy)"),"callee")
    end

    function TestNGCPProfilePrefs:test_tostring()
        local callee_xavp = NGCPProfilePrefs:xavp('callee')
        callee_xavp("testid",1)
        callee_xavp("foo","foo")
        local caller_xavp = NGCPProfilePrefs:xavp('caller')
        caller_xavp("other",1)
        caller_xavp("otherfoo","foo")
        lu.assertEquals(tostring(self.d), 'caller_prof_prefs:{other={1},otherfoo={"foo"},dummy={"caller"}}\ncallee_prof_prefs:{dummy={"callee"},testid={1},foo={"foo"}}\n')
    end
-- class TestNGCPProfilePrefs
--EOF