1 files changed, 10 insertions, 5 deletions

diff --git a/src/gnunet/service/store/store_dht_test.go b/src/gnunet/service/store/store_dht_test.go
index 9f80380..5278634 100644
--- a/src/gnunet/service/store/store_dht_test.go
+++ b/src/gnunet/service/store/store_dht_test.go
@@ -66,6 +66,8 @@ func TestDHTFilesStore(t *testing.T) {
         rf := blocks.NewGenericResultFilter()
 
         // First round: save blocks
+        btype := enums.BLOCK_TYPE_TEST
+        expire := util.AbsoluteTimeNever()
         for i := 0; i < fsNumBlocks; i++ {
                 // generate random block
                 size := 1024 + rand.Intn(62000) //nolint:gosec // good enough for testing
@@ -73,10 +75,13 @@ func TestDHTFilesStore(t *testing.T) {
                 if _, err = rand.Read(buf); err != nil { //nolint:gosec // good enough for testing
                         t.Fatal(err)
                 }
-                blk := blocks.NewGenericBlock(buf)
+                var blk blocks.Block
+                if blk, err = blocks.NewBlock(btype, expire, buf); err != nil {
+                        t.Fatal(err)
+                }
                 // generate associated key
                 k := crypto.Hash(buf)
-                key := blocks.NewGenericQuery(k, enums.BLOCK_TYPE_ANY, 0)
+                key := blocks.NewGenericQuery(k, enums.BLOCK_TYPE_TEST, 0)
 
                 // store entry
                 val := &DHTEntry{
@@ -105,9 +110,9 @@ func TestDHTFilesStore(t *testing.T) {
                 k := crypto.Hash(buf)
 
                 // do the keys match?
-                if !k.Equals(key.Key()) {
-                        t.Log(hex.EncodeToString(k.Bits))
-                        t.Log(hex.EncodeToString(key.Key().Bits))
+                if !k.Equal(key.Key()) {
+                        t.Log(hex.EncodeToString(k.Data))
+                        t.Log(hex.EncodeToString(key.Key().Data))
                         t.Fatal("key/value mismatch")
                 }
         }