From 1384523f620ffb59184e9eabc382bd1a6596b24c Mon Sep 17 00:00:00 2001
From: Pavel Karpy <carpawell@nspcc.ru>
Date: Wed, 16 Feb 2022 17:53:46 +0300
Subject: [PATCH] [#372] object: Add object notification unit tests

Signed-off-by: Pavel Karpy <carpawell@nspcc.ru>
---
 object/attributes_test.go | 89 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 89 insertions(+)
 create mode 100644 object/attributes_test.go

diff --git a/object/attributes_test.go b/object/attributes_test.go
new file mode 100644
index 0000000..8556f13
--- /dev/null
+++ b/object/attributes_test.go
@@ -0,0 +1,89 @@
+package object
+
+import (
+	"strconv"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestSetNotification(t *testing.T) {
+	o := new(Object)
+
+	var 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()))
+	})
+}