forked from lorenzodonini/ocpp-go
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathupdate_firmware_test.go
137 lines (129 loc) · 8.03 KB
/
update_firmware_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
package ocpp2_test
import (
"fmt"
"time"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/mock"
"github.com/stretchr/testify/require"
"github.com/lorenzodonini/ocpp-go/ocpp2.0.1/firmware"
"github.com/lorenzodonini/ocpp-go/ocpp2.0.1/types"
)
// Test
func (suite *OcppV2TestSuite) TestUpdateFirmwareRequestValidation() {
t := suite.T()
fw := firmware.Firmware{
Location: "https://someurl",
RetrieveDateTime: types.NewDateTime(time.Now()),
InstallDateTime: types.NewDateTime(time.Now()),
SigningCertificate: "1337c0de",
Signature: "deadc0de",
}
var requestTable = []GenericTestEntry{
{firmware.UpdateFirmwareRequest{Retries: newInt(5), RetryInterval: newInt(300), RequestID: 42, Firmware: fw}, true},
{firmware.UpdateFirmwareRequest{Retries: newInt(5), RequestID: 42, Firmware: fw}, true},
{firmware.UpdateFirmwareRequest{RequestID: 42, Firmware: fw}, true},
{firmware.UpdateFirmwareRequest{Firmware: fw}, true},
{firmware.UpdateFirmwareRequest{Retries: newInt(5), RetryInterval: newInt(300), RequestID: 42, Firmware: firmware.Firmware{Location: "https://someurl", RetrieveDateTime: types.NewDateTime(time.Now())}}, true},
{firmware.UpdateFirmwareRequest{}, false},
{firmware.UpdateFirmwareRequest{Retries: newInt(-1), RetryInterval: newInt(300), RequestID: 42, Firmware: fw}, false},
{firmware.UpdateFirmwareRequest{Retries: newInt(5), RetryInterval: newInt(-1), RequestID: 42, Firmware: fw}, false},
{firmware.UpdateFirmwareRequest{Retries: newInt(5), RetryInterval: newInt(300), RequestID: -1, Firmware: fw}, false},
{firmware.UpdateFirmwareRequest{Retries: newInt(5), RetryInterval: newInt(300), RequestID: 42, Firmware: firmware.Firmware{RetrieveDateTime: types.NewDateTime(time.Now()), InstallDateTime: types.NewDateTime(time.Now()), SigningCertificate: "1337c0de", Signature: "deadc0de"}}, false},
{firmware.UpdateFirmwareRequest{Retries: newInt(5), RetryInterval: newInt(300), RequestID: 42, Firmware: firmware.Firmware{Location: "https://someurl", InstallDateTime: types.NewDateTime(time.Now()), SigningCertificate: "1337c0de", Signature: "deadc0de"}}, false},
{firmware.UpdateFirmwareRequest{Retries: newInt(5), RetryInterval: newInt(300), RequestID: 42, Firmware: firmware.Firmware{Location: ">512.............................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................................", RetrieveDateTime: types.NewDateTime(time.Now()), InstallDateTime: types.NewDateTime(time.Now()), SigningCertificate: "1337c0de", Signature: "deadc0de"}}, false},
}
ExecuteGenericTestTable(t, requestTable)
}
func (suite *OcppV2TestSuite) TestUpdateFirmwareResponseValidation() {
t := suite.T()
var responseTable = []GenericTestEntry{
{firmware.UpdateFirmwareResponse{Status: firmware.UpdateFirmwareStatusAccepted, StatusInfo: &types.StatusInfo{ReasonCode: "ok", AdditionalInfo: "someInfo"}}, true},
{firmware.UpdateFirmwareResponse{Status: firmware.UpdateFirmwareStatusAccepted, StatusInfo: &types.StatusInfo{ReasonCode: "ok"}}, true},
{firmware.UpdateFirmwareResponse{Status: firmware.UpdateFirmwareStatusAccepted}, true},
{firmware.UpdateFirmwareResponse{}, false},
{firmware.UpdateFirmwareResponse{Status: "invalidFirmwareUpdateStatus"}, false},
{firmware.UpdateFirmwareResponse{Status: firmware.UpdateFirmwareStatusAccepted, StatusInfo: &types.StatusInfo{}}, false},
}
ExecuteGenericTestTable(t, responseTable)
}
func (suite *OcppV2TestSuite) TestUpdateFirmwareE2EMocked() {
t := suite.T()
wsId := "test_id"
messageId := defaultMessageId
wsUrl := "someUrl"
fw := firmware.Firmware{
Location: "https://someurl",
RetrieveDateTime: types.NewDateTime(time.Now()),
InstallDateTime: types.NewDateTime(time.Now()),
SigningCertificate: "1337c0de",
Signature: "deadc0de",
}
retries := newInt(5)
requestID := 42
retryInterval := newInt(300)
status := firmware.UpdateFirmwareStatusAccepted
statusInfo := types.StatusInfo{ReasonCode: "ok", AdditionalInfo: "someInfo"}
requestJson := fmt.Sprintf(`[2,"%v","%v",{"retries":%v,"retryInterval":%v,"requestId":%v,"firmware":{"location":"%v","retrieveDateTime":"%v","installDateTime":"%v","signingCertificate":"%v","signature":"%v"}}]`,
messageId, firmware.UpdateFirmwareFeatureName, *retries, *retryInterval, requestID, fw.Location, fw.RetrieveDateTime.FormatTimestamp(), fw.InstallDateTime.FormatTimestamp(), fw.SigningCertificate, fw.Signature)
responseJson := fmt.Sprintf(`[3,"%v",{"status":"%v","statusInfo":{"reasonCode":"%v","additionalInfo":"%v"}}]`,
messageId, status, statusInfo.ReasonCode, statusInfo.AdditionalInfo)
updateFirmwareResponse := firmware.NewUpdateFirmwareResponse(status)
updateFirmwareResponse.StatusInfo = &statusInfo
channel := NewMockWebSocket(wsId)
handler := &MockChargingStationFirmwareHandler{}
handler.On("OnUpdateFirmware", mock.Anything).Return(updateFirmwareResponse, nil).Run(func(args mock.Arguments) {
request := args.Get(0).(*firmware.UpdateFirmwareRequest)
require.NotNil(t, request)
require.NotNil(t, request.Retries)
assert.Equal(t, *retries, *request.Retries)
require.NotNil(t, request.RetryInterval)
assert.Equal(t, *retryInterval, *request.RetryInterval)
assert.Equal(t, requestID, request.RequestID)
assert.Equal(t, fw.Location, request.Firmware.Location)
assertDateTimeEquality(t, fw.RetrieveDateTime, request.Firmware.RetrieveDateTime)
assertDateTimeEquality(t, fw.InstallDateTime, request.Firmware.InstallDateTime)
assert.Equal(t, fw.SigningCertificate, request.Firmware.SigningCertificate)
assert.Equal(t, fw.Signature, request.Firmware.Signature)
})
setupDefaultCSMSHandlers(suite, expectedCSMSOptions{clientId: wsId, rawWrittenMessage: []byte(requestJson), forwardWrittenMessage: true})
setupDefaultChargingStationHandlers(suite, expectedChargingStationOptions{serverUrl: wsUrl, clientId: wsId, createChannelOnStart: true, channel: channel, rawWrittenMessage: []byte(responseJson), forwardWrittenMessage: true}, handler)
// Run Test
suite.csms.Start(8887, "somePath")
err := suite.chargingStation.Start(wsUrl)
assert.Nil(t, err)
resultChannel := make(chan bool, 1)
err = suite.csms.UpdateFirmware(wsId, func(resp *firmware.UpdateFirmwareResponse, err error) {
assert.Nil(t, err)
require.NotNil(t, resp)
assert.Equal(t, status, resp.Status)
require.NotNil(t, resp.StatusInfo)
assert.Equal(t, statusInfo.ReasonCode, resp.StatusInfo.ReasonCode)
assert.Equal(t, statusInfo.AdditionalInfo, resp.StatusInfo.AdditionalInfo)
resultChannel <- true
}, requestID, fw, func(request *firmware.UpdateFirmwareRequest) {
request.Retries = retries
request.RetryInterval = retryInterval
})
require.Nil(t, err)
result := <-resultChannel
assert.True(t, result)
}
func (suite *OcppV2TestSuite) TestUpdateFirmwareInvalidEndpoint() {
messageId := defaultMessageId
fw := firmware.Firmware{
Location: "https://someurl",
RetrieveDateTime: types.NewDateTime(time.Now()),
InstallDateTime: types.NewDateTime(time.Now()),
SigningCertificate: "1337c0de",
Signature: "deadc0de",
}
retries := newInt(5)
requestID := 42
retryInterval := newInt(300)
request := firmware.NewUpdateFirmwareRequest(requestID, fw)
request.Retries = retries
request.RetryInterval = retryInterval
requestJson := fmt.Sprintf(`[2,"%v","%v",{"retries":%v,"retryInterval":%v,"requestId":%v,"firmware":{"location":"%v","retrieveDateTime":"%v","installDateTime":"%v","signingCertificate":"%v","signature":"%v"}}]`,
messageId, firmware.UpdateFirmwareFeatureName, *retries, *retryInterval, requestID, fw.Location, fw.RetrieveDateTime.FormatTimestamp(), fw.InstallDateTime.FormatTimestamp(), fw.SigningCertificate, fw.Signature)
testUnsupportedRequestFromChargingStation(suite, request, requestJson, messageId)
}