fixed unit tests

Signed-off-by: Rahul Jadhav <nyrahul@gmail.com>
accuknox · Jan 24, 2022 · 300e32c · 300e32c
1 parent 4ff63ef
commit 300e32c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/.gitignore b/.gitignore
@@ -7,5 +7,5 @@ policies/*.yaml
 !policies/.keep
 tests/mysql/logs
 src/dist/
-src/coverage.out
-src/report.json
+src/coverage*.out
+src/report*.json
diff --git a/src/libs/dbHandler_test.go b/src/libs/dbHandler_test.go
@@ -48,7 +48,7 @@ func TestGetNetworkPolicies(t *testing.T) {
 	mock.ExpectQuery("^SELECT (.+) FROM network_policy*").
 		WillReturnRows(rows)
 
-	results := GetNetworkPolicies(types.ConfigDB{DBDriver: "mysql", TableNetworkPolicy: "network_policy"}, "", "", "")
+	results := GetNetworkPolicies(types.ConfigDB{DBDriver: "mysql"}, "", "", "")
 	assert.Equal(t, results[0].Kind, "test")
 
 	if err := mock.ExpectationsWereMet(); err != nil {
@@ -91,7 +91,7 @@ func TestInsertNetworkPolicies(t *testing.T) {
 		},
 	}
 
-	err := InsertNetworkPoliciesToMySQL(types.ConfigDB{DBDriver: "mysql", TableNetworkPolicy: "network_policy"}, nfe)
+	err := InsertNetworkPoliciesToMySQL(types.ConfigDB{DBDriver: "mysql"}, nfe)
 	assert.NoError(t, err)
 
 	if err := mock.ExpectationsWereMet(); err != nil {