swap test to viper

oipwg · Dec 11, 2018 · f5ec0ac · f5ec0ac
1 parent 8509f95
commit f5ec0ac
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/config/config_test.go b/config/config_test.go
@@ -2,19 +2,21 @@ package config
 
 import (
 	"testing"
+
+	"github.com/spf13/viper"
 )
 
 func TestSetTestnet(t *testing.T) {
 	// save pre test value
 	testnet := IsTestnet()
 
 	SetTestnet(true)
-	if Get("testnet").Bool(false) != true {
+	if viper.GetBool("testnet") != true {
 		t.Error("expected true, received false")
 	}
 
 	SetTestnet(false)
-	if Get("testnet").Bool(true) != false {
+	if viper.GetBool("testnet") != false {
 		t.Error("expected false, received true")
 	}