From 0bb82e0da290418cad3d860ade67ae9c630ced15 Mon Sep 17 00:00:00 2001
From: Robin Mueller <muellerr@irs.uni-stuttgart.de>
Date: Wed, 17 Aug 2022 16:47:46 +0200
Subject: [PATCH] add fs mock feed function

---
 unittests/hal/CMakeLists.txt       |  2 +-
 unittests/hal/testFsMock.cpp       | 49 ++++++++++++++++++++++++++++++
 unittests/mocks/FilesystemMock.cpp | 20 +++++++++++-
 unittests/mocks/FilesystemMock.h   |  6 ++++
 4 files changed, 75 insertions(+), 2 deletions(-)
 create mode 100644 unittests/hal/testFsMock.cpp

diff --git a/unittests/hal/CMakeLists.txt b/unittests/hal/CMakeLists.txt
index 5e2887b29..5d1c78465 100644
--- a/unittests/hal/CMakeLists.txt
+++ b/unittests/hal/CMakeLists.txt
@@ -1,2 +1,2 @@
 target_sources(${FSFW_TEST_TGT} PRIVATE testCommandExecutor.cpp
-                                        testHostFilesystem.cpp)
+                                        testHostFilesystem.cpp testFsMock.cpp)
diff --git a/unittests/hal/testFsMock.cpp b/unittests/hal/testFsMock.cpp
new file mode 100644
index 000000000..7040e7346
--- /dev/null
+++ b/unittests/hal/testFsMock.cpp
@@ -0,0 +1,49 @@
+#include <catch2/catch_test_macros.hpp>
+
+#include "fsfw/serialize/SerializeIF.h"
+#include "mocks/FilesystemMock.h"
+
+using namespace std;
+
+TEST_CASE("Filesystem Mock", "[mocks]") {
+  auto fsMock = FilesystemMock();
+
+  SECTION("Create File") {
+    FilesystemParams params("hello.txt");
+    CHECK(fsMock.createFile(params) == result::OK);
+    auto iter = fsMock.fileMap.find("hello.txt");
+    REQUIRE(iter != fsMock.fileMap.end());
+    FilesystemMock::FileInfo &stats = iter->second;
+    CHECK(stats.fileSegQueue.empty());
+    CHECK(stats.fileRaw.empty());
+  }
+
+  SECTION("Write to File") {
+    std::string testData = "test data";
+    FileOpParams params("hello.txt", testData.size());
+    CHECK(fsMock.writeToFile(params, reinterpret_cast<const uint8_t *>(testData.data())) ==
+          HasReturnvaluesIF::RETURN_OK);
+    auto iter = fsMock.fileMap.find("hello.txt");
+    REQUIRE(iter != fsMock.fileMap.end());
+    FilesystemMock::FileInfo &stats = iter->second;
+    CHECK(not stats.fileSegQueue.empty());
+    CHECK(not stats.fileRaw.empty());
+    auto &segment = stats.fileSegQueue.back();
+    CHECK(segment.offset == 0);
+    CHECK(std::string(reinterpret_cast<const char *>(segment.data.data()), segment.data.size()) ==
+          testData);
+    CHECK(std::string(reinterpret_cast<const char *>(stats.fileRaw.data()), segment.data.size()) ==
+          testData);
+  }
+
+  SECTION("Create Directory") {
+    FilesystemParams params("hello");
+    CHECK(fsMock.createDirectory(params) == result::OK);
+    REQUIRE(not fsMock.dirMap.empty());
+    auto iter = fsMock.dirMap.find("hello");
+    REQUIRE(iter != fsMock.dirMap.end());
+    auto &dirInfo = iter->second;
+    CHECK(dirInfo.createCallCount == 1);
+    CHECK(dirInfo.delCallCount == 0);
+  }
+}
\ No newline at end of file
diff --git a/unittests/mocks/FilesystemMock.cpp b/unittests/mocks/FilesystemMock.cpp
index 02620fd54..8f9e3f8f7 100644
--- a/unittests/mocks/FilesystemMock.cpp
+++ b/unittests/mocks/FilesystemMock.cpp
@@ -1,7 +1,21 @@
 #include "FilesystemMock.h"
 
+#include <filesystem>
+
 #include "fsfw/serialize/SerializeIF.h"
 
+ReturnValue_t FilesystemMock::feedFile(const std::string &filename, std::ifstream &file) {
+  if (not std::filesystem::exists(filename)) {
+    return HasReturnvaluesIF::RETURN_FAILED;
+  }
+  size_t fileSize = std::filesystem::file_size(filename);
+  FileOpParams params(filename.c_str(), fileSize);
+  std::vector<uint8_t> rawData(fileSize);
+  file.read(reinterpret_cast<char *>(rawData.data()), static_cast<unsigned int>(rawData.size()));
+  createOrAddToFile(params, rawData.data());
+  return HasReturnvaluesIF::RETURN_OK;
+}
+
 ReturnValue_t FilesystemMock::writeToFile(FileOpParams params, const uint8_t *data) {
   createOrAddToFile(params, data);
   return HasReturnvaluesIF::RETURN_OK;
@@ -75,7 +89,9 @@ void FilesystemMock::createOrAddToFile(FileOpParams params, const uint8_t *data)
   auto iter = fileMap.find(filename);
   if (iter == fileMap.end()) {
     FileSegmentQueue queue;
-    queue.push(FileWriteInfo(filename, params.offset, data, params.size));
+    if (params.size > 0) {
+      queue.push(FileWriteInfo(filename, params.offset, data, params.size));
+    }
     FileInfo info;
     info.fileSegQueue = queue;
     if (data != nullptr) {
@@ -105,4 +121,6 @@ void FilesystemMock::createOrAddToFile(FileOpParams params, const uint8_t *data)
 void FilesystemMock::reset() {
   fileMap.clear();
   dirMap.clear();
+  std::queue<RenameInfo> empty;
+  std::swap(renameQueue, empty);
 }
diff --git a/unittests/mocks/FilesystemMock.h b/unittests/mocks/FilesystemMock.h
index f7c53eda9..19d4d38cb 100644
--- a/unittests/mocks/FilesystemMock.h
+++ b/unittests/mocks/FilesystemMock.h
@@ -1,6 +1,7 @@
 #ifndef FSFW_MOCKS_FILESYSTEMMOCK_H
 #define FSFW_MOCKS_FILESYSTEMMOCK_H
 
+#include <fstream>
 #include <map>
 #include <queue>
 #include <string>
@@ -53,6 +54,7 @@ class FilesystemMock : public HasFileSystemIF {
   };
   std::queue<RenameInfo> renameQueue;
 
+  ReturnValue_t feedFile(const std::string &filename, std::ifstream &file);
   ReturnValue_t writeToFile(FileOpParams params, const uint8_t *data) override;
   ReturnValue_t readFromFile(FileOpParams params, uint8_t **buffer, size_t &readSize,
                              size_t maxSize) override;
@@ -64,6 +66,10 @@ class FilesystemMock : public HasFileSystemIF {
 
   void reset();
 
+  using HasFileSystemIF::createDirectory;
+  using HasFileSystemIF::createFile;
+  using HasFileSystemIF::readFromFile;
+
  private:
   void createOrAddToFile(FileOpParams params, const uint8_t *data);
 };