get balance checks completely working

strangelove-ventures · Jul 11, 2023 · aa41ee8 · aa41ee8
1 parent 13e89af
commit aa41ee8
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/chain/penumbra/penumbra_app_node.go b/chain/penumbra/penumbra_app_node.go
@@ -216,6 +216,7 @@ func (p *PenumbraAppNode) GenerateGenesisFile(
 		"testnet",
 		"generate",
 		"--chain-id", chainID,
+		"--preserve-chain-id",
 		"--validators-input-file", p.ValidatorsInputFileContainer(),
 		"--allocations-input-file", p.AllocationsInputFileContainer(),
 	}

diff --git a/chain/penumbra/penumbra_client_node.go b/chain/penumbra/penumbra_client_node.go
@@ -283,9 +283,7 @@ func (p *PenumbraClientNode) GetBalance(ctx context.Context, denom string) (*big
 			continue
 		}
 
-		fmt.Printf("%v '%s'\n", metadata.Display, translateHiAndLo(b.Amount.Hi, b.Amount.Lo).String())
-
-		if metadata.Display == denom {
+		if metadata.Base == denom {
 			return translateHiAndLo(b.Amount.Hi, b.Amount.Lo), nil
 		}
 	}

diff --git a/examples/penumbra/penumbra_chain_test.go b/examples/penumbra/penumbra_chain_test.go
@@ -54,7 +54,7 @@ func TestPenumbraChainStart(t *testing.T) {
 
 	node := chain.(*penumbra.PenumbraChain).PenumbraNodes[0]
 
-	bal, err := chain.GetBalance(ctx, node.PenumbraClientNodes["validator"].KeyName, "")
+	bal, err := chain.GetBalance(ctx, node.PenumbraClientNodes["validator"].KeyName, chain.Config().Denom)
 	require.NoError(t, err)
 
 	t.Logf("Balance: %d \n", bal)