Minor fix for missing fee types

tonicpow · Apr 25, 2022 · b9009c7 · b9009c7
1 parent 14420c8
commit b9009c7
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/policy_quote.go b/policy_quote.go
@@ -173,7 +173,15 @@ func (i *internalResult) parsePolicyQuote() (response PolicyQuoteResponse, err e
 
 	// If we have a valid payload
 	if len(response.Payload) > 0 {
-		err = json.Unmarshal([]byte(response.Payload), &response.Quote)
+		if err = json.Unmarshal([]byte(response.Payload), &response.Quote); err != nil {
+			return
+		}
+		if response.Quote != nil &&
+			len(response.Quote.Fees) > 0 &&
+			len(response.Quote.Fees[0].FeeType) == 0 { // This is an issue because go-bt json field is stripping the types
+			response.Quote.Fees[0].FeeType = FeeTypeStandard
+			response.Quote.Fees[1].FeeType = FeeTypeData
+		}
 	}
 	return
 }
diff --git a/policy_quote_test.go b/policy_quote_test.go
@@ -79,6 +79,9 @@ func TestClient_PolicyQuote(t *testing.T) {
 		assert.Equal(t, uint32(99999), response.Quote.Policies.MaxTxSizePolicy)
 		assert.Equal(t, uint64(10000000), response.Quote.Policies.MaxStackMemoryUsagePolicy)
 		assert.Equal(t, flags, response.Quote.Policies.SkipScriptFlags)
+
+		assert.Equal(t, 500, response.Quote.FeePayload.GetFee(FeeTypeStandard).MiningFee.Satoshis)
+		assert.Equal(t, 1000, response.Quote.FeePayload.GetFee(FeeTypeStandard).MiningFee.Bytes)
 	})
 
 	t.Run("invalid miner", func(t *testing.T) {