92 lines
2.2 KiB
Go
92 lines
2.2 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package orders_test
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"testing"
|
|
|
|
"github.com/golang/protobuf/ptypes"
|
|
"github.com/google/go-cmp/cmp"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"storj.io/storj/internal/testcontext"
|
|
"storj.io/storj/internal/testplanet"
|
|
"storj.io/storj/pkg/auth/signing"
|
|
"storj.io/storj/pkg/pb"
|
|
"storj.io/storj/pkg/storj"
|
|
"storj.io/storj/storagenode"
|
|
"storj.io/storj/storagenode/orders"
|
|
"storj.io/storj/storagenode/storagenodedb/storagenodedbtest"
|
|
)
|
|
|
|
func TestOrders(t *testing.T) {
|
|
storagenodedbtest.Run(t, func(t *testing.T, db storagenode.DB) {
|
|
ctx := testcontext.New(t)
|
|
defer ctx.Cleanup()
|
|
|
|
ordersdb := db.Orders()
|
|
|
|
storagenode := testplanet.MustPregeneratedSignedIdentity(0)
|
|
|
|
satellite0 := testplanet.MustPregeneratedSignedIdentity(1)
|
|
|
|
uplink := testplanet.MustPregeneratedSignedIdentity(3)
|
|
piece := storj.NewPieceID()
|
|
|
|
serialNumber := newRandomSerial()
|
|
|
|
// basic test
|
|
_, err := ordersdb.ListUnsent(ctx, 100)
|
|
require.NoError(t, err)
|
|
|
|
now := ptypes.TimestampNow()
|
|
|
|
limit, err := signing.SignOrderLimit(signing.SignerFromFullIdentity(satellite0), &pb.OrderLimit2{
|
|
SerialNumber: serialNumber,
|
|
SatelliteId: satellite0.ID,
|
|
UplinkId: uplink.ID,
|
|
StorageNodeId: storagenode.ID,
|
|
PieceId: piece,
|
|
Limit: 100,
|
|
Action: pb.PieceAction_GET,
|
|
PieceExpiration: now,
|
|
OrderExpiration: now,
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
order, err := signing.SignOrder(signing.SignerFromFullIdentity(uplink), &pb.Order2{
|
|
SerialNumber: serialNumber,
|
|
Amount: 50,
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
info := &orders.Info{
|
|
Limit: limit,
|
|
Order: order,
|
|
Uplink: uplink.PeerIdentity(),
|
|
}
|
|
|
|
// basic add
|
|
err = ordersdb.Enqueue(ctx, info)
|
|
require.NoError(t, err)
|
|
|
|
// duplicate add
|
|
err = ordersdb.Enqueue(ctx, info)
|
|
require.Error(t, err, "duplicate add")
|
|
|
|
unsent, err := ordersdb.ListUnsent(ctx, 100)
|
|
require.NoError(t, err)
|
|
|
|
require.Empty(t, cmp.Diff([]*orders.Info{info}, unsent, cmp.Comparer(pb.Equal)))
|
|
})
|
|
}
|
|
|
|
// TODO: move somewhere better
|
|
func newRandomSerial() storj.SerialNumber {
|
|
var serial storj.SerialNumber
|
|
_, _ = rand.Read(serial[:])
|
|
return serial
|
|
}
|