octoplex/internal/protocol/event_test.go

package protocol_test

import (
	"errors"
	"testing"

	"git.netflux.io/rob/octoplex/internal/domain"
	"git.netflux.io/rob/octoplex/internal/event"
	pb "git.netflux.io/rob/octoplex/internal/generated/grpc"
	"git.netflux.io/rob/octoplex/internal/protocol"
	"github.com/google/go-cmp/cmp"
	gocmp "github.com/google/go-cmp/cmp"
	"github.com/stretchr/testify/assert"
	"google.golang.org/protobuf/testing/protocmp"
)

func TestEventToProto(t *testing.T) {
	testCases := []struct {
		name string
		in   event.Event
		want *pb.Event
	}{
		{
			name: "AppStateChanged",
			in: event.AppStateChangedEvent{
				State: domain.AppState{
					Source: domain.Source{
						Container: domain.Container{
							ID: "abc123",
						},
						Live: true,
					},
					Destinations: []domain.Destination{
						{
							Name: "dest1",
							URL:  "rtmp://dest1.example.com",
							Container: domain.Container{
								ID: "bcd456",
							},
						},
					},
					BuildInfo: domain.BuildInfo{GoVersion: "go1.16", Version: "v1.0.0"},
				},
			},
			want: &pb.Event{
				EventType: &pb.Event_AppStateChanged{
					AppStateChanged: &pb.AppStateChangedEvent{
						AppState: &pb.AppState{
							Source: &pb.Source{
								Container: &pb.Container{
									Id: "abc123",
								},
								Live: true,
							},
							Destinations: []*pb.Destination{
								{
									Name: "dest1",
									Url:  "rtmp://dest1.example.com",
									Container: &pb.Container{
										Id: "bcd456",
									},
								},
							},
							BuildInfo: &pb.BuildInfo{GoVersion: "go1.16", Version: "v1.0.0"},
						},
					},
				},
			},
		},
		{
			name: "DestinationAdded",
			in:   event.DestinationAddedEvent{URL: "rtmp://dest.example.com"},
			want: &pb.Event{
				EventType: &pb.Event_DestinationAdded{
					DestinationAdded: &pb.DestinationAddedEvent{
						Url: "rtmp://dest.example.com",
					},
				},
			},
		},
		{
			name: "AddDestinationFailed",
			in:   event.AddDestinationFailedEvent{URL: "rtmp://fail.example.com", Err: errors.New("failed")},
			want: &pb.Event{
				EventType: &pb.Event_AddDestinationFailed{
					AddDestinationFailed: &pb.AddDestinationFailedEvent{
						Url:   "rtmp://fail.example.com",
						Error: "failed",
					},
				},
			},
		},
		{
			name: "DestinationStreamExited",
			in:   event.DestinationStreamExitedEvent{Name: "stream1", Err: errors.New("exit reason")},
			want: &pb.Event{
				EventType: &pb.Event_DestinationStreamExited{
					DestinationStreamExited: &pb.DestinationStreamExitedEvent{
						Name:  "stream1",
						Error: "exit reason",
					},
				},
			},
		},
		{
			name: "FatalErrorOccurred",
			in:   event.FatalErrorOccurredEvent{Message: "fatal error"},
			want: &pb.Event{
				EventType: &pb.Event_FatalError{
					FatalError: &pb.FatalErrorEvent{Message: "fatal error"},
				},
			},
		},
		{
			name: "OtherInstanceDetected",
			in:   event.OtherInstanceDetectedEvent{},
			want: &pb.Event{
				EventType: &pb.Event_OtherInstanceDetected{
					OtherInstanceDetected: &pb.OtherInstanceDetectedEvent{},
				},
			},
		},
		{
			name: "MediaServerStarted",
			in:   event.MediaServerStartedEvent{RTMPURL: "rtmp://media", RTMPSURL: "rtmps://media"},
			want: &pb.Event{
				EventType: &pb.Event_MediaServerStarted{
					MediaServerStarted: &pb.MediaServerStartedEvent{
						RtmpUrl:  "rtmp://media",
						RtmpsUrl: "rtmps://media",
					},
				},
			},
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			assert.Empty(t, gocmp.Diff(tc.want, protocol.EventToProto(tc.in), protocmp.Transform()))
		})
	}
}

func TestEventFromProto(t *testing.T) {
	testCases := []struct {
		name string
		in   *pb.Event
		want event.Event
	}{
		{
			name: "AppStateChanged",
			in: &pb.Event{
				EventType: &pb.Event_AppStateChanged{
					AppStateChanged: &pb.AppStateChangedEvent{
						AppState: &pb.AppState{
							Source: &pb.Source{
								Container: &pb.Container{Id: "abc123"},
								Live:      true,
							},
							Destinations: []*pb.Destination{
								{
									Name:      "dest1",
									Url:       "rtmp://dest1.example.com",
									Container: &pb.Container{Id: "bcd456"},
								},
							},
							BuildInfo: &pb.BuildInfo{
								GoVersion: "go1.16",
								Version:   "v1.0.0",
							},
						},
					},
				},
			},
			want: event.AppStateChangedEvent{
				State: domain.AppState{
					Source: domain.Source{
						Container: domain.Container{ID: "abc123"},
						Live:      true,
					},
					Destinations: []domain.Destination{
						{
							Name:      "dest1",
							URL:       "rtmp://dest1.example.com",
							Container: domain.Container{ID: "bcd456"},
						},
					},
					BuildInfo: domain.BuildInfo{
						GoVersion: "go1.16",
						Version:   "v1.0.0",
					},
				},
			},
		},
		{
			name: "DestinationAdded",
			in: &pb.Event{
				EventType: &pb.Event_DestinationAdded{
					DestinationAdded: &pb.DestinationAddedEvent{
						Url: "rtmp://dest.example.com",
					},
				},
			},
			want: event.DestinationAddedEvent{URL: "rtmp://dest.example.com"},
		},
		{
			name: "AddDestinationFailed",
			in: &pb.Event{
				EventType: &pb.Event_AddDestinationFailed{
					AddDestinationFailed: &pb.AddDestinationFailedEvent{
						Url:   "rtmp://fail.example.com",
						Error: "failed",
					},
				},
			},
			want: event.AddDestinationFailedEvent{URL: "rtmp://fail.example.com", Err: errors.New("failed")},
		},
		{
			name: "DestinationStreamExited",
			in: &pb.Event{
				EventType: &pb.Event_DestinationStreamExited{
					DestinationStreamExited: &pb.DestinationStreamExitedEvent{
						Name:  "stream1",
						Error: "exit reason",
					},
				},
			},
			want: event.DestinationStreamExitedEvent{Name: "stream1", Err: errors.New("exit reason")},
		},
		{
			name: "FatalErrorOccurred",
			in: &pb.Event{
				EventType: &pb.Event_FatalError{
					FatalError: &pb.FatalErrorEvent{Message: "fatal error"},
				},
			},
			want: event.FatalErrorOccurredEvent{Message: "fatal error"},
		},
		{
			name: "OtherInstanceDetected",
			in: &pb.Event{
				EventType: &pb.Event_OtherInstanceDetected{
					OtherInstanceDetected: &pb.OtherInstanceDetectedEvent{},
				},
			},
			want: event.OtherInstanceDetectedEvent{},
		},
		{
			name: "MediaServerStarted",
			in: &pb.Event{
				EventType: &pb.Event_MediaServerStarted{
					MediaServerStarted: &pb.MediaServerStartedEvent{
						RtmpUrl:  "rtmp://media",
						RtmpsUrl: "rtmps://media",
					},
				},
			},
			want: event.MediaServerStartedEvent{RTMPURL: "rtmp://media", RTMPSURL: "rtmps://media"},
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			assert.Empty(t, cmp.Diff(tc.want, protocol.EventFromProto(tc.in), gocmp.Comparer(compareErrorMessages)))
		})
	}
}