minor fixes

amirylm · Nov 27, 2023 · 0b1aed2 · 0b1aed2
1 parent f6c7152
commit 0b1aed2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git 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
@@ -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)
 	}
 }