tests: topic from bytes

pactus-project · Jan 18, 2025 · b646b3b · b646b3b
1 parent db6d2a8
commit b646b3b
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/www/zmq/topic.go b/www/zmq/topic.go
@@ -42,5 +42,5 @@ func TopicFromBytes(b []byte) Topic {
 		return 0
 	}
 
-	return Topic(binary.BigEndian.Uint16(b[:2]))
+	return Topic(binary.BigEndian.Uint16(b))
 }
diff --git a/www/zmq/topic_test.go b/www/zmq/topic_test.go
@@ -0,0 +1,18 @@
+package zmq
+
+import (
+	"github.com/stretchr/testify/require"
+	"testing"
+)
+
+func TestTopicFromBytes(t *testing.T) {
+	validRawTopic := TopicRawTransaction.Bytes()
+	invalidRawTopic := make([]byte, 0)
+
+	topic := TopicFromBytes(validRawTopic)
+	require.Equal(t, TopicRawTransaction, topic)
+
+	topic = TopicFromBytes(invalidRawTopic)
+	require.Equal(t, 0, int(topic))
+
+}