Cleanup flag

onflow · Mar 13, 2024 · 1b16948 · 1b16948
1 parent 4a6cd2c
commit 1b16948
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/internal/super/generate_test.go b/internal/super/generate_test.go
@@ -79,6 +79,10 @@ func TestGenerateNewContractSkipTests(t *testing.T) {
 
 	generateFlags.SkipTests = true
 
+	t.Cleanup(func() {
+		generateFlags.SkipTests = false
+	})
+
 	// Test contract generation
 	_, err := generateNew([]string{"TestContract"}, "contract", logger, state)
 	assert.NoError(t, err, "Failed to generate contract")