frostfs-api-go/object/attributes_test.go
Evgenii Stratonikov a85146250b [#68] Makefile: Use gofumpt for formatting
Signed-off-by: Evgenii Stratonikov <e.stratonikov@yadro.com>
2024-03-14 10:31:58 +03:00

89 lines
1.7 KiB
Go

package object
import (
"strconv"
"testing"
"github.com/stretchr/testify/require"
)
func TestSetNotification(t *testing.T) {
o := new(Object)
ni := NotificationInfo{
epoch: 10,
topic: "test",
}
WriteNotificationInfo(o, ni)
var foundEpoch, foundTopic bool
for _, attr := range o.GetHeader().GetAttributes() {
switch key := attr.GetKey(); key {
case SysAttributeTickEpoch:
require.Equal(t, false, foundEpoch)
uEpoch, err := strconv.ParseUint(attr.GetValue(), 10, 64)
require.NoError(t, err)
require.Equal(t, ni.Epoch(), uEpoch)
foundEpoch = true
case SysAttributeTickTopic:
require.Equal(t, false, foundTopic)
require.Equal(t, ni.Topic(), attr.GetValue())
foundTopic = true
}
}
require.Equal(t, true, foundEpoch && foundTopic)
}
func TestGetNotification(t *testing.T) {
o := new(Object)
attr := []Attribute{
{SysAttributeTickEpoch, "10"},
{SysAttributeTickTopic, "test"},
}
h := new(Header)
h.SetAttributes(attr)
o.SetHeader(h)
t.Run("No error", func(t *testing.T) {
ni, err := GetNotificationInfo(o)
require.NoError(t, err)
require.Equal(t, uint64(10), ni.Epoch())
require.Equal(t, "test", ni.Topic())
})
}
func TestIntegration(t *testing.T) {
o := new(Object)
var (
ni1 = NotificationInfo{
epoch: 10,
topic: "",
}
ni2 = NotificationInfo{
epoch: 11,
topic: "test",
}
)
WriteNotificationInfo(o, ni1)
WriteNotificationInfo(o, ni2)
t.Run("double set", func(t *testing.T) {
ni, err := GetNotificationInfo(o)
require.NoError(t, err)
require.Equal(t, ni2.epoch, ni.Epoch())
require.Equal(t, ni2.topic, ni.Topic())
require.Equal(t, 2, len(o.GetHeader().GetAttributes()))
})
}