generated from PenaSide/GolangTemplate
113 lines
2.9 KiB
Go
113 lines
2.9 KiB
Go
package transfer_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"penahub.gitlab.yandexcloud.net/pena-services/customer/internal/models"
|
|
"penahub.gitlab.yandexcloud.net/pena-services/customer/internal/proto/broker"
|
|
"penahub.gitlab.yandexcloud.net/pena-services/customer/internal/utils/transfer"
|
|
)
|
|
|
|
func TestPrivilegeModelToProto(t *testing.T) {
|
|
t.Run("Успешный перевод модели привелегии в прото", func(t *testing.T) {
|
|
assert.Equal(t,
|
|
&broker.PrivilegeMessage{
|
|
PrivilegeID: "12",
|
|
ServiceKey: "key-1",
|
|
Type: "type",
|
|
Value: "12",
|
|
},
|
|
transfer.PrivilegeModelToProto(&models.Privilege{
|
|
ID: "1",
|
|
Name: "name",
|
|
PrivilegeID: "12",
|
|
ServiceKey: "key-1",
|
|
Description: "description",
|
|
Type: "type",
|
|
Value: "12",
|
|
Price: 20,
|
|
}),
|
|
)
|
|
})
|
|
|
|
t.Run("Успешный перевод nil модели привелегии в прото", func(t *testing.T) {
|
|
assert.Equal(t, &broker.PrivilegeMessage{}, transfer.PrivilegeModelToProto(nil))
|
|
})
|
|
}
|
|
|
|
func TestPrivilegeArrayModelToProto(t *testing.T) {
|
|
t.Run("Успешный перевод массива модели привелегий в массив прото", func(t *testing.T) {
|
|
assert.Equal(t,
|
|
[]*broker.PrivilegeMessage{
|
|
{
|
|
PrivilegeID: "12",
|
|
ServiceKey: "key-1",
|
|
Type: "type",
|
|
Value: "12",
|
|
},
|
|
},
|
|
transfer.PrivilegeArrayModelToProto([]models.Privilege{
|
|
{
|
|
ID: "1",
|
|
Name: "name",
|
|
PrivilegeID: "12",
|
|
ServiceKey: "key-1",
|
|
Description: "description",
|
|
Type: "type",
|
|
Value: "12",
|
|
Price: 20,
|
|
},
|
|
}),
|
|
)
|
|
})
|
|
}
|
|
|
|
func TestPrivilegeProtoToModel(t *testing.T) {
|
|
t.Run("Успешный перевод прото привелегии в модель", func(t *testing.T) {
|
|
assert.Equal(t,
|
|
&models.Privilege{
|
|
PrivilegeID: "12",
|
|
ServiceKey: "key-1",
|
|
Type: "type",
|
|
Value: "12",
|
|
},
|
|
transfer.PrivilegeProtoToModel(&broker.PrivilegeMessage{
|
|
PrivilegeID: "12",
|
|
ServiceKey: "key-1",
|
|
Type: "type",
|
|
Value: "12",
|
|
}),
|
|
)
|
|
})
|
|
|
|
t.Run("Успешный перевод nil прото привелегии в модель", func(t *testing.T) {
|
|
assert.Equal(t, &models.Privilege{}, transfer.PrivilegeProtoToModel(nil))
|
|
})
|
|
}
|
|
|
|
func TestPrivilegeArrayProtoToModel(t *testing.T) {
|
|
t.Run("Успешный перевод массива прото привелегий в массив моделей", func(t *testing.T) {
|
|
assert.Equal(t,
|
|
[]models.Privilege{
|
|
{
|
|
PrivilegeID: "12",
|
|
ServiceKey: "key-1",
|
|
Type: "type",
|
|
Value: "12",
|
|
},
|
|
{},
|
|
},
|
|
transfer.PrivilegeArrayProtoToModel([]*broker.PrivilegeMessage{
|
|
{
|
|
PrivilegeID: "12",
|
|
ServiceKey: "key-1",
|
|
Type: "type",
|
|
Value: "12",
|
|
},
|
|
nil,
|
|
}),
|
|
)
|
|
})
|
|
}
|