Skip to content

Commit fb2e4d6

Browse files
authored
[apiserversdk] add RayService envtests (#3904)
Signed-off-by: win5923 <[email protected]>
1 parent f6b4f17 commit fb2e4d6

File tree

1 file changed

+61
-1
lines changed

1 file changed

+61
-1
lines changed

apiserversdk/proxy_test.go

Lines changed: 61 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -185,7 +185,67 @@ var _ = Describe("RayJob", Ordered, func() {
185185
})
186186
})
187187

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+
})
189249

190250
var _ = Describe("events", Ordered, func() {
191251
It("List events", func() {

0 commit comments

Comments
 (0)