Open-IM-Server/pkg/utils/platform_number_id_to_name_test.go

package utils

import (
	"testing"

	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/constant"

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

func Test_PlatformIDToName(t *testing.T) {
	assert.Equal(t, constant.PlatformIDToName(1), "IOS")
	assert.Equal(t, constant.PlatformIDToName(2), "Android")
	assert.Equal(t, constant.PlatformIDToName(3), "Windows")
	assert.Equal(t, constant.PlatformIDToName(4), "OSX")
	assert.Equal(t, constant.PlatformIDToName(5), "Web")
	assert.Equal(t, constant.PlatformIDToName(6), "MiniWeb")
	assert.Equal(t, constant.PlatformIDToName(7), "Linux")

	assert.Equal(t, constant.PlatformIDToName(0), "")
}

func Test_PlatformNameToID(t *testing.T) {
	assert.Equal(t, constant.PlatformNameToID("IOS"), int32(1))
	assert.Equal(t, constant.PlatformNameToID("Android"), int32(2))
	assert.Equal(t, constant.PlatformNameToID("Windows"), int32(3))
	assert.Equal(t, constant.PlatformNameToID("OSX"), int32(4))
	assert.Equal(t, constant.PlatformNameToID("Web"), int32(5))
	assert.Equal(t, constant.PlatformNameToID("MiniWeb"), int32(6))
	assert.Equal(t, constant.PlatformNameToID("Linux"), int32(7))

	assert.Equal(t, constant.PlatformNameToID("UnknownDevice"), int32(0))
	assert.Equal(t, constant.PlatformNameToID(""), int32(0))
}

func Test_PlatformNameToClass(t *testing.T) {
	assert.Equal(t, constant.PlatformNameToClass("IOS"), "Mobile")
	assert.Equal(t, constant.PlatformNameToClass("Android"), "Mobile")
	assert.Equal(t, constant.PlatformNameToClass("OSX"), "PC")
	assert.Equal(t, constant.PlatformNameToClass("Windows"), "PC")
	assert.Equal(t, constant.PlatformNameToClass("Web"), "PC")
	assert.Equal(t, constant.PlatformNameToClass("MiniWeb"), "Mobile")
	assert.Equal(t, constant.PlatformNameToClass("Linux"), "PC")

	assert.Equal(t, constant.PlatformNameToClass("UnknownDevice"), "")
	assert.Equal(t, constant.PlatformNameToClass(""), "")
}