test(grant): cleanup grants table after each test in grant repository…

… test
raystack · Oct 9, 2023 · 3e541da · 3e541da
1 parent 6330380
commit 3e541da
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/internal/store/postgres/grant_repository_test.go b/internal/store/postgres/grant_repository_test.go
@@ -103,6 +103,17 @@ func (s *GrantRepositoryTestSuite) SetupSuite() {
 	s.Require().NoError(err)
 }
 
+func (s *GrantRepositoryTestSuite) AfterTest(suiteName, testName string) {
+	// clean grants table
+	db, err := s.store.DB().DB()
+	if err != nil {
+		s.T().Fatal(err)
+	}
+	if _, err := db.Exec("DELETE FROM grants"); err != nil {
+		s.T().Fatal(err)
+	}
+}
+
 func (s *GrantRepositoryTestSuite) TearDownSuite() {
 	// Clean tests
 	db, err := s.store.DB().DB()