mk6i-open-oscar-server/server/oscar/handler/oservice_test.go

package handler

import (
	"bytes"
	"context"
	"log/slog"
	"testing"

	"github.com/mk6i/retro-aim-server/server/oscar/middleware"
	"github.com/mk6i/retro-aim-server/state"
	"github.com/mk6i/retro-aim-server/wire"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/mock"
)

func TestOServiceBOSHandler_ClientOnline(t *testing.T) {
	input := wire.SNACMessage{
		Frame: wire.SNACFrame{
			FoodGroup: wire.OService,
			SubGroup:  wire.OServiceClientOnline,
		},
		Body: wire.SNAC_0x01_0x02_OServiceClientOnline{
			GroupVersions: []struct {
				FoodGroup   uint16
				Version     uint16
				ToolID      uint16
				ToolVersion uint16
			}{
				{
					FoodGroup: 10,
				},
			},
		},
	}

	svc := newMockOServiceService(t)
	svc.EXPECT().
		ClientOnline(mock.Anything, input.Body, mock.Anything).
		Return(nil)

	h := NewOServiceHandler(slog.Default(), svc)

	responseWriter := newMockResponseWriter(t)

	buf := &bytes.Buffer{}
	assert.NoError(t, wire.MarshalBE(input.Body, buf))

	assert.NoError(t, h.ClientOnline(nil, nil, input.Frame, buf, responseWriter))
}

func TestOServiceBOSHandler_ServiceRequest(t *testing.T) {
	input := wire.SNACMessage{
		Frame: wire.SNACFrame{
			FoodGroup: wire.OService,
			SubGroup:  wire.OServiceServiceRequest,
		},
		Body: wire.SNAC_0x01_0x04_OServiceServiceRequest{
			FoodGroup: wire.Chat,
		},
	}
	output := wire.SNACMessage{
		Frame: wire.SNACFrame{
			FoodGroup: wire.OService,
			SubGroup:  wire.OServiceServiceResponse,
		},
		Body: wire.SNAC_0x01_0x05_OServiceServiceResponse{
			TLVRestBlock: wire.TLVRestBlock{
				TLVList: wire.TLVList{
					wire.NewTLVBE(0x01, uint16(1000)),
				},
			},
		},
	}

	svc := newMockOServiceService(t)
	svc.EXPECT().
		ServiceRequest(mock.Anything, mock.Anything, input.Frame, input.Body).
		Return(output, nil)

	h := NewOServiceHandler(slog.Default(), svc)

	responseWriter := newMockResponseWriter(t)
	responseWriter.EXPECT().
		SendSNAC(output.Frame, output.Body).
		Return(nil)

	buf := &bytes.Buffer{}
	assert.NoError(t, wire.MarshalBE(input.Body, buf))

	assert.NoError(t, h.ServiceRequest(nil, nil, input.Frame, buf, responseWriter))
}

func TestOServiceChatHandler_ClientOnline(t *testing.T) {
	input := wire.SNACMessage{
		Frame: wire.SNACFrame{
			FoodGroup: wire.OService,
			SubGroup:  wire.OServiceClientOnline,
		},
		Body: wire.SNAC_0x01_0x02_OServiceClientOnline{
			GroupVersions: []struct {
				FoodGroup   uint16
				Version     uint16
				ToolID      uint16
				ToolVersion uint16
			}{
				{
					FoodGroup: 10,
				},
			},
		},
	}

	svc := newMockOServiceService(t)
	svc.EXPECT().
		ClientOnline(mock.Anything, mock.Anything, mock.Anything).
		Return(nil)

	h := NewOServiceHandler(slog.Default(), svc)

	responseWriter := newMockResponseWriter(t)

	buf := &bytes.Buffer{}
	assert.NoError(t, wire.MarshalBE(input.Body, buf))

	assert.NoError(t, h.ClientOnline(nil, nil, input.Frame, buf, responseWriter))
}

func TestOServiceHandler_IdleNotification(t *testing.T) {
	input := wire.SNACMessage{
		Frame: wire.SNACFrame{
			FoodGroup: wire.OService,
			SubGroup:  wire.OServiceIdleNotification,
		},
		Body: wire.SNAC_0x01_0x11_OServiceIdleNotification{
			IdleTime: 10,
		},
	}

	svc := newMockOServiceService(t)
	svc.EXPECT().
		IdleNotification(mock.Anything, mock.Anything, input.Body).
		Return(nil)

	h := OServiceHandler{
		OServiceService: svc,
		RouteLogger: middleware.RouteLogger{
			Logger: slog.Default(),
		},
	}

	responseWriter := newMockResponseWriter(t)

	buf := &bytes.Buffer{}
	assert.NoError(t, wire.MarshalBE(input.Body, buf))

	assert.NoError(t, h.IdleNotification(nil, nil, input.Frame, buf, responseWriter))
}

func TestOServiceHandler_ClientVersions(t *testing.T) {
	input := wire.SNACMessage{
		Frame: wire.SNACFrame{
			FoodGroup: wire.OService,
			SubGroup:  wire.OServiceClientVersions,
		},
		Body: wire.SNAC_0x01_0x17_OServiceClientVersions{
			Versions: []uint16{
				10,
			},
		},
	}
	output := wire.SNACMessage{
		Frame: wire.SNACFrame{
			FoodGroup: wire.OService,
			SubGroup:  wire.OServiceHostVersions,
		},
		Body: wire.SNAC_0x01_0x18_OServiceHostVersions{
			Versions: []uint16{
				10,
			},
		},
	}

	sess := state.NewSession()
	svc := newMockOServiceService(t)
	svc.EXPECT().
		ClientVersions(mock.Anything, sess, input.Frame, input.Body).
		Return(output)

	h := OServiceHandler{
		OServiceService: svc,
		RouteLogger: middleware.RouteLogger{
			Logger: slog.Default(),
		},
	}

	responseWriter := newMockResponseWriter(t)
	responseWriter.EXPECT().
		SendSNAC(output.Frame, output.Body).
		Return(nil)

	buf := &bytes.Buffer{}
	assert.NoError(t, wire.MarshalBE(input.Body, buf))

	assert.NoError(t, h.ClientVersions(context.Background(), sess, input.Frame, buf, responseWriter))
}

func TestOServiceHandler_RateParamsQuery(t *testing.T) {
	input := wire.SNACMessage{
		Frame: wire.SNACFrame{
			FoodGroup: wire.OService,
			SubGroup:  wire.OServiceRateParamsQuery,
		},
		Body: struct{}{},
	}
	output := wire.SNACMessage{
		Frame: wire.SNACFrame{
			FoodGroup: wire.OService,
			SubGroup:  wire.OServiceRateParamsReply,
		},
		Body: wire.SNAC_0x01_0x07_OServiceRateParamsReply{
			RateGroups: []struct {
				ID    uint16
				Pairs []struct {
					FoodGroup uint16
					SubGroup  uint16
				} `oscar:"count_prefix=uint16"`
			}{
				{
					ID: 1,
				},
			},
		},
	}

	svc := newMockOServiceService(t)
	svc.EXPECT().
		RateParamsQuery(mock.Anything, mock.Anything, input.Frame).
		Return(output)

	h := OServiceHandler{
		OServiceService: svc,
		RouteLogger: middleware.RouteLogger{
			Logger: slog.Default(),
		},
	}

	responseWriter := newMockResponseWriter(t)
	responseWriter.EXPECT().
		SendSNAC(output.Frame, output.Body).
		Return(nil)

	buf := &bytes.Buffer{}
	assert.NoError(t, wire.MarshalBE(input.Body, buf))

	assert.NoError(t, h.RateParamsQuery(nil, nil, input.Frame, buf, responseWriter))
}

func TestOServiceHandler_RateParamsSubAdd(t *testing.T) {
	input := wire.SNACMessage{
		Frame: wire.SNACFrame{
			FoodGroup: wire.OService,
			SubGroup:  wire.OServiceRateParamsSubAdd,
		},
		Body: wire.SNAC_0x01_0x08_OServiceRateParamsSubAdd{
			ClassIDs: []uint16{1, 2, 3, 4},
		},
	}

	sess := state.NewSession()

	svc := newMockOServiceService(t)
	svc.EXPECT().
		RateParamsSubAdd(mock.Anything, sess, input.Body)

	h := OServiceHandler{
		OServiceService: svc,
		RouteLogger: middleware.RouteLogger{
			Logger: slog.Default(),
		},
	}

	buf := &bytes.Buffer{}
	assert.NoError(t, wire.MarshalBE(input.Body, buf))

	assert.NoError(t, h.RateParamsSubAdd(context.Background(), sess, input.Frame, buf, nil))
}

func TestOServiceHandler_SetUserInfoFields(t *testing.T) {
	input := wire.SNACMessage{
		Frame: wire.SNACFrame{
			FoodGroup: wire.OService,
			SubGroup:  wire.OServiceSetUserInfoFields,
		},
		Body: wire.SNAC_0x01_0x1E_OServiceSetUserInfoFields{
			TLVRestBlock: wire.TLVRestBlock{
				TLVList: wire.TLVList{
					wire.NewTLVBE(0x01, []byte{1, 2, 3, 4}),
				},
			},
		},
	}
	output := wire.SNACMessage{
		Frame: wire.SNACFrame{
			FoodGroup: wire.OService,
			SubGroup:  wire.OServiceUserInfoUpdate,
		},
		Body: wire.SNAC_0x01_0x0F_OServiceUserInfoUpdate{
			UserInfo: []wire.TLVUserInfo{
				{ScreenName: "screen-name"},
				{ScreenName: "screen-name"},
			},
		},
	}

	svc := newMockOServiceService(t)
	svc.EXPECT().
		SetUserInfoFields(mock.Anything, mock.Anything, input.Frame, input.Body).
		Return(output, nil)

	h := OServiceHandler{
		OServiceService: svc,
		RouteLogger: middleware.RouteLogger{
			Logger: slog.Default(),
		},
	}

	responseWriter := newMockResponseWriter(t)
	responseWriter.EXPECT().
		SendSNAC(output.Frame, output.Body).
		Return(nil)

	buf := &bytes.Buffer{}
	assert.NoError(t, wire.MarshalBE(input.Body, buf))

	assert.NoError(t, h.SetUserInfoFields(nil, nil, input.Frame, buf, responseWriter))
}

func TestOServiceHandler_UserInfoQuery(t *testing.T) {
	input := wire.SNACMessage{
		Frame: wire.SNACFrame{
			FoodGroup: wire.OService,
			SubGroup:  wire.OServiceUserInfoQuery,
		},
		Body: struct{}{},
	}
	output := wire.SNACMessage{
		Frame: wire.SNACFrame{
			FoodGroup: wire.OService,
			SubGroup:  wire.OServiceUserInfoUpdate,
		},
		Body: wire.SNAC_0x01_0x0F_OServiceUserInfoUpdate{
			UserInfo: []wire.TLVUserInfo{
				{ScreenName: "screen-name"},
				{ScreenName: "screen-name"},
			},
		},
	}

	svc := newMockOServiceService(t)
	svc.EXPECT().
		UserInfoQuery(mock.Anything, mock.Anything, input.Frame).
		Return(output)

	h := OServiceHandler{
		OServiceService: svc,
		RouteLogger: middleware.RouteLogger{
			Logger: slog.Default(),
		},
	}

	responseWriter := newMockResponseWriter(t)
	responseWriter.EXPECT().
		SendSNAC(output.Frame, output.Body).
		Return(nil)

	buf := &bytes.Buffer{}
	assert.NoError(t, wire.MarshalBE(input.Body, buf))

	assert.NoError(t, h.UserInfoQuery(nil, nil, input.Frame, buf, responseWriter))
}

func TestOServiceHandler_Noop(t *testing.T) {
	input := wire.SNACMessage{
		Frame: wire.SNACFrame{
			FoodGroup: wire.OService,
			SubGroup:  wire.OServiceNoop,
		},
		Body: struct{}{},
	}

	h := OServiceHandler{
		RouteLogger: middleware.RouteLogger{
			Logger: slog.Default(),
		},
	}

	responseWriter := newMockResponseWriter(t)
	buf := &bytes.Buffer{}
	assert.NoError(t, wire.MarshalBE(input.Body, buf))

	assert.NoError(t, h.Noop(nil, nil, input.Frame, buf, responseWriter))
}

func TestOServiceHandler_SetPrivacyFlags(t *testing.T) {
	input := wire.SNACMessage{
		Frame: wire.SNACFrame{
			FoodGroup: wire.OService,
			SubGroup:  wire.OServiceSetPrivacyFlags,
		},
		Body: wire.SNAC_0x01_0x14_OServiceSetPrivacyFlags{
			PrivacyFlags: wire.OServicePrivacyFlagMember,
		},
	}

	svc := newMockOServiceService(t)
	svc.EXPECT().
		SetPrivacyFlags(mock.Anything, input.Body)

	h := OServiceHandler{
		OServiceService: svc,
		RouteLogger: middleware.RouteLogger{
			Logger: slog.Default(),
		},
	}

	responseWriter := newMockResponseWriter(t)

	buf := &bytes.Buffer{}
	assert.NoError(t, wire.MarshalBE(input.Body, buf))

	assert.NoError(t, h.SetPrivacyFlags(nil, nil, input.Frame, buf, responseWriter))
}

func TestOServiceChatNavHandler_ClientOnline(t *testing.T) {
	input := wire.SNACMessage{
		Frame: wire.SNACFrame{
			FoodGroup: wire.OService,
			SubGroup:  wire.OServiceClientOnline,
		},
		Body: wire.SNAC_0x01_0x02_OServiceClientOnline{
			GroupVersions: []struct {
				FoodGroup   uint16
				Version     uint16
				ToolID      uint16
				ToolVersion uint16
			}{
				{
					FoodGroup: 10,
				},
			},
		},
	}

	svc := newMockOServiceService(t)
	svc.EXPECT().
		ClientOnline(mock.Anything, input.Body, mock.Anything).
		Return(nil)
	h := NewOServiceHandler(slog.Default(), svc)

	responseWriter := newMockResponseWriter(t)

	buf := &bytes.Buffer{}
	assert.NoError(t, wire.MarshalBE(input.Body, buf))

	assert.NoError(t, h.ClientOnline(nil, nil, input.Frame, buf, responseWriter))
}

func TestOServiceAlertHandler_ClientOnline(t *testing.T) {
	input := wire.SNACMessage{
		Frame: wire.SNACFrame{
			FoodGroup: wire.OService,
			SubGroup:  wire.OServiceClientOnline,
		},
		Body: wire.SNAC_0x01_0x02_OServiceClientOnline{
			GroupVersions: []struct {
				FoodGroup   uint16
				Version     uint16
				ToolID      uint16
				ToolVersion uint16
			}{
				{
					FoodGroup: 10,
				},
			},
		},
	}

	svc := newMockOServiceService(t)
	svc.EXPECT().
		ClientOnline(mock.Anything, input.Body, mock.Anything).
		Return(nil)
	h := NewOServiceHandler(slog.Default(), svc)

	responseWriter := newMockResponseWriter(t)

	buf := &bytes.Buffer{}
	assert.NoError(t, wire.MarshalBE(input.Body, buf))

	assert.NoError(t, h.ClientOnline(nil, nil, input.Frame, buf, responseWriter))
}