countstore: a bit more mem test coverage (defensive)

bluesky-social · Dec 15, 2023 · d349d3e · d349d3e
1 parent 90ffed8
commit d349d3e
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/automod/countstore/countstore_test.go b/automod/countstore/countstore_test.go
@@ -20,9 +20,12 @@ func TestMemCountStoreBasics(t *testing.T) {
 	assert.Equal(0, c)
 	assert.NoError(cs.Increment(ctx, "test1", "val1"))
 	assert.NoError(cs.Increment(ctx, "test1", "val1"))
-	c, err = cs.GetCount(ctx, "test1", "val1", PeriodTotal)
-	assert.NoError(err)
-	assert.Equal(2, c)
+
+	for _, period := range []string{PeriodTotal, PeriodDay, PeriodHour} {
+		c, err = cs.GetCount(ctx, "test1", "val1", period)
+		assert.NoError(err)
+		assert.Equal(2, c)
+	}
 
 	c, err = cs.GetCountDistinct(ctx, "test2", "val2", PeriodTotal)
 	assert.NoError(err)
@@ -36,9 +39,12 @@ func TestMemCountStoreBasics(t *testing.T) {
 
 	assert.NoError(cs.IncrementDistinct(ctx, "test2", "val2", "two"))
 	assert.NoError(cs.IncrementDistinct(ctx, "test2", "val2", "three"))
-	c, err = cs.GetCountDistinct(ctx, "test2", "val2", PeriodTotal)
-	assert.NoError(err)
-	assert.Equal(3, c)
+
+	for _, period := range []string{PeriodTotal, PeriodDay, PeriodHour} {
+		c, err = cs.GetCountDistinct(ctx, "test2", "val2", period)
+		assert.NoError(err)
+		assert.Equal(3, c)
+	}
 }
 
 func TestMemCountStoreConcurrent(t *testing.T) {