update testing framework

berops · Nov 14, 2024 · 13b7a47 · 13b7a47
1 parent 3832a64
commit 13b7a47
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/services/testing-framework/claudie_test.go b/services/testing-framework/claudie_test.go
@@ -122,7 +122,7 @@ func testClaudie(ctx context.Context) error {
 
 			log.Info().Msgf("Starting test set: %s", path)
 			err, cleanup := processTestSet(ctx, path, false, manager, testLonghornDeployment)
-			if err == nil || errors.Is(err, errInterrupt) {
+			if err == nil || errors.Is(err, errInterrupt) || errors.Is(err, context.Canceled) {
 				if err := cleanup(); err != nil {
 					log.Err(err).Msgf("Error in cleaning up test set %s", path)
 				}
@@ -141,7 +141,7 @@ func testClaudie(ctx context.Context) error {
 
 			log.Info().Msgf("Starting test set: %s", path)
 			err, cleanup := processTestSet(ctx, path, true, manager, testLonghornDeployment)
-			if err == nil || errors.Is(err, errInterrupt) {
+			if err == nil || errors.Is(err, errInterrupt) || errors.Is(err, context.Canceled) {
 				if err := cleanup(); err != nil {
 					log.Err(err).Msgf("Error in cleaning up test set %s", path)
 				}
@@ -173,7 +173,7 @@ func testClaudie(ctx context.Context) error {
 						return testAutoscaler(ctx, c)
 					},
 				)
-				if err == nil || errors.Is(err, errInterrupt) {
+				if err == nil || errors.Is(err, errInterrupt) || errors.Is(err, context.Canceled) {
 					if err := cleanup(); err != nil {
 						log.Err(err).Msgf("Error in cleaning up test set %s", path)
 					}