Account store test (#303)

[bytom/vapor.git] / database / wallet_store.go

diff --git a/database/wallet_store.go b/database/wallet_store.go
index 85eb2e1..a0c2a36 100644 (file)
--- a/database/wallet_store.go
+++ b/database/wallet_store.go
@@ -246,7 +246,7 @@ func (store *WalletStore) GetGlobalTransactionIndex(txID string) []byte {
 func (store *WalletStore) GetStandardUTXO(outid bc.Hash) (*acc.UTXO, error) {
        rawUTXO := store.db.Get(StandardUTXOKey(outid))
        if rawUTXO == nil {
-               return nil, fmt.Errorf("failed get standard UTXO, outputID: %s ", outid.String())
+               return nil, wallet.ErrGetStandardUTXO
        }
 
        UTXO := new(acc.UTXO)
@@ -342,6 +342,7 @@ func (store *WalletStore) ListAccountUTXOs(id string, isSmartContract bool) ([]*
 
                confirmedUTXOs = append(confirmedUTXOs, utxo)
        }
+
        return confirmedUTXOs, nil
 }
 
@@ -376,8 +377,11 @@ func (store *WalletStore) ListTransactions(accountID string, StartTxID string, c
                        return nil, err
                }
 
-               annotatedTxs = append(annotatedTxs, annotatedTx)
-               txNum--
+               if accountID == "" || wallet.FindTransactionsByAccount(annotatedTx, accountID) {
+                       annotatedTxs = append([]*query.AnnotatedTx{annotatedTx}, annotatedTxs...)
+                       txNum--
+               }
+
        }
 
        return annotatedTxs, nil