Fix Integration Test Cleanup

Signed-off-by: Mahad Zaryab <[email protected]>
jaegertracing · Jan 18, 2025 · 6936344 · 6936344
1 parent 5a06df6
commit 6936344
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/plugin/storage/integration/cassandra_test.go b/plugin/storage/integration/cassandra_test.go
@@ -42,6 +42,7 @@ func newCassandraStorageIntegration() *CassandraStorageIntegration {
 
 func (s *CassandraStorageIntegration) cleanUp(t *testing.T) {
 	require.NoError(t, s.factory.Purge(context.Background()))
+	require.NoError(t, s.archiveFactory.Purge(context.Background()))
 }
 
 func (*CassandraStorageIntegration) initializeCassandraFactory(t *testing.T, flags []string, factoryInit func() *cassandra.Factory) *cassandra.Factory {

diff --git a/plugin/storage/integration/elasticsearch_test.go b/plugin/storage/integration/elasticsearch_test.go
@@ -101,6 +101,7 @@ func (s *ESStorageIntegration) initializeES(t *testing.T, c *http.Client, allTag
 
 func (s *ESStorageIntegration) esCleanUp(t *testing.T) {
 	require.NoError(t, s.factory.Purge(context.Background()))
+	require.NoError(t, s.archiveFactory.Purge(context.Background()))
 }
 
 func (*ESStorageIntegration) initializeESFactory(t *testing.T, args []string, factoryInit func() *es.Factory) *es.Factory {

diff --git a/plugin/storage/integration/grpc_test.go b/plugin/storage/integration/grpc_test.go
@@ -62,6 +62,7 @@ func (s *GRPCStorageIntegrationTestSuite) initialize(t *testing.T) {
 
 func (s *GRPCStorageIntegrationTestSuite) close(t *testing.T) {
 	require.NoError(t, s.factory.Close())
+	require.NoError(t, s.archiveFactory.Close())
 	s.remoteStorage.Close(t)
 }