close the server via context

Signed-off-by: EshaanAgg <[email protected]>

internal/provider/file/file.go

@@ -151,8 +151,8 @@ func (p *Provider) Start(ctx context.Context) error {
 
 			switch event.Op {
 			case fsnotify.Create, fsnotify.Write, fsnotify.Remove:
-				// Since we do not watch any events in the subdirectories, any events involving files
-				// modifications in current directory will trigger the event handling.
+				// Since we do not watch any events related to creation/deletion of subdirectories,
+				// any events involving files & folder modifications should trigger the event handling.
 				goto handle
 			default:
 				// do nothing
@@ -209,4 +209,5 @@ func (p *Provider) startHealthProbeServer(ctx context.Context, readyzChecker hea
 	if err := srv.ListenAndServe(); err != nil {
 		p.logger.Error(err, "failed to start health probe server")
 	}
+
 }

internal/provider/file/file_test.go

@@ -87,8 +87,9 @@ func TestFileProvider(t *testing.T) {
 	require.NoError(t, err)
 
 	// Start file provider.
+	ctx, cancelFpContext := context.WithCancel(context.Background())
 	go func() {
-		if err := fp.Start(context.Background()); err != nil {
+		if err := fp.Start(ctx); err != nil {
 			t.Errorf("failed to start file provider: %v", err)
 		}
 	}()
@@ -166,6 +167,7 @@ func TestFileProvider(t *testing.T) {
 	})
 
 	t.Cleanup(func() {
+		cancelFpContext()
 		err := os.RemoveAll(watchFileBase)
 		require.NoError(t, err)
 		err = os.RemoveAll(watchDirPath)
@@ -214,8 +216,9 @@ func TestRecursiveFileProvider(t *testing.T) {
 	require.NoError(t, err)
 
 	// Start file provider.
+	ctx, cancelFpContext := context.WithCancel(context.Background())
 	go func() {
-		if err := fp.Start(context.Background()); err != nil {
+		if err := fp.Start(ctx); err != nil {
 			t.Errorf("failed to start file provider: %v", err)
 		}
 	}()
@@ -281,6 +284,7 @@ func TestRecursiveFileProvider(t *testing.T) {
 	})
 
 	t.Cleanup(func() {
+		cancelFpContext()
 		err := os.RemoveAll(baseDir)
 		require.NoError(t, err)
 		err = os.RemoveAll(unwatchedDir)