@@ -63,6 +63,8 @@ var _ = Describe("TLS Config", func() {
6363 Expect (feast .isOpenShiftTls (OfflineFeastType )).To (BeFalse ())
6464 Expect (feast .isOpenShiftTls (OnlineFeastType )).To (BeFalse ())
6565 Expect (feast .isOpenShiftTls (RegistryFeastType )).To (BeFalse ())
66+ Expect (feast .isOpenShiftTls (UIFeastType )).To (BeFalse ())
67+
6668 openshiftTls , err := feast .checkOpenshiftTls ()
6769 Expect (err ).ToNot (HaveOccurred ())
6870 Expect (openshiftTls ).To (BeFalse ())
@@ -79,6 +81,9 @@ var _ = Describe("TLS Config", func() {
7981 tls = feast .getTlsConfigs (OnlineFeastType )
8082 Expect (tls ).To (BeNil ())
8183 Expect (tls .IsTLS ()).To (BeFalse ())
84+ tls = feast .getTlsConfigs (UIFeastType )
85+ Expect (tls ).To (BeNil ())
86+ Expect (tls .IsTLS ()).To (BeFalse ())
8287 tls = feast .getTlsConfigs (RegistryFeastType )
8388 Expect (tls ).NotTo (BeNil ())
8489 Expect (tls .IsTLS ()).To (BeTrue ())
@@ -90,7 +95,9 @@ var _ = Describe("TLS Config", func() {
9095 Expect (feast .localRegistryTls ()).To (BeTrue ())
9196 Expect (feast .isOpenShiftTls (OfflineFeastType )).To (BeFalse ())
9297 Expect (feast .isOpenShiftTls (OnlineFeastType )).To (BeFalse ())
98+ Expect (feast .isOpenShiftTls (UIFeastType )).To (BeFalse ())
9399 Expect (feast .isOpenShiftTls (RegistryFeastType )).To (BeTrue ())
100+
94101 openshiftTls , err = feast .checkOpenshiftTls ()
95102 Expect (err ).ToNot (HaveOccurred ())
96103 Expect (openshiftTls ).To (BeTrue ())
@@ -124,12 +131,19 @@ var _ = Describe("TLS Config", func() {
124131 Expect (tls .SecretRef .Name ).To (Equal ("feast-test-registry-tls" ))
125132 Expect (tls .SecretKeyNames ).To (Equal (secretKeyNames ))
126133 Expect (tls .IsTLS ()).To (BeTrue ())
134+ tls = feast .getTlsConfigs (UIFeastType )
135+ Expect (tls ).NotTo (BeNil ())
136+ Expect (tls .SecretRef ).NotTo (BeNil ())
137+ Expect (tls .SecretRef .Name ).To (Equal ("feast-test-ui-tls" ))
138+ Expect (tls .SecretKeyNames ).To (Equal (secretKeyNames ))
139+ Expect (tls .IsTLS ()).To (BeTrue ())
127140
128141 Expect (feast .remoteRegistryTls ()).To (BeFalse ())
129142 Expect (feast .localRegistryTls ()).To (BeTrue ())
130143 Expect (feast .isOpenShiftTls (OfflineFeastType )).To (BeTrue ())
131144 Expect (feast .isOpenShiftTls (OnlineFeastType )).To (BeTrue ())
132145 Expect (feast .isOpenShiftTls (RegistryFeastType )).To (BeTrue ())
146+ Expect (feast .isOpenShiftTls (UIFeastType )).To (BeTrue ())
133147 openshiftTls , err = feast .checkOpenshiftTls ()
134148 Expect (err ).ToNot (HaveOccurred ())
135149 Expect (openshiftTls ).To (BeTrue ())
@@ -139,18 +153,22 @@ var _ = Describe("TLS Config", func() {
139153 err = feast .setDeployment (feastDeploy )
140154 Expect (err ).ToNot (HaveOccurred ())
141155 Expect (feastDeploy .Spec .Template .Spec .InitContainers ).To (HaveLen (1 ))
142- Expect (feastDeploy .Spec .Template .Spec .Containers ).To (HaveLen (3 ))
156+ Expect (feastDeploy .Spec .Template .Spec .Containers ).To (HaveLen (4 ))
143157 Expect (feastDeploy .Spec .Template .Spec .Containers [0 ].Command ).To (ContainElements (ContainSubstring ("--key" )))
144158 Expect (feastDeploy .Spec .Template .Spec .Containers [1 ].Command ).To (ContainElements (ContainSubstring ("--key" )))
145159 Expect (feastDeploy .Spec .Template .Spec .Containers [2 ].Command ).To (ContainElements (ContainSubstring ("--key" )))
146- Expect (feastDeploy .Spec .Template .Spec .Volumes ).To (HaveLen (4 ))
160+ Expect (feastDeploy .Spec .Template .Spec .Containers [3 ].Command ).To (ContainElements (ContainSubstring ("--key" )))
161+ Expect (feastDeploy .Spec .Template .Spec .Volumes ).To (HaveLen (5 ))
147162
148163 // registry service w/ tls and in an openshift cluster
149164 feast .Handler .FeatureStore = minimalFeatureStore ()
150165 feast .Handler .FeatureStore .Spec .Services = & feastdevv1alpha1.FeatureStoreServices {
151166 OnlineStore : & feastdevv1alpha1.OnlineStore {
152167 TLS : & feastdevv1alpha1.TlsConfigs {},
153168 },
169+ UI : & feastdevv1alpha1.UIService {
170+ TLS : & feastdevv1alpha1.TlsConfigs {},
171+ },
154172 Registry : & feastdevv1alpha1.Registry {
155173 Local : & feastdevv1alpha1.LocalRegistryConfig {
156174 TLS : & feastdevv1alpha1.TlsConfigs {
@@ -171,17 +189,20 @@ var _ = Describe("TLS Config", func() {
171189 tls = feast .getTlsConfigs (OnlineFeastType )
172190 Expect (tls ).NotTo (BeNil ())
173191 Expect (tls .IsTLS ()).To (BeFalse ())
192+ tls = feast .getTlsConfigs (UIFeastType )
193+ Expect (tls ).NotTo (BeNil ())
194+ Expect (tls .IsTLS ()).To (BeFalse ())
174195 tls = feast .getTlsConfigs (RegistryFeastType )
175196 Expect (tls ).NotTo (BeNil ())
176197 Expect (tls .IsTLS ()).To (BeTrue ())
177198 Expect (tls .SecretKeyNames ).NotTo (Equal (secretKeyNames ))
178199 Expect (getPortStr (tls )).To (Equal ("443" ))
179200 Expect (GetTlsPath (RegistryFeastType )).To (Equal ("/tls/registry/" ))
180-
181201 Expect (feast .remoteRegistryTls ()).To (BeFalse ())
182202 Expect (feast .localRegistryTls ()).To (BeTrue ())
183203 Expect (feast .isOpenShiftTls (OfflineFeastType )).To (BeFalse ())
184204 Expect (feast .isOpenShiftTls (OnlineFeastType )).To (BeFalse ())
205+ Expect (feast .isOpenShiftTls (UIFeastType )).To (BeFalse ())
185206 Expect (feast .isOpenShiftTls (RegistryFeastType )).To (BeFalse ())
186207 openshiftTls , err = feast .checkOpenshiftTls ()
187208 Expect (err ).ToNot (HaveOccurred ())
@@ -193,6 +214,9 @@ var _ = Describe("TLS Config", func() {
193214 feast .Handler .FeatureStore .Spec .Services .OnlineStore .TLS = & feastdevv1alpha1.TlsConfigs {
194215 Disable : & disable ,
195216 }
217+ feast .Handler .FeatureStore .Spec .Services .UI .TLS = & feastdevv1alpha1.TlsConfigs {
218+ Disable : & disable ,
219+ }
196220 feast .Handler .FeatureStore .Spec .Services .Registry = & feastdevv1alpha1.Registry {
197221 Local : & feastdevv1alpha1.LocalRegistryConfig {
198222 TLS : & feastdevv1alpha1.TlsConfigs {
@@ -219,6 +243,10 @@ var _ = Describe("TLS Config", func() {
219243 Expect (tls ).NotTo (BeNil ())
220244 Expect (tls .IsTLS ()).To (BeFalse ())
221245 Expect (tls .SecretKeyNames ).NotTo (Equal (secretKeyNames ))
246+ tls = feast .getTlsConfigs (UIFeastType )
247+ Expect (tls ).NotTo (BeNil ())
248+ Expect (tls .IsTLS ()).To (BeFalse ())
249+ Expect (tls .SecretKeyNames ).NotTo (Equal (secretKeyNames ))
222250 tls = feast .getTlsConfigs (RegistryFeastType )
223251 Expect (tls ).NotTo (BeNil ())
224252 Expect (tls .IsTLS ()).To (BeFalse ())
@@ -230,6 +258,7 @@ var _ = Describe("TLS Config", func() {
230258 Expect (feast .localRegistryTls ()).To (BeFalse ())
231259 Expect (feast .isOpenShiftTls (OfflineFeastType )).To (BeTrue ())
232260 Expect (feast .isOpenShiftTls (OnlineFeastType )).To (BeFalse ())
261+ Expect (feast .isOpenShiftTls (UIFeastType )).To (BeFalse ())
233262 Expect (feast .isOpenShiftTls (RegistryFeastType )).To (BeFalse ())
234263 openshiftTls , err = feast .checkOpenshiftTls ()
235264 Expect (err ).ToNot (HaveOccurred ())
@@ -249,11 +278,17 @@ var _ = Describe("TLS Config", func() {
249278 Expect (onlineSvc .Annotations ).To (BeEmpty ())
250279 Expect (onlineSvc .Spec .Ports [0 ].Name ).To (Equal (HttpScheme ))
251280
281+ uiSvc := feast .initFeastSvc (UIFeastType )
282+ err = feast .setService (uiSvc , UIFeastType )
283+ Expect (err ).ToNot (HaveOccurred ())
284+ Expect (uiSvc .Annotations ).To (BeEmpty ())
285+ Expect (uiSvc .Spec .Ports [0 ].Name ).To (Equal (HttpScheme ))
286+
252287 // check k8s deployment objects
253288 feastDeploy = feast .initFeastDeploy ()
254289 err = feast .setDeployment (feastDeploy )
255290 Expect (err ).ToNot (HaveOccurred ())
256- Expect (feastDeploy .Spec .Template .Spec .Containers ).To (HaveLen (3 ))
291+ Expect (feastDeploy .Spec .Template .Spec .Containers ).To (HaveLen (4 ))
257292 Expect (GetOfflineContainer (* feastDeploy )).NotTo (BeNil ())
258293 Expect (feastDeploy .Spec .Template .Spec .Volumes ).To (HaveLen (2 ))
259294
@@ -263,6 +298,9 @@ var _ = Describe("TLS Config", func() {
263298 Expect (GetOfflineContainer (* feastDeploy ).VolumeMounts ).To (HaveLen (2 ))
264299 Expect (GetOnlineContainer (* feastDeploy ).Command ).NotTo (ContainElements (ContainSubstring ("--key" )))
265300 Expect (GetOnlineContainer (* feastDeploy ).VolumeMounts ).To (HaveLen (1 ))
301+ Expect (GetUIContainer (* feastDeploy ).Command ).NotTo (ContainElements (ContainSubstring ("--key" )))
302+ Expect (GetUIContainer (* feastDeploy ).VolumeMounts ).To (HaveLen (1 ))
303+
266304 })
267305 })
268306})
0 commit comments