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

     1  //
     2  // Copyright (C) 2020 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  	"github.com/stretchr/testify/assert"
    13  )
    14  
    15  func TestNewDeviceResponse(t *testing.T) {
    16  	expectedRequestId := "123456"
    17  	expectedStatusCode := 200
    18  	expectedMessage := "unit test message"
    19  	expectedDevice := dtos.Device{Name: "test device"}
    20  	actual := NewDeviceResponse(expectedRequestId, expectedMessage, expectedStatusCode, expectedDevice)
    21  
    22  	assert.Equal(t, expectedRequestId, actual.RequestId)
    23  	assert.Equal(t, expectedStatusCode, actual.StatusCode)
    24  	assert.Equal(t, expectedMessage, actual.Message)
    25  	assert.Equal(t, expectedDevice, actual.Device)
    26  }
    27  
    28  func TestNewMultiDevicesResponse(t *testing.T) {
    29  	expectedRequestId := "123456"
    30  	expectedStatusCode := 200
    31  	expectedMessage := "unit test message"
    32  	expectedDevices := []dtos.Device{
    33  		{Name: "test device1"},
    34  		{Name: "test device2"},
    35  	}
    36  	expectedTotalCount := uint32(2)
    37  	actual := NewMultiDevicesResponse(expectedRequestId, expectedMessage, expectedStatusCode, expectedTotalCount, expectedDevices)
    38  
    39  	assert.Equal(t, expectedRequestId, actual.RequestId)
    40  	assert.Equal(t, expectedStatusCode, actual.StatusCode)
    41  	assert.Equal(t, expectedMessage, actual.Message)
    42  	assert.Equal(t, expectedTotalCount, actual.TotalCount)
    43  	assert.Equal(t, expectedDevices, actual.Devices)
    44  }