MTH

[nncp.git] / src / tx_test.go

diff --git a/src/tx_test.go b/src/tx_test.go
index bde218f99043413a8be2134b533567fe1172e6a5..72c6f614e60405311cb1e37805f5e5050d4e04fe 100644 (file)
--- a/src/tx_test.go
+++ b/src/tx_test.go
@@ -1,6 +1,6 @@
 /*
 NNCP -- Node to Node copy, utilities for store-and-forward data exchange
-Copyright (C) 2016-2019 Sergey Matveev <stargrave@stargrave.org>
+Copyright (C) 2016-2021 Sergey Matveev <stargrave@stargrave.org>
 
 This program is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
@@ -27,8 +27,7 @@ import (
        "testing"
        "testing/quick"
 
-       "github.com/davecgh/go-xdr/xdr2"
-       "golang.org/x/crypto/blake2b"
+       xdr "github.com/davecgh/go-xdr/xdr2"
 )
 
 func TestTx(t *testing.T) {
@@ -84,6 +83,7 @@ func TestTx(t *testing.T) {
                        int64(src.Len()),
                        int64(padSize),
                        src,
+                       "pktName",
                )
                if err != nil {
                        return false
@@ -97,9 +97,13 @@ func TestTx(t *testing.T) {
                        return false
                }
                txJob := sentJobs[0]
-               defer txJob.Fd.Close()
+               fd, err := os.Open(txJob.Path)
+               if err != nil {
+                       panic(err)
+               }
+               defer fd.Close()
                var bufR bytes.Buffer
-               if _, err = io.Copy(&bufR, txJob.Fd); err != nil {
+               if _, err = io.Copy(&bufR, fd); err != nil {
                        panic(err)
                }
                var bufW bytes.Buffer
@@ -136,7 +140,7 @@ func TestTx(t *testing.T) {
                                if pkt.Type != PktTypeTrns {
                                        return false
                                }
-                               if bytes.Compare(pkt.Path[:blake2b.Size256], vias[i+1][:]) != 0 {
+                               if bytes.Compare(pkt.Path[:MTHSize], vias[i+1][:]) != 0 {
                                        return false
                                }
                        }