Clean up stakeholders after Import CSV API test

Signed-off-by: Sam Lucidi <slucidi@redhat.com>
konveyor · Nov 8, 2023 · 9a5cd5b · 9a5cd5b
1 parent 57bb3f0
commit 9a5cd5b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/test/api/importcsv/api_test.go b/test/api/importcsv/api_test.go
@@ -173,13 +173,20 @@ func TestImportCSV(t *testing.T) {
 
 			// Delete imported Applications.
 			for _, apps := range gotApps {
+				if apps.Owner != nil {
+					assert.Must(t, Stakeholder.Delete(apps.Owner.ID))
+				}
+				for _, contributor := range apps.Contributors {
+					assert.Must(t, Stakeholder.Delete(contributor.ID))
+				}
 				assert.Must(t, Application.Delete(apps.ID))
 			}
 
 			// Delete imported Dependencies.
 			for _, deps := range gotDeps {
 				assert.Must(t, Dependency.Delete(deps.ID))
 			}
+
 		})
 	}
 }
diff --git a/test/api/importcsv/pkg.go b/test/api/importcsv/pkg.go
@@ -10,6 +10,7 @@ var (
 	Client      *binding.Client
 	Application binding.Application
 	Dependency  binding.Dependency
+	Stakeholder binding.Stakeholder
 )
 
 func init() {