Fix tests

Author: RobertIndie

pkg/ctl/topic/stats_test.go

@@ -128,6 +128,15 @@ func TestGetPerPartitionedStatsCmd(t *testing.T) {
 		t.Fatal(err)
 	}
 
+	assert.Greater(t, stats.TopicCreationTimeStamp, int64(0))
+	stats.TopicCreationTimeStamp = 0
+	partitionKey := "persistent://public/default/test-topic-per-partitioned-stats-partition-0"
+	assert.Contains(t, stats.Partitions, partitionKey)
+	assert.Greaterf(t, stats.Partitions[partitionKey].TopicCreationTimeStamp, int64(0), "TopicCreationTimeStamp should be greater than 0")
+	v := stats.Partitions[partitionKey]
+	v.TopicCreationTimeStamp = 0
+	stats.Partitions[partitionKey] = v
+
 	defaultStats := utils.PartitionedTopicStats{
 		MsgRateIn:           0,
 		MsgRateOut:          0,
@@ -141,7 +150,7 @@ func TestGetPerPartitionedStatsCmd(t *testing.T) {
 		DeDuplicationStatus: "",
 		Metadata:            utils.PartitionedTopicMetadata{Partitions: 1},
 		Partitions: map[string]utils.TopicStats{
-			"persistent://public/default/test-topic-per-partitioned-stats-partition-0": {
+			partitionKey: {
 				MsgRateIn:           0,
 				MsgRateOut:          0,
 				MsgThroughputIn:     0,
@@ -216,6 +225,8 @@ func TestGetStatsWithoutSubscriptionBacklogSize(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
+	assert.Greaterf(t, stats.TopicCreationTimeStamp, int64(0), "TopicCreationTimeStamp should be greater than 0")
+	stats.TopicCreationTimeStamp = 0
 
 	assert.Equal(t, defaultStats, stats)
 }
@@ -234,6 +245,8 @@ func TestGetStatsWithEarliestTimeInBacklog(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
+	assert.Greaterf(t, stats.TopicCreationTimeStamp, int64(0), "TopicCreationTimeStamp should be greater than 0")
+	stats.TopicCreationTimeStamp = 0
 
 	assert.Equal(t, defaultStats, stats)
 }
@@ -252,6 +265,8 @@ func TestGetStatsWithMultipleNewFlags(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
+	assert.Greaterf(t, stats.TopicCreationTimeStamp, int64(0), "TopicCreationTimeStamp should be greater than 0")
+	stats.TopicCreationTimeStamp = 0
 
 	assert.Equal(t, defaultStats, stats)
 }
@@ -300,6 +315,15 @@ func TestGetPerPartitionStatsWithNewFlags(t *testing.T) {
 		t.Fatal(err)
 	}
 
+	assert.Greater(t, stats.TopicCreationTimeStamp, int64(0))
+	stats.TopicCreationTimeStamp = 0
+	partitionKey := "persistent://public/default/test-per-part-stats-new-flags-partition-0"
+	assert.Contains(t, stats.Partitions, partitionKey)
+	assert.Greaterf(t, stats.Partitions[partitionKey].TopicCreationTimeStamp, int64(0), "TopicCreationTimeStamp should be greater than 0")
+	v := stats.Partitions[partitionKey]
+	v.TopicCreationTimeStamp = 0
+	stats.Partitions[partitionKey] = v
+
 	defaultStats := utils.PartitionedTopicStats{
 		MsgRateIn:           0,
 		MsgRateOut:          0,
@@ -313,7 +337,7 @@ func TestGetPerPartitionStatsWithNewFlags(t *testing.T) {
 		DeDuplicationStatus: "",
 		Metadata:            utils.PartitionedTopicMetadata{Partitions: 1},
 		Partitions: map[string]utils.TopicStats{
-			"persistent://public/default/test-per-part-stats-new-flags-partition-0": {
+			partitionKey: {
 				MsgRateIn:           0,
 				MsgRateOut:          0,
 				MsgThroughputIn:     0,