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

     1  //
     2  // Copyright (C) 2020-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 TestNewSubscriptionResponse(t *testing.T) {
    17  	expectedRequestId := "123456"
    18  	expectedStatusCode := 200
    19  	expectedMessage := "unit test message"
    20  	expectedSubscription := dtos.Subscription{Name: "test Subscription"}
    21  	actual := NewSubscriptionResponse(expectedRequestId, expectedMessage, expectedStatusCode, expectedSubscription)
    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, expectedSubscription, actual.Subscription)
    27  }
    28  
    29  func TestNewMultiSubscriptionsResponse(t *testing.T) {
    30  	expectedRequestId := "123456"
    31  	expectedStatusCode := 200
    32  	expectedMessage := "unit test message"
    33  	expectedSubscriptions := []dtos.Subscription{
    34  		{Name: "test Subscription1"},
    35  		{Name: "test Subscription2"},
    36  	}
    37  	expectedTotalCount := uint32(len(expectedSubscriptions))
    38  	actual := NewMultiSubscriptionsResponse(expectedRequestId, expectedMessage, expectedStatusCode, expectedTotalCount, expectedSubscriptions)
    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, expectedSubscriptions, actual.Subscriptions)
    45  }