From 52b60db6b25a4c3042198a87ac4855c9b0f35d65 Mon Sep 17 00:00:00 2001 From: DanB Date: Wed, 20 Nov 2013 14:47:21 +0100 Subject: [PATCH] Adding the apier_test file again --- apier/v1/apier_test.go | 180 +++++++++++++++++++++++++++++++++++++++++ 1 file changed, 180 insertions(+) create mode 100644 apier/v1/apier_test.go diff --git a/apier/v1/apier_test.go b/apier/v1/apier_test.go new file mode 100644 index 000000000..56b087a40 --- /dev/null +++ b/apier/v1/apier_test.go @@ -0,0 +1,180 @@ +/* +Rating system designed to be used in VoIP Carriers World +Copyright (C) 2013 ITsysCOM + +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 apier + +import ( + "flag" + "github.com/cgrates/cgrates/config" + "github.com/cgrates/cgrates/utils" + "github.com/cgrates/cgrates/engine" + "os/exec" + "path" + "fmt" + "net/rpc" + "net/rpc/jsonrpc" + "time" + "reflect" + "testing" +) + +/* +README: + + Enable local tests by passing '-local' to the go test command + It is expected that the data folder of CGRateS exists at path /usr/share/cgrates/data or passed via command arguments. + Prior running the tests, create database and users by running: + mysql -pyourrootpwd < /usr/share/cgrates/data/storage/mysql/create_db_with_users.sql + What these tests do: + * Flush tables in storDb to start clean. + * Start engine with default configuration and give it some time to listen (here caching can slow down, hence the command argument parameter). + * Connect rpc client depending on encoding defined in configuration. + * Execute remote Apis and test their replies(follow prepaid1cent scenario so we can test load in dataDb also). +*/ + +var cfg *config.CGRConfig +var rater *rpc.Client + +var testLocal = flag.Bool("local", false, "Perform the tests only on local test environment, not by default.") // This flag will be passed here via "go test -local" args +var dataDir = flag.String("data_dir", "/usr/share/cgrates/data", "CGR data dir path here") +var waitRater = flag.Int("wait_rater", 200, "Number of miliseconds to wait for rater to start and cache") + +func TestCreateTables(t *testing.T) { + if !*testLocal { + return + } + cfg, _ = config.NewDefaultCGRConfig() + var mysql *engine.MySQLStorage + if d, err := engine.NewMySQLStorage(cfg.StorDBHost, cfg.StorDBPort, cfg.StorDBName, cfg.StorDBUser, cfg.StorDBPass); err != nil { + t.Fatal("Error on opening database connection: ", err) + } else { + mysql = d.(*engine.MySQLStorage) + } + for _, scriptName := range []string{engine.CREATE_CDRS_TABLES_SQL, engine.CREATE_COSTDETAILS_TABLES_SQL, engine.CREATE_MEDIATOR_TABLES_SQL, engine.CREATE_TARIFFPLAN_TABLES_SQL} { + if err := mysql.CreateTablesFromScript(path.Join(*dataDir, "storage", "mysql", scriptName)); err != nil { + t.Fatal("Error on mysql creation: ", err.Error()) + return // No point in going further + } + } + for _, tbl := range []string{utils.TBL_CDRS_PRIMARY, utils.TBL_CDRS_EXTRA} { + if _, err := mysql.Db.Query(fmt.Sprintf("SELECT 1 from %s", tbl)); err != nil { + t.Fatal(err.Error()) + } + } +} + +// Finds cgr-engine executable and starts it with default configuration +func TestStartEngine(t *testing.T) { + if !*testLocal { + return + } + enginePath, err := exec.LookPath("cgr-engine") + if err != nil { + t.Fatal("Cannot find cgr-engine executable") + } + exec.Command("pkill", "cgr-engine").Run() // Just to make sure another one is not running, bit brutal maybe we can fine tune it + engine := exec.Command(enginePath, "-rater", "-config", path.Join(*dataDir,"conf","cgrates.cfg")) + if err := engine.Start(); err != nil { + t.Fatal("Cannot start cgr-engine: ", err.Error()) + } + time.Sleep(time.Duration(*waitRater)*time.Millisecond) // Give time to rater to fire up +} + +// Connect rpc client to rater +func TestRpcConn(t *testing.T) { + if !*testLocal { + return + } + var err error + if cfg.RPCEncoding == utils.JSON { + rater, err = jsonrpc.Dial("tcp", cfg.MediatorRater) + } else { + rater, err = rpc.Dial("tcp", cfg.MediatorRater) + } + if err != nil { + t.Fatal("Could not connect to rater: ", err.Error()) + } +} + +// Test here TPTiming APIs +func TestApierTPTiming(t *testing.T) { + if !*testLocal { + return + } + // ALWAYS,*any,*any,*any,*any,00:00:00 + tmAlways := &utils.ApierTPTiming{TPid: engine.TEST_SQL, + TimingId: "ALWAYS", + Years: "*any", + Months: "*any", + MonthDays: "*any", + WeekDays: "*any", + Time: "00:00:00", + } + tmAlways2 := new(utils.ApierTPTiming) + *tmAlways2 = *tmAlways + tmAlways2.TimingId = "ALWAYS2" + tmAsap := &utils.ApierTPTiming{TPid: engine.TEST_SQL, + TimingId: "ASAP", + Years: "*any", + Months: "*any", + MonthDays: "*any", + WeekDays: "*any", + Time: "*asap", + } + reply := "" + for _, tm := range []*utils.ApierTPTiming{tmAlways, tmAsap, tmAlways2} { + if err := rater.Call("ApierV1.SetTPTiming", tm, &reply); err!=nil { + t.Error("Got error on ApierV1.SetTPTiming: ", err.Error()) + } else if reply != "OK" { + t.Error("Unexpected reply received when calling ApierV1.SetTPTiming: ", reply) + } + } + // Check second set + if err := rater.Call("ApierV1.SetTPTiming", tmAlways, &reply); err!=nil { + t.Error("Got error on second ApierV1.SetTPTiming: ", err.Error()) + } else if reply != "OK" { + t.Error("Calling ApierV1.SetTPTiming got reply: ", reply) + } + // Check missing params + if err := rater.Call("ApierV1.SetTPTiming", new(utils.ApierTPTiming), &reply); err==nil { + t.Error("Calling ApierV1.SetTPTiming, expected error, received: ", reply) + } else if err.Error() != "MANDATORY_IE_MISSING:[TPid TimingId Years Months MonthDays WeekDays Time]" { + t.Error("Calling ApierV1.SetTPTiming got unexpected error: ", err.Error()) + } + // Test get + var rplyTmAlways2 *utils.ApierTPTiming + if err := rater.Call("ApierV1.GetTPTiming", AttrGetTPTiming{tmAlways2.TPid, tmAlways2.TimingId}, &rplyTmAlways2); err!=nil { + t.Error("Calling ApierV1.GetTPTiming, got error: ", err.Error()) + } else if !reflect.DeepEqual(tmAlways2, rplyTmAlways2) { + t.Errorf("Calling ApierV1.GetTPTiming expected: %v, received: %v", tmAlways, rplyTmAlways2) + } + // Test remove + if err := rater.Call("ApierV1.RemTPTiming", AttrGetTPTiming{tmAlways2.TPid, tmAlways2.TimingId}, &reply); err!=nil { + t.Error("Calling ApierV1.RemTPTiming, got error: ", err.Error()) + } else if reply != "OK" { + t.Error("Calling ApierV1.RemTPTiming received: ", reply) + } + // Test getIds + var rplyTmIds []string + expectedTmIds := []string{"ALWAYS", "ASAP"} + if err := rater.Call("ApierV1.GetTPTimingIds", AttrGetTPTimingIds{tmAlways.TPid}, &rplyTmIds); err!=nil { + t.Error("Calling ApierV1.GetTPTimingIds, got error: ", err.Error()) + } else if !reflect.DeepEqual(expectedTmIds, rplyTmIds) { + t.Errorf("Calling ApierV1.GetTPTimingIds expected: %v, received: %v", expectedTmIds, rplyTmIds) + } +}