TestFile.hpp

Go to the documentation of this file.

#pragma once
 
#include <filesystem>
#include <fstream>
 
namespace stc::testutil {
 
struct TestFile {
    std::filesystem::path file;
 
    [[nodiscard("Discarding this object will immediately delete the file")]]
    TestFile(const std::filesystem::path& f, bool create = false) : file(f) {
        if (std::filesystem::exists(file) && !std::filesystem::is_regular_file(f)) {
            throw std::runtime_error(
                "The file you supplied to TestFile exists, and is not a regular file. "
                "This utility is only compatible with standard files. "
                "If you were trying to take ownership of a directory, use TestDirectory instead."
            );
        }
        deleteFile();
 
        if (create) {
            std::ofstream of(f);
            if (!of) {
                throw std::runtime_error(
                    "Failed to open test file. Do you have dir permissions?"
                );
            }
        }
    }
    ~TestFile() {
        deleteFile();
    }
    void deleteFile() {
        if (std::filesystem::exists(file)) {
            std::filesystem::remove(file);
        }
    }
};
 
}