Add unit test

cloudtrust · Jan 10, 2019 · 454f5ac · 454f5ac
1 parent 9e06ec9
commit 454f5ac
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/pkg/event/module_test.go b/pkg/event/module_test.go
@@ -39,7 +39,7 @@ func TestConsoleModule(t *testing.T) {
 	assert.Nil(t, err)
 }
 
-func TestStatisticsModule(t *testing.T) {
+func TestStatisticsModuleWithMissingValues(t *testing.T) {
 	var mockCtrl = gomock.NewController(t)
 	defer mockCtrl.Finish()
 	var mockInflux = mock.NewInflux(mockCtrl)
@@ -54,3 +54,19 @@ func TestStatisticsModule(t *testing.T) {
 	var err = statisticModule.Stats(context.Background(), map[string]interface{}{"key": "val"})
 	assert.Nil(t, err)
 }
+
+func TestStatisticsModule(t *testing.T) {
+	var mockCtrl = gomock.NewController(t)
+	defer mockCtrl.Finish()
+	var mockInflux = mock.NewInflux(mockCtrl)
+
+	var batchPointsConfig = influx.BatchPointsConfig{
+		Precision: "s",
+		Database:  "db",
+	}
+
+	var statisticModule = NewStatisticModule(mockInflux, batchPointsConfig)
+	mockInflux.EXPECT().Write(gomock.Any()).Return(nil).Times(1)
+	var err = statisticModule.Stats(context.Background(), map[string]interface{}{"type": "val", "realmId": "realm", "userId": "user"})
+	assert.Nil(t, err)
+}