add CreateEgress to test server

livekit · Oct 18, 2023 · 5c0beef · 5c0beef
1 parent 5fd61d4
commit 5c0beef
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/pkg/service/ioclient.go b/pkg/service/ioclient.go
@@ -27,6 +27,11 @@ func NewIOClient(nodeID string, bus psrpc.MessageBus) (rpc.IOInfoClient, error)
 	}, nil
 }
 
+func (c *IOClient) CreateEgress(ctx context.Context, info *livekit.EgressInfo, opts ...psrpc.RequestOption) (*emptypb.Empty, error) {
+	// TODO: add retries, log errors
+	return c.IOInfoClient.CreateEgress(ctx, info, opts...)
+}
+
 func (c *IOClient) UpdateEgress(ctx context.Context, info *livekit.EgressInfo, opts ...psrpc.RequestOption) (*emptypb.Empty, error) {
 	requestType, outputType := egress.GetTypes(info.Request)
 	switch info.Status {
@@ -52,5 +57,5 @@ func (c *IOClient) UpdateEgress(ctx context.Context, info *livekit.EgressInfo, o
 	}
 
 	// TODO: add retries, log errors
-	return c.IOInfoClient.UpdateEgress(ctx, info)
+	return c.IOInfoClient.UpdateEgress(ctx, info, opts...)
 }
diff --git a/test/ioserver.go b/test/ioserver.go
@@ -44,7 +44,11 @@ func newIOTestServer(bus psrpc.MessageBus, updates chan *livekit.EgressInfo) (*i
 	return s, nil
 }
 
-func (s *ioTestServer) UpdateEgressInfo(_ context.Context, info *livekit.EgressInfo) (*emptypb.Empty, error) {
+func (s *ioTestServer) CreateEgress(_ context.Context, info *livekit.EgressInfo) (*emptypb.Empty, error) {
+	return &emptypb.Empty{}, nil
+}
+
+func (s *ioTestServer) UpdateEgress(_ context.Context, info *livekit.EgressInfo) (*emptypb.Empty, error) {
 	s.updates <- info
 	return &emptypb.Empty{}, nil
 }