Update test to match new data structure

gbjk · Oct 3, 2023 · 11edc1c · 11edc1c
1 parent 9c340dd
commit 11edc1c
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/exchanges/bitstamp/bitstamp_test.go b/exchanges/bitstamp/bitstamp_test.go
@@ -160,9 +160,12 @@ func TestGetAccountTradingFee(t *testing.T) {
 
 	fee, err := b.GetAccountTradingFee(context.Background(), currency.NewPair(currency.LTC, currency.BTC))
 	assert.NoError(t, err, "GetAccountTradingFee should not error")
-	assert.Len(t, fee, 1, "GetAccountTradingFee should return only 1 fee")
-	assert.Positive(t, fee.Fees[0], "Maker should be positive")
-	assert.Positive(t, fee.Fees[1], "Taker should be positive")
+	assert.Equal(t, fee.Symbol, "ltcbtc", "Symbol should be correct")
+	if mockTests {
+		// Live endpoint may well return 0 for these Fees on Bitstamp for new users
+		assert.Positive(t, fee.Fees.Maker, "Maker should be positive")
+		assert.Positive(t, fee.Fees.Taker, "Taker should be positive")
+	}
 }
 
 func TestGetAccountTradingFees(t *testing.T) {