From 0b1aed255ac941b9a90ace0080142f8aea783a0d Mon Sep 17 00:00:00 2001
From: amirylm <amirylm.dev@gmail.com>
Date: Mon, 27 Nov 2023 17:34:56 -0300
Subject: [PATCH] minor fixes

---
 commons/net_test.go | 6 ++++++
 core/ctrl_test.go   | 2 ++
 2 files changed, 8 insertions(+)

diff --git a/commons/net_test.go b/commons/net_test.go
index a84c841..1d6018b 100644
--- a/commons/net_test.go
+++ b/commons/net_test.go
@@ -17,4 +17,10 @@ func TestGetOrGeneratePrivateKey(t *testing.T) {
 	require.NotNil(t, sk2)
 	require.Equal(t, skb64, sk2b64)
 	require.True(t, sk.Equals(sk2))
+
+	t.Run("bad input", func(t *testing.T) {
+		sk, _, err := GetOrGeneratePrivateKey("bad input")
+		require.Error(t, err)
+		require.Nil(t, sk)
+	})
 }
diff --git a/core/ctrl_test.go b/core/ctrl_test.go
index 895d971..e4ae079 100644
--- a/core/ctrl_test.go
+++ b/core/ctrl_test.go
@@ -75,6 +75,8 @@ func TestController_Sanity(t *testing.T) {
 
 	for topic, counter := range msgHitMap {
 		count := int(counter.Load()) / n // per node
+		// add 1 to account for the first message sent by the node
+		count += 1
 		require.GreaterOrEqual(t, count, rounds, "should get %d messages on topic %s", rounds, topic)
 	}
 }