unittest for small file transfer done

unittests/cfdp/handler/testSourceHandler.cpp

@@ -1,3 +1,5 @@
+#include <etl/crc32.h>
+
 #include <catch2/catch_test_macros.hpp>
 #include <filesystem>
 
@@ -6,6 +8,7 @@
 #include "fsfw/cfdp/handler/SourceHandler.h"
 #include "fsfw/cfdp/pdu/EofPduCreator.h"
 #include "fsfw/cfdp/pdu/EofPduReader.h"
+#include "fsfw/cfdp/pdu/FileDataReader.h"
 #include "fsfw/cfdp/pdu/MetadataPduCreator.h"
 #include "fsfw/cfdp/pdu/MetadataPduReader.h"
 #include "fsfw/tmtcservices/TmTcMessage.h"
@@ -57,15 +60,20 @@ TEST_CASE("CFDP Source Handler", "[cfdp]") {
   PutRequest putRequest(id, srcNameLv, destNameLv);
   CHECK(sourceHandler.initialize() == OK);
 
-  auto genericMetadataCheck = [&](SourceHandler::FsmResult& fsmResult, size_t expectedFileSize) {
-    TmTcMessage tmtcMessage;
-    const uint8_t* pduPtr;
+  auto onePduSentCheck = [&](SourceHandler::FsmResult& fsmResult, TmTcMessage& tmtcMessage,
+                             const uint8_t** pduPtr) {
     CHECK(fsmResult.packetsSent == 1);
     CHECK(mqMock.numberOfSentMessages() == 1);
     REQUIRE(mqMock.getNextSentMessage(destQueueId, tmtcMessage) == OK);
     auto accessor = tmStore.getData(tmtcMessage.getStorageId());
     REQUIRE(accessor.first == OK);
-    pduPtr = accessor.second.data();
+    *pduPtr = accessor.second.data();
+    return std::move(accessor);
+  };
+  auto genericMetadataCheck = [&](SourceHandler::FsmResult& fsmResult, size_t expectedFileSize) {
+    TmTcMessage tmtcMessage;
+    const uint8_t* pduPtr;
+    auto accessor = onePduSentCheck(fsmResult, tmtcMessage, &pduPtr);
     CHECK(accessor.second.size() == 55);
     MetadataGenericInfo metadataInfo;
     MetadataPduReader metadataReader(pduPtr, accessor.second.size(), metadataInfo, nullptr, 0);
@@ -87,12 +95,7 @@ TEST_CASE("CFDP Source Handler", "[cfdp]") {
                              uint32_t expectedChecksum) {
     TmTcMessage tmtcMessage;
     const uint8_t* pduPtr;
-    CHECK(fsmResult.packetsSent == 1);
-    CHECK(mqMock.numberOfSentMessages() == 1);
-    REQUIRE(mqMock.getNextSentMessage(destQueueId, tmtcMessage) == OK);
-    auto accessor = tmStore.getData(tmtcMessage.getStorageId());
-    REQUIRE(accessor.first == OK);
-    pduPtr = accessor.second.data();
+    auto accessor = onePduSentCheck(fsmResult, tmtcMessage, &pduPtr);
     // 10 byte PDU header, 1 byte directive field, 1 byte condition code, 4 byte checksum,
     // 4 byte FSS
     CHECK(accessor.second.size() == 20);
@@ -136,10 +139,32 @@ TEST_CASE("CFDP Source Handler", "[cfdp]") {
     // Verify metadata PDU was sent.
     genericMetadataCheck(fsmResult, expectedFileSize);
     fsmResult = sourceHandler.stateMachine();
-    // TODO: Verify one file data PDU was sent.
-    CHECK(fsmResult.packetsSent == 1);
+    TmTcMessage tmtcMessage;
+    const uint8_t* pduPtr;
+    auto accessor = onePduSentCheck(fsmResult, tmtcMessage, &pduPtr);
+    FileDataInfo fdInfo;
+    FileDataReader fdReader(pduPtr, accessor.second.size(), fdInfo);
+    // 10 byte PDU header, 4 byte offset, 12 bytes file data.
+    CHECK(accessor.second.size() == 26);
+    CHECK(fdReader.parseData() == OK);
+    CHECK(fdInfo.getOffset().value() == 0);
+    size_t fileSize = 0;
+    const uint8_t* fileData = fdInfo.getFileData(&fileSize);
+    REQUIRE(fileSize == fileContent.size());
+    CHECK(fileData != nullptr);
+    std::string dataReadBack(reinterpret_cast<const char*>(fileData), fileSize);
+    CHECK(dataReadBack == fileContent);
+    mqMock.clearMessages();
 
-    // TODO: Verify one EOF PDU was sent.
+    fsmResult = sourceHandler.stateMachine();
+
+    etl::crc32 crcCalc;
+    crcCalc.add(fileContent.data(), fileContent.data() + fileContent.size());
+    // Verify EOF PDU was sent.
+    genericEofCheck(fsmResult, expectedFileSize, crcCalc.value());
+
+    CHECK(sourceHandler.getStep() == SourceHandler::TransactionStep::IDLE);
+    CHECK(sourceHandler.getState() == CfdpState::IDLE);
   }
 
   SECTION("Transfer two segment file") {}

unittests/cfdp/pdu/testFileData.cpp

@@ -107,8 +107,7 @@ TEST_CASE("File Data PDU", "[cfdp][pdu]") {
         serializer.serialize(&buffer, &sz, fileDataBuffer.size(), SerializeIF::Endianness::NETWORK);
     REQUIRE(result == returnvalue::OK);
 
-    Fss emptyOffset;
-    FileDataInfo emptyInfo(emptyOffset);
+    FileDataInfo emptyInfo;
     FileDataReader deserializer(fileDataBuffer.data(), fileDataBuffer.size(), emptyInfo);
     result = deserializer.parseData();
     REQUIRE(result == returnvalue::OK);