cgrates/cmd/cgr-migrator/cgr-migrator_test.go

/*
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 Affero 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 Affero General Public License for more details.

You should have received a copy of the GNU Affero General Public License
along with this program.  If not, see <https://www.gnu.org/licenses/>
*/

package main

import (
	"testing"
	"time"
)

// if the flag change this should fail
// do not use constants in this test
func TestFlags(t *testing.T) {
	if err := cgrMigratorFlags.Parse([]string{"-config_path", "true"}); err != nil {
		t.Fatal(err)
	} else if *cfgPath != "true" {
		t.Errorf("Expected true received:%v ", *cfgPath)
	}
	if err := cgrMigratorFlags.Parse([]string{"-print_config", "true"}); err != nil {
		t.Fatal(err)
	} else if *printConfig != true {
		t.Errorf("Expected true, received %+v", *printConfig)
	}
	if err := cgrMigratorFlags.Parse([]string{"-exec", "true"}); err != nil {
		t.Fatal(err)
	} else if *exec != "true" {
		t.Errorf("Expected true received:%v ", *exec)
	}
	if err := cgrMigratorFlags.Parse([]string{"-version", "true"}); err != nil {
		t.Fatal(err)
	} else if !*version {
		t.Errorf("Expected true received:%v ", *version)
	}
	if err := cgrMigratorFlags.Parse([]string{"-dbdata_encoding", "true"}); err != nil {
		t.Fatal(err)
	} else if *inDBDataEncoding != "true" {
		t.Errorf("Expected true received:%v ", *inDBDataEncoding)
	}
	if err := cgrMigratorFlags.Parse([]string{"-redisSentinel", "true"}); err != nil {
		t.Fatal(err)
	} else if *inDataDBRedisSentinel != "true" {
		t.Errorf("Expected true received:%v ", *inDataDBRedisSentinel)
	}
	if err := cgrMigratorFlags.Parse([]string{"-redisCluster", "true"}); err != nil {
		t.Fatal(err)
	} else if !*dbRedisCluster {
		t.Errorf("Expected true received:%v ", *dbRedisCluster)
	}
	if err := cgrMigratorFlags.Parse([]string{"-redisClusterSync", "3s"}); err != nil {
		t.Fatal(err)
	} else if *dbRedisClusterSync != 3*time.Second {
		t.Errorf("Expected true received:%v ", *dbRedisClusterSync)
	}
	if err := cgrMigratorFlags.Parse([]string{"-redisClusterOndownDelay", "3s"}); err != nil {
		t.Fatal(err)
	} else if *dbRedisClusterDownDelay != 3*time.Second {
		t.Errorf("Expected true received:%v ", *dbRedisClusterDownDelay)
	}
	if err := cgrMigratorFlags.Parse([]string{"-mongoQueryTimeout", "3s"}); err != nil {
		t.Fatal(err)
	} else if *dbQueryTimeout != 3*time.Second {
		t.Errorf("Expected true received:%v ", *dbQueryTimeout)
	}
	if err := cgrMigratorFlags.Parse([]string{"-redisTLS", "true"}); err != nil {
		t.Fatal(err)
	} else if !*dbRedisTls {
		t.Errorf("Expected true received:%v ", *dbRedisTls)
	}
	if err := cgrMigratorFlags.Parse([]string{"-redisClientCertificate", "true"}); err != nil {
		t.Fatal(err)
	} else if *dbRedisClientCertificate != "true" {
		t.Errorf("Expected true received:%v ", *dbRedisClientCertificate)
	}
	if err := cgrMigratorFlags.Parse([]string{"-redisClientKey", "true"}); err != nil {
		t.Fatal(err)
	} else if *dbRedisClientKey != "true" {
		t.Errorf("Expected true received:%v ", *dbRedisClientKey)
	}
	if err := cgrMigratorFlags.Parse([]string{"-redisCACertificate", "true"}); err != nil {
		t.Fatal(err)
	} else if *dbRedisCACertificate != "true" {
		t.Errorf("Expected true received:%v ", *dbRedisCACertificate)
	}
	if err := cgrMigratorFlags.Parse([]string{"-out_redis_sentinel", "true"}); err != nil {
		t.Fatal(err)
	} else if *outDataDBRedisSentinel != "true" {
		t.Errorf("Expected true received:%v ", *outDataDBRedisSentinel)
	}

	if err := cgrMigratorFlags.Parse([]string{"-dry_run", "true"}); err != nil {
		t.Fatal(err)
	} else if !*dryRun {
		t.Errorf("Expected true received:%v ", *dryRun)
	}
	if err := cgrMigratorFlags.Parse([]string{"-verbose", "true"}); err != nil {
		t.Fatal(err)
	} else if !*verbose {
		t.Errorf("Expected true received:%v ", *verbose)
	}

}