-
Notifications
You must be signed in to change notification settings - Fork 22
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
9716649
commit 2617704
Showing
16 changed files
with
212 additions
and
28 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,165 @@ | ||
| package spine | ||
|
|
||
| import ( | ||
| "encoding/json" | ||
| "fmt" | ||
| "os" | ||
| "sync" | ||
| "testing" | ||
|
|
||
| "github.com/enbility/eebus-go/spine/model" | ||
| "github.com/google/go-cmp/cmp" | ||
| "github.com/google/go-cmp/cmp/cmpopts" | ||
| "github.com/stretchr/testify/assert" | ||
| ) | ||
|
|
||
| const ( | ||
| wallbox_detaileddiscoverydata_recv_reply_file_path = "./testdata/wallbox_detaileddiscoverydata_recv_reply.json" | ||
| wallbox_detaileddiscoverydata_recv_notify_file_path = "./testdata/wallbox_detaileddiscoverydata_recv_notify.json" | ||
| ) | ||
|
|
||
| type WriteMessageHandler struct { | ||
| sentMessages [][]byte | ||
|
|
||
| mux sync.Mutex | ||
| } | ||
|
|
||
| var _ SpineDataConnection = (*WriteMessageHandler)(nil) | ||
|
|
||
| func (t *WriteMessageHandler) WriteSpineMessage(message []byte) { | ||
| t.mux.Lock() | ||
| defer t.mux.Unlock() | ||
|
|
||
| t.sentMessages = append(t.sentMessages, message) | ||
| } | ||
|
|
||
| func (t *WriteMessageHandler) LastMessage() []byte { | ||
| t.mux.Lock() | ||
| defer t.mux.Unlock() | ||
|
|
||
| if len(t.sentMessages) == 0 { | ||
| return nil | ||
| } | ||
|
|
||
| return t.sentMessages[len(t.sentMessages)-1] | ||
| } | ||
|
|
||
| func (t *WriteMessageHandler) MessageWithReference(msgCounterReference *model.MsgCounterType) []byte { | ||
| t.mux.Lock() | ||
| defer t.mux.Unlock() | ||
|
|
||
| var datagram model.Datagram | ||
|
|
||
| for _, msg := range t.sentMessages { | ||
| if err := json.Unmarshal(msg, &datagram); err != nil { | ||
| return nil | ||
| } | ||
| if datagram.Datagram.Header.MsgCounterReference == nil { | ||
| continue | ||
| } | ||
| if uint(*datagram.Datagram.Header.MsgCounterReference) != uint(*msgCounterReference) { | ||
| continue | ||
| } | ||
| if datagram.Datagram.Payload.Cmd[0].ResultData != nil { | ||
| continue | ||
| } | ||
|
|
||
| return msg | ||
| } | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| func (t *WriteMessageHandler) ResultWithReference(msgCounterReference *model.MsgCounterType) []byte { | ||
| t.mux.Lock() | ||
| defer t.mux.Unlock() | ||
|
|
||
| var datagram model.Datagram | ||
|
|
||
| for _, msg := range t.sentMessages { | ||
| if err := json.Unmarshal(msg, &datagram); err != nil { | ||
| return nil | ||
| } | ||
| if datagram.Datagram.Header.MsgCounterReference == nil { | ||
| continue | ||
| } | ||
| if uint(*datagram.Datagram.Header.MsgCounterReference) != uint(*msgCounterReference) { | ||
| continue | ||
| } | ||
| if datagram.Datagram.Payload.Cmd[0].ResultData == nil { | ||
| continue | ||
| } | ||
|
|
||
| return msg | ||
| } | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| func loadFileData(t *testing.T, fileName string) []byte { | ||
| fileData, err := os.ReadFile(fileName) | ||
| if err != nil { | ||
| t.Fatal(err) | ||
| } | ||
|
|
||
| return fileData | ||
| } | ||
|
|
||
| func checkSentData(t *testing.T, sendBytes []byte, msgSendFilePrefix string) { | ||
| msgSendExpectedBytes, err := os.ReadFile(msgSendFilePrefix + "_expected.json") | ||
| if err != nil { | ||
| t.Fatal(err) | ||
| } | ||
|
|
||
| msgSendActualFileName := msgSendFilePrefix + "_actual.json" | ||
| equal := jsonDatagramEqual(t, msgSendExpectedBytes, sendBytes) | ||
| if !equal { | ||
| saveJsonToFile(t, sendBytes, msgSendActualFileName) | ||
| } | ||
| assert.Truef(t, equal, "Assert equal failed! Check '%s' ", msgSendActualFileName) | ||
| } | ||
|
|
||
| func jsonDatagramEqual(t *testing.T, expectedJson, actualJson []byte) bool { | ||
| var actualDatagram model.Datagram | ||
| if err := json.Unmarshal(actualJson, &actualDatagram); err != nil { | ||
| t.Fatal(err) | ||
| } | ||
| var expectedDatagram model.Datagram | ||
| if err := json.Unmarshal(expectedJson, &expectedDatagram); err != nil { | ||
| t.Fatal(err) | ||
| } | ||
|
|
||
| less := func(a, b model.FunctionPropertyType) bool { return string(*a.Function) < string(*b.Function) } | ||
| return cmp.Equal(expectedDatagram, actualDatagram, cmpopts.SortSlices(less)) | ||
| } | ||
|
|
||
| func saveJsonToFile(t *testing.T, data json.RawMessage, fileName string) { | ||
| jsonIndent, err := json.MarshalIndent(data, "", " ") | ||
| if err != nil { | ||
| t.Fatal(err) | ||
| } | ||
| err = os.WriteFile(fileName, jsonIndent, os.ModePerm) | ||
| if err != nil { | ||
| t.Fatal(err) | ||
| } | ||
| } | ||
|
|
||
| func waitForAck(t *testing.T, msgCounterReference *model.MsgCounterType, writeHandler *WriteMessageHandler) { | ||
| var datagram model.Datagram | ||
|
|
||
| msg := writeHandler.ResultWithReference(msgCounterReference) | ||
| if msg == nil { | ||
| t.Fatal("acknowledge message was not sent!!") | ||
| } | ||
|
|
||
| if err := json.Unmarshal(msg, &datagram); err != nil { | ||
| t.Fatal(err) | ||
| } | ||
|
|
||
| cmd := datagram.Datagram.Payload.Cmd[0] | ||
| if cmd.ResultData != nil { | ||
| if cmd.ResultData.ErrorNumber != nil && uint(*cmd.ResultData.ErrorNumber) != uint(model.ErrorNumberTypeNoError) { | ||
| t.Fatal(fmt.Errorf("error '%d' result data received", uint(*cmd.ResultData.ErrorNumber))) | ||
| } | ||
| } | ||
| } |
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
File renamed without changes.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.