add time limit for votetx (#102)

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

diff --git a/database/store_test.go b/database/store_test.go
index e9d7527..45ff016 100644 (file)
--- a/database/store_test.go
+++ b/database/store_test.go
@@ -169,6 +169,8 @@ func TestSaveChainStatus(t *testing.T) {
                        bc.Hash{V0: 1, V1: 1, V2: 3, V3: 4}: &storage.UtxoEntry{Type: storage.NormalUTXOType, BlockHeight: 100, Spent: true},
                        bc.Hash{V0: 1, V1: 1, V2: 3, V3: 5}: &storage.UtxoEntry{Type: storage.CrosschainUTXOType, BlockHeight: 100, Spent: false},
                        bc.Hash{V0: 1, V1: 1, V2: 3, V3: 6}: &storage.UtxoEntry{Type: storage.CrosschainUTXOType, BlockHeight: 100, Spent: true},
+                       bc.Hash{V0: 1, V1: 3, V2: 3, V3: 7}: &storage.UtxoEntry{Type: storage.VoteUTXOType, BlockHeight: 100, Spent: false},
+                       bc.Hash{V0: 1, V1: 3, V2: 3, V3: 7}: &storage.UtxoEntry{Type: storage.VoteUTXOType, BlockHeight: 100, Spent: true},
                },
        }
 
@@ -188,6 +190,9 @@ func TestSaveChainStatus(t *testing.T) {
                if (utxo.Type == storage.CrosschainUTXOType) && (!utxo.Spent) {
                        continue
                }
+               if (utxo.Type == storage.VoteUTXOType) && (utxo.Spent) {
+                       continue
+               }
 
                gotUtxo, err := store.GetUtxo(&hash)
                if err != nil {