/* Copyright (C) CJ Affiliate
*
* You may use, distribute and modify this code under the
* terms of the GNU General Public License version 2 or
* later.
*
* You should have received a copy of the license with this
* file. If not, you will find a copy in the "LICENSE" file
* at https://github.com/cjdev/dual-control.
*/
#include <memory>
#include <cstring>
#include <pwd.h>
#include <cstdio>
#include <sys/stat.h>
#include <fstream>
#include <sstream>
#include "sys_fstream.h"
#include "random_source.h"
#include "test_util.h"
class fake_fstreams : public fstreams_ifc
{
private:
std::string expected_file_path_;
std::string file_contents_;
public:
fake_fstreams (const std::string &expected_file_path,
const std::string &file_contents)
: expected_file_path_ (expected_file_path),
file_contents_ (file_contents) {}
pstream open_fstream (const std::string &file_path) const override
{
if (file_path == expected_file_path_) {
return fstreams::pstream (new std::istringstream (file_contents_));
} else {
return fstreams_ifc::open_fstream (file_path);
}
}
};
int reads_from_the_right_file ()
{
//given
std::string random_source = random_source_ifc::file_path;
std::string very_random_bytes { 4, 2 };
// hardcoded file name is .dual_control in the user's home directory
fstreams test_streams (fstreams::delegate (new fake_fstreams (random_source,
very_random_bytes)));
//file_reader test_file_reader (file_reader::delegate (new fake_file_reader));
class random_source supplier (random_source::create (test_streams));
//when
std::vector<uint8_t> actual_v = supplier.get_random_bytes (2);
std::string actual (actual_v.begin(), actual_v.end());
//then
check (actual == very_random_bytes, "didn't get the expected bytes");
succeed();
}
int handles_missing_random_source_correctly ()
{
//given
std::string very_random_bytes { 4, 2 };
// hardcoded file name is .dual_control in the user's home directory
fstreams test_streams (fstreams::delegate (new fake_fstreams ("/nowhere",
very_random_bytes)));
//file_reader test_file_reader (file_reader::delegate (new fake_file_reader));
class random_source supplier (random_source::create (test_streams));
//when
try {
supplier.get_random_bytes (2);
fail ("get_random_bytes() should throw if it can't access /dev/urandom");
} catch (std::exception e) {
succeed();
}
//then
succeed();
}
int run_tests()
{
test (reads_from_the_right_file);
test (handles_missing_random_source_correctly);
succeed();
}
int main (int argc, char *argv[])
{
return !run_tests();
}