81 lines
2.0 KiB
Go
81 lines
2.0 KiB
Go
|
// Copyright (C) 2023 Storj Labs, Inc.
|
||
|
|
// See LICENSE for copying information.
|
||
|
|
|
||
|
|
package pieces
|
||
|
|
|
||
|
|
import (
|
||
|
|
"crypto/sha256"
|
||
|
|
"encoding/hex"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/spacemonkeygo/monkit/v3"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
"go.uber.org/zap/zaptest"
|
||
|
|
|
||
|
|
"storj.io/common/testcontext"
|
||
|
|
"storj.io/storj/storagenode/blobstore"
|
||
|
|
"storj.io/storj/storagenode/blobstore/filestore"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestMonitoredBlobWriter(t *testing.T) {
|
||
|
|
|
||
|
|
dir, err := filestore.NewDir(zaptest.NewLogger(t), t.TempDir())
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
blobs := filestore.New(zaptest.NewLogger(t), dir, filestore.Config{})
|
||
|
|
defer func() { require.NoError(t, blobs.Close()) }()
|
||
|
|
|
||
|
|
ctx := testcontext.New(t)
|
||
|
|
f1, err := blobs.Create(ctx, blobstore.BlobRef{
|
||
|
|
Namespace: []byte("test"),
|
||
|
|
Key: []byte("key"),
|
||
|
|
}, 10)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
statName := "test_monitored_blob_writer"
|
||
|
|
m1 := MonitorBlobWriter(statName, f1)
|
||
|
|
|
||
|
|
_, err = m1.Write([]byte("01234"))
|
||
|
|
require.NoError(t, err)
|
||
|
|
_, err = m1.Write([]byte("56789"))
|
||
|
|
require.NoError(t, err)
|
||
|
|
err = m1.Commit(ctx)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
found := false
|
||
|
|
mon.Stats(func(key monkit.SeriesKey, field string, val float64) {
|
||
|
|
if key.Measurement == statName && field == "count" {
|
||
|
|
found = true
|
||
|
|
require.Equal(t, float64(1), val)
|
||
|
|
require.Equal(t, "small", key.Tags.Get("size"))
|
||
|
|
}
|
||
|
|
})
|
||
|
|
require.True(t, found)
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestMonitoredHash(t *testing.T) {
|
||
|
|
|
||
|
|
statName := "test_monitored_hash"
|
||
|
|
m1 := MonitorHash(statName, sha256.New())
|
||
|
|
|
||
|
|
_, err := m1.Write([]byte("01234"))
|
||
|
|
require.NoError(t, err)
|
||
|
|
_, err = m1.Write([]byte("56789"))
|
||
|
|
require.NoError(t, err)
|
||
|
|
hash := m1.Sum([]byte{})
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, "84d89877f0d4041efb6bf91a16f0248f2fd573e6af05c19f96bedb9f882f7882", hex.EncodeToString(hash))
|
||
|
|
|
||
|
|
found := false
|
||
|
|
mon.Stats(func(key monkit.SeriesKey, field string, val float64) {
|
||
|
|
if key.Measurement == statName && field == "count" {
|
||
|
|
found = true
|
||
|
|
require.Equal(t, float64(1), val)
|
||
|
|
require.Equal(t, "small", key.Tags.Get("size"))
|
||
|
|
}
|
||
|
|
})
|
||
|
|
require.True(t, found)
|
||
|
|
|
||
|
|
}
|