41 lines
1.3 KiB
C++
41 lines
1.3 KiB
C++
#include "gtest/gtest.h"
|
|
#include "test.h"
|
|
|
|
/* Utility functions */
|
|
std::vector<unsigned char> generic_read_file(const std::string& filename){
|
|
auto ifstream = std::ifstream{filename, std::ios::binary};
|
|
auto path = std::filesystem::path{filename};
|
|
auto size = std::filesystem::file_size(path);
|
|
std::vector<unsigned char> buf(size);
|
|
ifstream.read(reinterpret_cast<char*>(buf.data()), size);
|
|
return buf;
|
|
}
|
|
|
|
std::vector<std::string> get_created_filenames(const horcrux::FsCryptoOutput& out){
|
|
std::vector<std::string> result(out.created_files.size());
|
|
transform(out.created_files.begin(), out.created_files.end(), result.begin(),
|
|
[](auto path){return path.string();});
|
|
return result;
|
|
}
|
|
void delete_created_files(const horcrux::FsCryptoOutput& out){
|
|
std::for_each(out.created_files.begin(), out.created_files.end(),
|
|
[](auto f){std::filesystem::remove(f);});
|
|
}
|
|
|
|
std::vector<std::string> get_encrypted_files(const std::string& folder, const std::string& basename) {
|
|
std::vector<std::string> result;
|
|
for (auto& p: std::filesystem::directory_iterator(folder)){
|
|
if (p.path().string().starts_with(basename) &&
|
|
p.path().string().ends_with(".enc"))
|
|
result.push_back(p.path().string());
|
|
}
|
|
std::sort(result.begin(), result.end());
|
|
return result;
|
|
}
|
|
|
|
|
|
int main(int argc, char *argv[]) {
|
|
::testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|