@@ -185,7 +185,67 @@ var _ = Describe("RayJob", Ordered, func() {
185
185
})
186
186
})
187
187
188
- // TODO: add tests for RayServices
188
+ var _ = Describe ("RayService" , Ordered , func () {
189
+ It ("Create RayService" , func () {
190
+ _ , err := rayClient .RayServices ("default" ).Create (context .Background (), & rayv1.RayService {
191
+ ObjectMeta : metav1.ObjectMeta {Name : "proxy-test-service" },
192
+ Spec : rayv1.RayServiceSpec {
193
+ RayClusterSpec : rayv1.RayClusterSpec {
194
+ HeadGroupSpec : rayv1.HeadGroupSpec {
195
+ RayStartParams : make (map [string ]string ),
196
+ Template : corev1.PodTemplateSpec {
197
+ Spec : corev1.PodSpec {
198
+ Containers : []corev1.Container {
199
+ {
200
+ Name : "test" ,
201
+ Image : "test" ,
202
+ },
203
+ },
204
+ },
205
+ },
206
+ },
207
+ },
208
+ },
209
+ }, metav1.CreateOptions {})
210
+ Expect (err ).ToNot (HaveOccurred ())
211
+ Expect (lastReq .Load ().Method ).To (Equal (http .MethodPost ))
212
+ Expect (lastReq .Load ().RequestURI ).To (Equal ("/apis/ray.io/v1/namespaces/default/rayservices" ))
213
+ })
214
+ It ("Get RayService" , func () {
215
+ service , err := rayClient .RayServices ("default" ).Get (context .Background (), "proxy-test-service" , metav1.GetOptions {})
216
+ Expect (err ).ToNot (HaveOccurred ())
217
+ Expect (service .Name ).To (Equal ("proxy-test-service" ))
218
+ Expect (lastReq .Load ().Method ).To (Equal (http .MethodGet ))
219
+ Expect (lastReq .Load ().RequestURI ).To (Equal ("/apis/ray.io/v1/namespaces/default/rayservices/proxy-test-service" ))
220
+ })
221
+ It ("Update RayService" , func () {
222
+ service , err := rayClient .RayServices ("default" ).Get (context .Background (), "proxy-test-service" , metav1.GetOptions {})
223
+ Expect (err ).ToNot (HaveOccurred ())
224
+ service .Spec .RayClusterSpec .HeadGroupSpec .Template .Spec .Containers [0 ].Image = "new-image"
225
+ _ , err = rayClient .RayServices ("default" ).Update (context .Background (), service , metav1.UpdateOptions {})
226
+ Expect (err ).ToNot (HaveOccurred ())
227
+ Expect (lastReq .Load ().Method ).To (Equal (http .MethodPut ))
228
+ Expect (lastReq .Load ().RequestURI ).To (Equal ("/apis/ray.io/v1/namespaces/default/rayservices/proxy-test-service" ))
229
+
230
+ updatedSvc , err := rayClient .RayServices ("default" ).Get (context .Background (), "proxy-test-service" , metav1.GetOptions {})
231
+ Expect (err ).ToNot (HaveOccurred ())
232
+ Expect (updatedSvc .Spec .RayClusterSpec .HeadGroupSpec .Template .Spec .Containers [0 ].Image ).To (Equal ("new-image" ))
233
+ })
234
+ It ("List RayService" , func () {
235
+ services , err := rayClient .RayServices ("default" ).List (context .Background (), metav1.ListOptions {})
236
+ Expect (err ).ToNot (HaveOccurred ())
237
+ Expect (services .Items ).To (HaveLen (1 ))
238
+ Expect (services .Items [0 ].Name ).To (Equal ("proxy-test-service" ))
239
+ Expect (lastReq .Load ().Method ).To (Equal (http .MethodGet ))
240
+ Expect (lastReq .Load ().RequestURI ).To (Equal ("/apis/ray.io/v1/namespaces/default/rayservices" ))
241
+ })
242
+ It ("Delete RayService" , func () {
243
+ err := rayClient .RayServices ("default" ).Delete (context .Background (), "proxy-test-service" , metav1.DeleteOptions {})
244
+ Expect (err ).ToNot (HaveOccurred ())
245
+ Expect (lastReq .Load ().Method ).To (Equal (http .MethodDelete ))
246
+ Expect (lastReq .Load ().RequestURI ).To (Equal ("/apis/ray.io/v1/namespaces/default/rayservices/proxy-test-service" ))
247
+ })
248
+ })
189
249
190
250
var _ = Describe ("events" , Ordered , func () {
191
251
It ("List events" , func () {
0 commit comments