Skip to content

Commit

Permalink
added qchain tests
Browse files Browse the repository at this point in the history
  • Loading branch information
0xluk committed Mar 11, 2024
1 parent f2ae970 commit b9a97ab
Show file tree
Hide file tree
Showing 2 changed files with 62 additions and 0 deletions.
34 changes: 34 additions & 0 deletions store/store_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -541,3 +541,37 @@ func TestPebbleStore_TransferTransactions(t *testing.T) {
_, err = store.GetTransferTransactionsPerTick(ctx, idOne, 14)
require.EqualError(t, err, ErrNotFound.Error())
}

func TestPebbleStore_QChainHash(t *testing.T) {
ctx := context.Background()

// Setup test environment
dbDir, err := os.MkdirTemp("", "pebble_test")
require.NoError(t, err)
defer os.RemoveAll(dbDir)

db, err := pebble.Open(filepath.Join(dbDir, "testdb"), &pebble.Options{})
require.NoError(t, err)
defer db.Close()

logger, _ := zap.NewDevelopment()
store := NewPebbleStore(db, logger)

// Sample QChainHash for testing
tickNumber := uint64(12795005)
qChainHash := []byte("qChainHash")

// Set QChainHash
err = store.PutQChainDigest(ctx, tickNumber, qChainHash)
require.NoError(t, err)

// Get QChainHash
retrievedQChainHash, err := store.GetQChainDigest(ctx, tickNumber)
require.NoError(t, err)
require.Equal(t, qChainHash, retrievedQChainHash)

// Test retrieval of non-existent QChainHash
_, err = store.GetQChainDigest(ctx, 999) // Assuming 999 is a tick number that wasn't stored
require.Error(t, err)
require.Equal(t, ErrNotFound, err)
}
28 changes: 28 additions & 0 deletions validator/qchain/qchain_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,28 @@
package qchain

import (
"github.com/stretchr/testify/require"
"testing"
)

func TestQChain_Digest(t *testing.T) {
qChain := QChain{
ComputorIndex: 1,
Epoch: 20,
Tick: 30,
Millisecond: 40,
Second: 50,
Minute: 60,
Hour: 70,
Day: 80,
Month: 90,
Year: 10,
PreviousResourceTestingDigest: 13423432,
PreviousSpectrumDigest: [32]byte{},
PreviousUniverseDigest: [32]byte{},
PreviousComputerDigest: [32]byte{},
TxDigest: [32]byte{},
}
_, err := qChain.Digest()
require.NoError(t, err)
}

0 comments on commit b9a97ab

Please sign in to comment.