feat(standalone): Polishing

Author: BeegiiK

pkg/controllers/daemon/standalone/controller.go

@@ -56,7 +56,7 @@ func New(config *kcfg.StandaloneConfig, cache *standalone.Cache, metricsModule *
 		cache:         cache,
 		config:        config,
 		metricsModule: metricsModule,
-		l:             log.Logger().Named(string("Controller")),
+		l:             log.Logger().Named(string("RetinaEndpointController")),
 	}, nil
 }
 
@@ -104,7 +104,7 @@ func (c *Controller) Reconcile(ctx context.Context) error {
 
 // Run starts the controller loop
 func (c *Controller) Run(ctx context.Context) {
-	c.l.Info("Starting controller")
+	c.l.Info("Starting RetinaEndpoint controller")
 
 	ticker := time.NewTicker(c.config.MetricsInterval)
 	defer ticker.Stop()
@@ -124,7 +124,7 @@ func (c *Controller) Run(ctx context.Context) {
 
 // Stop stops the controller and cleans up resources
 func (c *Controller) Stop() {
-	c.l.Info("Stopping controller")
+	c.l.Info("Stopping RetinaEndpoint controller")
 	c.cache.Clear()
 	c.metricsModule.Clear()
 }

pkg/controllers/daemon/standalone/controller_test.go

@@ -46,14 +46,17 @@ func TestControllerReconcile(t *testing.T) {
 	newEndpoint := common.NewRetinaEndpoint("new-pod", "default", &common.IPAddresses{IPv4: net.ParseIP("1.1.1.1")})
 	mockSource.EXPECT().GetAllEndpoints().Return([]*common.RetinaEndpoint{newEndpoint}, nil)
 
+	// Need valid file path to initialize statefile source
+	testMockStatefilePath := "./source/statefile/azure/azure-vnet-mock.json"
+
 	// Setup test controller with invalid config to test error handling
-	invalidCfg := &kcfg.StandaloneConfig{MetricsInterval: time.Second, EnrichmentMode: "gcp-statefile"}
+	invalidCfg := &kcfg.StandaloneConfig{MetricsInterval: time.Second, EnrichmentMode: "gcp-statefile", StateFileLocation: testMockStatefilePath}
 	controller, err := New(invalidCfg, cache, metricsModule)
 	require.Error(t, err)
 	require.Nil(t, controller)
 
 	// Setup test controller with valid config
-	cfg := &kcfg.StandaloneConfig{MetricsInterval: time.Second, EnrichmentMode: "azure-vnet-statefile"}
+	cfg := &kcfg.StandaloneConfig{MetricsInterval: time.Second, EnrichmentMode: "azure-vnet-statefile", StateFileLocation: testMockStatefilePath}
 	controller, err = New(cfg, cache, metricsModule)
 	require.NoError(t, err)
 	require.NotNil(t, controller)