dev.azure.com/aidainnovazione0090/DeviceManager/_git/go-mod-core-contracts@v1.0.2/dtos/responses/transmission_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 TestNewTransmissionResponse(t *testing.T) {
    17  	expectedRequestId := "123456"
    18  	expectedStatusCode := 200
    19  	expectedMessage := "unit test message"
    20  	expectedTransmission := dtos.Transmission{}
    21  	actual := NewTransmissionResponse(expectedRequestId, expectedMessage, expectedStatusCode, expectedTransmission)
    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, expectedTransmission, actual.Transmission)
    27  }
    28  
    29  func TestNewMultiTransmissionsResponse(t *testing.T) {
    30  	expectedRequestId := "123456"
    31  	expectedStatusCode := 200
    32  	expectedMessage := "unit test message"
    33  	expectedTransmissions := []dtos.Transmission{
    34  		{Id: "abc"},
    35  		{Id: "def"},
    36  	}
    37  	expectedTotalCount := uint32(len(expectedTransmissions))
    38  	actual := NewMultiTransmissionsResponse(expectedRequestId, expectedMessage, expectedStatusCode, expectedTotalCount, expectedTransmissions)
    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, expectedTransmissions, actual.Transmissions)
    45  }