update test

nginx · Nov 6, 2024 · 15e6b8a · 15e6b8a
1 parent dbdfa1e
commit 15e6b8a
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/internal/file/file_manager_service_test.go b/internal/file/file_manager_service_test.go
@@ -59,12 +59,13 @@ func TestFileManagerService_UpdateFile(t *testing.T) {
 		},
 	}
 
+	tempDir := os.TempDir()
+
 	for _, test := range tests {
 		ctx := context.Background()
 
-		tempDir := os.TempDir()
 		testFile := helpers.CreateFileWithErrorCheck(t, tempDir, "nginx.conf")
-		defer helpers.RemoveFileWithErrorCheck(t, testFile.Name())
+
 		var fileMeta *mpi.FileMeta
 		if test.isCert {
 			fileMeta = protos.CertMeta(testFile.Name(), "")
@@ -80,6 +81,8 @@ func TestFileManagerService_UpdateFile(t *testing.T) {
 
 		require.NoError(t, err)
 		assert.Equal(t, 1, fakeFileServiceClient.UpdateFileCallCount())
+
+		helpers.RemoveFileWithErrorCheck(t, testFile.Name())
 	}
 }