refactor to pass test

FactomProject · Jan 24, 2018 · 02cabcb · 02cabcb
1 parent 8f8bb9e
commit 02cabcb
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/state/loadDatabase.go b/state/loadDatabase.go
@@ -79,12 +79,14 @@ func LoadDatabase(s *State) {
 		s.Println("******* New Database **************")
 		s.Println("***********************************\n")
 
-		customIdentity, err := primitives.HexToHash(s.CustomBootstrapIdentity)
-		if err != nil {
-			panic(fmt.Sprintf("Could not decode Custom Bootstrap Identity (likely in config file) found: %s\n", s.CustomBootstrapIdentity))
+		var customIdentity interfaces.IHash
+		if s.Network == "CUSTOM" {
+			customIdentity, err = primitives.HexToHash(s.CustomBootstrapIdentity)
+			if err != nil {
+				panic(fmt.Sprintf("Could not decode Custom Bootstrap Identity (likely in config file) found: %s\n", s.CustomBootstrapIdentity))
+			}
 		}
 		dblk, ablk, fblk, ecblk := GenerateGenesisBlocks(s.GetNetworkID(), customIdentity)
-
 		msg := messages.NewDBStateMsg(s.GetTimestamp(), dblk, ablk, fblk, ecblk, nil, nil, nil)
 		s.InMsgQueue().Enqueue(msg)
 	}