diff --git a/engine/globalvars_test.go b/engine/globalvars_test.go new file mode 100644 index 000000000..cdcc49175 --- /dev/null +++ b/engine/globalvars_test.go @@ -0,0 +1,52 @@ +/* +Real-time Online/Offline Charging System (OCS) for Telecom & ISP environments +Copyright (C) ITsysCOM GmbH + +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 program 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 +*/ +package engine + +import ( + "net/http" + "testing" +) + +func TestGlobalvarsSetHTTPPstrTransport(t *testing.T) { + tmpTransport := httpPstrTransport + defer func() { + SetHTTPPstrTransport(tmpTransport) + }() + newTransport := &http.Transport{ + DisableKeepAlives: true, + } + SetHTTPPstrTransport(newTransport) + if httpPstrTransport != newTransport { + t.Errorf("\nExpected <%+v>, \nReceived <%+v>", newTransport, httpPstrTransport) + } +} + +func TestGlobalvarsGetHTTPPstrTransport(t *testing.T) { + tmpTransport := httpPstrTransport + defer func() { + SetHTTPPstrTransport(tmpTransport) + }() + newTransport := &http.Transport{ + DisableKeepAlives: true, + } + SetHTTPPstrTransport(newTransport) + getTransport := GetHTTPPstrTransport() + if getTransport != newTransport { + t.Errorf("\nExpected <%+v>, \nReceived <%+v>", newTransport, getTransport) + } +}