70 lines
1.6 KiB
Go
70 lines
1.6 KiB
Go
// Copyright (C) 2021 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package audit_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/zeebo/errs"
|
|
|
|
"storj.io/common/rpc"
|
|
"storj.io/common/rpc/rpcstatus"
|
|
"storj.io/storj/satellite/audit"
|
|
)
|
|
|
|
func TestPieceAuditFromErr(t *testing.T) {
|
|
cases := []struct {
|
|
Error error
|
|
Audit audit.PieceAudit
|
|
}{
|
|
{
|
|
Error: nil,
|
|
Audit: audit.PieceAuditSuccess,
|
|
},
|
|
{
|
|
Error: rpc.Error.Wrap(context.DeadlineExceeded),
|
|
Audit: audit.PieceAuditOffline,
|
|
},
|
|
{
|
|
Error: rpc.Error.New("unknown rpc error"),
|
|
Audit: audit.PieceAuditOffline,
|
|
},
|
|
{
|
|
Error: rpc.Error.Wrap(rpcstatus.Error(rpcstatus.InvalidArgument, "rpc wrapped rpcstatus invalid arg error")),
|
|
Audit: audit.PieceAuditOffline,
|
|
},
|
|
{
|
|
Error: rpc.Error.Wrap(rpcstatus.Error(rpcstatus.NotFound, "rpc wrapped rpcstatus not found error")),
|
|
// TODO: should not this be failure?
|
|
Audit: audit.PieceAuditOffline,
|
|
},
|
|
{
|
|
Error: rpcstatus.Error(rpcstatus.NotFound, "rpcstatus not found error"),
|
|
Audit: audit.PieceAuditFailure,
|
|
},
|
|
{
|
|
Error: context.DeadlineExceeded,
|
|
Audit: audit.PieceAuditContained,
|
|
},
|
|
{
|
|
Error: rpcstatus.Error(rpcstatus.DeadlineExceeded, "deadline exceeded rpcstatus error"),
|
|
Audit: audit.PieceAuditContained,
|
|
},
|
|
{
|
|
Error: errs.New("unknown error"),
|
|
Audit: audit.PieceAuditUnknown,
|
|
},
|
|
{
|
|
Error: rpcstatus.Error(rpcstatus.Unknown, "unknown rpcstatus error"),
|
|
Audit: audit.PieceAuditUnknown,
|
|
},
|
|
}
|
|
for _, c := range cases {
|
|
pieceAudit := audit.PieceAuditFromErr(c.Error)
|
|
require.Equal(t, c.Audit, pieceAudit, c.Error)
|
|
}
|
|
}
|