Mock list capabilities of FileManager

daemon/include/FileManager.h

@@ -49,6 +49,14 @@ private:
    */
   int remainingListChunks;
 
+  /**
+   * Get vector of all files in current directory.
+   * Ignores directories
+   *
+   * @return vector of all files in the current directory
+   */
+  std::vector<std::string> listFiles();
+
 public:
   enum { max_data_length = 512 };
 
@@ -129,30 +137,22 @@ public:
    */
   virtual std::vector<char> readGet();
 
-  /**
-   * Get vector of all files in current directory.
-   * Ignores directories
-   *
-   * @return vector of all files in the current directory
-   */
-  std::vector<std::string> listFiles();
-
   /**
    * Open list command. Set list vector and claculate chunks
    *
    * @return chunks of the resulting list.
    */
-  int openList();
+  virtual int openList();
 
   /**
    * @return remaining chunks to be send with listdata command
    */
-  int getRemainingListChunks();
+  virtual int getRemainingListChunks();
 
   /**
    * @return size of the list vector
    */
-  int getListSize();
+  virtual int getListSize();
 
   /**
    * Return next chunk for listdata command.
@@ -166,7 +166,7 @@ public:
    * Cancel current list command.
    * Clear list vector and set remainingListChunks zero.
    */
-   void cancelList();
+  void cancelList();
 };
 
 #endif

daemon/test/FileManagerMock.h

@@ -24,6 +24,10 @@ public:
 
   MOCK_METHOD(void, writePut, (const std::vector<char> &data), (override));
   MOCK_METHOD(std::vector<char>, readGet, (), (override));
+
+  MOCK_METHOD(int, openList,(), (override));
+  MOCK_METHOD(int, getRemainingListChunks, (), (override));
+  MOCK_METHOD(int, getListSize, (), (override));
 };
 
 #endif