[VOL-3880] Correctly reporting software image status in OMCI Get
[VOL-3900] OMCI ONU Software Image Download
Change-Id: I8d91be832f3a89404d0af0dd98e6b53359e6a738
diff --git a/internal/common/omci/image_test.go b/internal/common/omci/image_test.go
new file mode 100644
index 0000000..ad07eac
--- /dev/null
+++ b/internal/common/omci/image_test.go
@@ -0,0 +1,93 @@
+/*
+ * Copyright 2018-present Open Networking Foundation
+
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+
+ * http://www.apache.org/licenses/LICENSE-2.0
+
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package omci
+
+import (
+ "github.com/google/gopacket"
+ "github.com/opencord/omci-lib-go"
+ me "github.com/opencord/omci-lib-go/generated"
+ "gotest.tools/assert"
+ "testing"
+)
+
+func omciToStartSoftwareDownloadResponse(t *testing.T, omciPkt *gopacket.Packet) *omci.StartSoftwareDownloadResponse {
+ msgLayer := (*omciPkt).Layer(omci.LayerTypeStartSoftwareDownloadResponse)
+ if msgLayer == nil {
+ t.Fatal("omci Msg layer could not be detected for StartSoftwareDownloadResponse")
+ }
+ msgObj, msgOk := msgLayer.(*omci.StartSoftwareDownloadResponse)
+ if !msgOk {
+ t.Fatal("omci Msg layer could not be assigned for StartSoftwareDownloadResponse")
+ }
+ return msgObj
+}
+
+func TestCreateStartSoftwareDownloadResponse(t *testing.T) {
+ omciReq := &omci.StartSoftwareDownloadRequest{
+ MeBasePacket: omci.MeBasePacket{
+ EntityClass: me.SoftwareImageClassID,
+ EntityInstance: 1,
+ },
+ ImageSize: 32768,
+ NumberOfCircuitPacks: 1,
+ WindowSize: 31,
+ CircuitPacks: []uint16{0},
+ }
+
+ omciReqPkt, err := Serialize(omci.StartSoftwareDownloadRequestType, omciReq, 66)
+ if err != nil {
+ t.Fatal(err.Error())
+ }
+
+ omciReqPkt, _ = HexEncode(omciReqPkt)
+
+ // start test
+ pkt, msg, _ := ParseOpenOltOmciPacket(omciReqPkt)
+ responsePkt, err := CreateStartSoftwareDownloadResponse(pkt, msg)
+ assert.NilError(t, err)
+
+ omciResponseMsg, omciResponsePkt := omciBytesToMsg(t, responsePkt)
+ assert.Equal(t, omciResponseMsg.MessageType, omci.StartSoftwareDownloadResponseType)
+
+ getResponseLayer := omciToStartSoftwareDownloadResponse(t, omciResponsePkt)
+
+ assert.Equal(t, getResponseLayer.Result, me.Success)
+}
+
+func TestComputeDownloadSectionsCount(t *testing.T) {
+ omciReq := &omci.StartSoftwareDownloadRequest{
+ MeBasePacket: omci.MeBasePacket{
+ EntityClass: me.SoftwareImageClassID,
+ EntityInstance: 1,
+ },
+ ImageSize: 32768,
+ NumberOfCircuitPacks: 1,
+ WindowSize: 31,
+ CircuitPacks: []uint16{0},
+ }
+
+ omciReqPkt, err := Serialize(omci.StartSoftwareDownloadRequestType, omciReq, 66)
+ if err != nil {
+ t.Fatal(err.Error())
+ }
+
+ omciReqPkt, _ = HexEncode(omciReqPkt)
+ pkt, _, _ := ParseOpenOltOmciPacket(omciReqPkt)
+
+ count := ComputeDownloadSectionsCount(pkt)
+ assert.Equal(t, count, 1058)
+}