dev.azure.com/aidainnovazione0090/DeviceManager/_git/go-mod-core-contracts@v1.0.2/dtos/responses/notification_test.go (about)

     1  //
     2  // Copyright (C) 2021 IOTech Ltd
     3  //
     4  // SPDX-License-Identifier: Apache-2.0
     5  
     6  package responses
     7  
     8  import (
     9  	"testing"
    10  
    11  	"dev.azure.com/aidainnovazione0090/DeviceManager/_git/go-mod-core-contracts/dtos"
    12  
    13  	"github.com/stretchr/testify/assert"
    14  )
    15  
    16  func TestNewNotificationResponse(t *testing.T) {
    17  	expectedRequestId := "123456"
    18  	expectedStatusCode := 200
    19  	expectedMessage := "unit test message"
    20  	expectedNotification := dtos.Notification{}
    21  	actual := NewNotificationResponse(expectedRequestId, expectedMessage, expectedStatusCode, expectedNotification)
    22  
    23  	assert.Equal(t, expectedRequestId, actual.RequestId)
    24  	assert.Equal(t, expectedStatusCode, actual.StatusCode)
    25  	assert.Equal(t, expectedMessage, actual.Message)
    26  	assert.Equal(t, expectedNotification, actual.Notification)
    27  }
    28  
    29  func TestNewMultiNotificationsResponse(t *testing.T) {
    30  	expectedRequestId := "123456"
    31  	expectedStatusCode := 200
    32  	expectedMessage := "unit test message"
    33  	expectedNotifications := []dtos.Notification{
    34  		{Id: "abc"},
    35  		{Id: "def"},
    36  	}
    37  	expectedTotalCount := uint32(len(expectedNotifications))
    38  	actual := NewMultiNotificationsResponse(expectedRequestId, expectedMessage, expectedStatusCode, expectedTotalCount, expectedNotifications)
    39  
    40  	assert.Equal(t, expectedRequestId, actual.RequestId)
    41  	assert.Equal(t, expectedStatusCode, actual.StatusCode)
    42  	assert.Equal(t, expectedMessage, actual.Message)
    43  	assert.Equal(t, expectedTotalCount, actual.TotalCount)
    44  	assert.Equal(t, expectedNotifications, actual.Notifications)
    45  }