From 5c0beef30ff250cc3c4e780afbfdd1d2fc933ecd Mon Sep 17 00:00:00 2001 From: David Colburn Date: Wed, 18 Oct 2023 02:07:20 -0700 Subject: [PATCH] add CreateEgress to test server --- pkg/service/ioclient.go | 7 ++++++- test/ioserver.go | 6 +++++- 2 files changed, 11 insertions(+), 2 deletions(-) diff --git a/pkg/service/ioclient.go b/pkg/service/ioclient.go index 5c07334f..fc07de73 100644 --- 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 index bb50382f..ef2a3517 100644 --- 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 }