diff --git a/tests/src/App/ProjectFile.cpp b/tests/src/App/ProjectFile.cpp index d7f1f89389..d726b7d350 100644 --- a/tests/src/App/ProjectFile.cpp +++ b/tests/src/App/ProjectFile.cpp @@ -3,6 +3,7 @@ #include "InitApplication.h" #include #include +#include #include // NOLINTBEGIN @@ -23,6 +24,10 @@ protected: resDir.append("/tests/ProjectTest.FCStd"); return resDir; } + std::string imageFileName() const + { + return {"thumbnails/Thumbnail.png"}; + } std::list getInventorObjects() const { return {"Body"}; @@ -103,4 +108,25 @@ TEST_F(ProjectFileTest, getTypeId) Base::Type id = proj.getTypeId("Body"); EXPECT_EQ(id, getInventorId()); } + +TEST_F(ProjectFileTest, getThumbnailBuffer) +{ + App::ProjectFile proj(fileName()); + std::stringstream str; + proj.readInputFileDirect(imageFileName(), str); + std::string buffer = str.str(); + EXPECT_EQ(buffer.size(), 2857); +} + +TEST_F(ProjectFileTest, getThumbnailFile) +{ + App::ProjectFile proj(fileName()); + Base::FileInfo fi(Base::FileInfo::getTempFileName()); + Base::ofstream file(fi, std::ios::out | std::ios::binary); + proj.readInputFileDirect(imageFileName(), file); + file.flush(); + file.close(); + EXPECT_EQ(fi.size(), 2857); + fi.deleteFile(); +} // NOLINTEND