fix: schedule sign transaction

schedule_create_transaction_test.go

@@ -122,11 +122,14 @@ func TestScheduleCreateTransaction_Signature_Execute(t *testing.T) {
 
 	assert.Equal(t, HbarUnits.Hbar.numberOfTinybar(), newBalance.tinybar)
 
+	accountTransactionID := TransactionIDGenerate(client.GetOperatorAccountID())
+
 	tx, err := NewAccountCreateTransaction().
 		SetKey(newKey.PublicKey()).
 		SetMaxTransactionFee(NewHbar(2)).
 		SetInitialBalance(newBalance).
 		SetNodeAccountIDs([]AccountID{{Account: 3}}).
+		SetTransactionID(accountTransactionID).
 		FreezeWith(client)
 	assert.NoError(t, err)
 
@@ -173,17 +176,4 @@ func TestScheduleCreateTransaction_Signature_Execute(t *testing.T) {
 
 	receipt, err = resp.GetReceipt(client)
 	assert.NoError(t, err)
-
-	tx4, err := NewScheduleDeleteTransaction().
-		SetScheduleID(scheduleID).
-		FreezeWith(client)
-	assert.NoError(t, err)
-
-	resp, err = tx4.
-		Sign(newKey).
-		Execute(client)
-	assert.NoError(t, err)
-
-	_, err = resp.GetReceipt(client)
-	assert.NoError(t, err)
 }

schedule_sign_transaction.go

@@ -45,19 +45,8 @@ func (transaction *ScheduleSignTransaction) GetScheduleID() ScheduleID {
 func (transaction *ScheduleSignTransaction) AddScheduleSignature(publicKey PublicKey, signature []byte) *ScheduleSignTransaction {
 	transaction.requireOneNodeAccountID()
 
-	if transaction.pb.SigMap != nil{
-		if transaction.pb.SigMap.SigPair != nil {
-			transaction.pb.SigMap.SigPair = append(transaction.pb.SigMap.SigPair, publicKey.toSignaturePairProtobuf(signature))
-		} else {
-			transaction.pb.SigMap.SigPair = make([]*proto.SignaturePair, 0)
-			transaction.pb.SigMap.SigPair = append(transaction.pb.SigMap.SigPair, publicKey.toSignaturePairProtobuf(signature))
-		}
-	} else {
-		transaction.pb.SigMap = &proto.SignatureMap{
-			SigPair: make([]*proto.SignaturePair, 0),
-		}
-		transaction.pb.SigMap.SigPair = append(transaction.pb.SigMap.SigPair, publicKey.toSignaturePairProtobuf(signature))
-	}
+	sigPairs := transaction.pb.GetSigMap().GetSigPair()
+	sigPairs = append(sigPairs, publicKey.toSignaturePairProtobuf(signature))
 
 	return transaction
 }

transaction_receipt_query.go

@@ -42,7 +42,7 @@ func (query *TransactionReceiptQuery) GetCost(client *Client) (Hbar, error) {
 		request{
 			query: &query.Query,
 		},
-		query_shouldRetry,
+		transactionReceiptQuery_shouldRetry,
 		costQuery_makeRequest,
 		costQuery_advanceRequest,
 		costQuery_getNodeAccountID,
@@ -63,6 +63,10 @@ func transactionReceiptQuery_shouldRetry(status Status, response response) bool
 	switch status {
 	case StatusBusy, StatusUnknown, StatusReceiptNotFound:
 		return true
+	case StatusOk:
+		break
+	default:
+		return false
 	}
 
 	status = Status(response.query.GetTransactionGetReceipt().GetReceipt().GetStatus())
@@ -76,6 +80,17 @@ func transactionReceiptQuery_shouldRetry(status Status, response response) bool
 }
 
 func transactionReceiptQuery_mapResponseStatus(_ request, response response) Status {
+	status := Status(response.query.GetTransactionGetReceipt().GetHeader().GetNodeTransactionPrecheckCode())
+
+	switch status {
+	case StatusBusy, StatusUnknown, StatusReceiptNotFound:
+		return status
+	case StatusOk:
+		break
+	default:
+		return status
+	}
+
 	return Status(response.query.GetTransactionGetReceipt().GetReceipt().GetStatus())
 }
 
@@ -141,5 +156,5 @@ func (query *TransactionReceiptQuery) Execute(client *Client) (TransactionReceip
 		return TransactionReceipt{}, err
 	}
 
-	return transactionReceiptFromProtobuf(resp.query.GetTransactionGetReceipt().Receipt), nil
+	return transactionReceiptFromProtobuf(resp.query.GetTransactionGetReceipt().GetReceipt()), nil
 }