ee62e2a9d8
* wip * linter fixes * linter fixes * test fixes * linter fixes * fix merge + restructure piecestore packages * review feedback * linter fixes * linter fixes * remove unnecessary aliases to piecestore * more merge fixing
56 lines
1.2 KiB
Go
56 lines
1.2 KiB
Go
// Copyright (C) 2018 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package psclient
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/mr-tron/base58/base58"
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"storj.io/storj/pkg/dht"
|
|
"storj.io/storj/pkg/node"
|
|
"storj.io/storj/pkg/provider"
|
|
)
|
|
|
|
func TestNewPieceID(t *testing.T) {
|
|
t.Run("should return an id string", func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
id := NewPieceID()
|
|
assert.Equal(id.IsValid(), true)
|
|
})
|
|
|
|
t.Run("should return a different string on each call", func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
assert.NotEqual(NewPieceID(), NewPieceID())
|
|
})
|
|
}
|
|
|
|
func TestDerivePieceID(t *testing.T) {
|
|
pid := NewPieceID()
|
|
fid, err := newTestIdentity()
|
|
assert.NoError(t, err)
|
|
nid := dht.NodeID(fid.ID)
|
|
|
|
did, err := pid.Derive(nid.Bytes())
|
|
assert.NoError(t, err)
|
|
assert.NotEqual(t, pid, did)
|
|
|
|
did2, err := pid.Derive(nid.Bytes())
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, did, did2)
|
|
|
|
_, err = base58.Decode(did.String())
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
// helper function to generate new node identities with
|
|
// correct difficulty and concurrency
|
|
func newTestIdentity() (*provider.FullIdentity, error) {
|
|
ctx := context.Background()
|
|
fid, err := node.NewFullIdentity(ctx, 12, 4)
|
|
return fid, err
|
|
}
|