test: Extend test to assert rest-api container args

ntkathole · ntkathole · commit d42d9e5cac65 · 2025-05-22T09:59:39.000+05:30
Signed-off-by: ntkathole &lt;nikhilkathole2683@gmail.com&gt;
diff --git a/infra/feast-operator/internal/controller/featurestore_controller_test.go b/infra/feast-operator/internal/controller/featurestore_controller_test.go
@@ -1256,6 +1256,29 @@ var _ = Describe("FeatureStore Controller", func() {
 			err = k8sClient.Get(ctx, nsName, resource)
 			Expect(err).NotTo(HaveOccurred())
 			Expect(resource.Status.Applied.Services.Registry.Local.Server.RestAPI).To(BeTrue())
+			By("checking the registry container args include --rest-api")
+			feast := services.FeastServices{
+				Handler: handler.FeastHandler{
+					Client:       controllerReconciler.Client,
+					Context:      ctx,
+					Scheme:       controllerReconciler.Scheme,
+					FeatureStore: resource,
+				},
+			}
+
+			// check deployment
+			deploy := &appsv1.Deployment{}
+			objMeta := feast.GetObjectMeta()
+			err = k8sClient.Get(ctx, types.NamespacedName{
+				Name:      objMeta.Name,
+				Namespace: objMeta.Namespace,
+			}, deploy)
+			Expect(err).NotTo(HaveOccurred())
+
+			registryContainer := services.GetRegistryContainer(*deploy)
+			Expect(registryContainer).NotTo(BeNil())
+			Expect(registryContainer.Command).To(ContainElement("--rest-api"),
+				"expected --rest-api to be present in registry container args: %v", registryContainer.Args)
 		})
 
 		It("should error on reconcile", func() {
