1 #include <gtest/gtest.h>
3 #include <boost/process.hpp>
11 #include "crypto/keymanager.h"
12 #include "http/httpclient.h"
13 #include "libaktualizr/config.h"
14 #include "logging/logging.h"
15 #include "package_manager/packagemanagerfake.h"
16 #include "storage/sqlstorage.h"
17 #include "test_utils.h"
18 #include "uptane/fetcher.h"
19 #include "uptane/tuf.h"
20 #include "utilities/apiqueue.h"
22 static const int die_after = 50;
23 static const int pause_duration = 2;
27 static std::mutex pause_m;
28 static std::condition_variable cv;
29 static bool die =
false;
30 static bool resumed =
false;
34 static void progress_cb(
const Uptane::Target& target,
const std::string& description,
unsigned int progress) {
37 std::cout <<
"progress: " << progress << std::endl;
38 if (progress >= die_after) {
39 std::lock_guard<std::mutex> lk(pause_m);
44 EXPECT_GE(progress, die_after);
49 std::shared_ptr<INvStorage> storage(
new SQLStorage(config.storage,
false));
50 auto http = std::make_shared<HttpClient>();
51 auto pacman = std::make_shared<PackageManagerFake>(config.pacman, config.bootloader, storage, http);
53 KeyManager keys(storage, config.keymanagerConfig());
57 bool res = pacman->fetchTarget(target, fetcher, keys, progress_cb, &token);
63 std::shared_ptr<INvStorage> storage(
new SQLStorage(config.storage,
false));
64 auto http = std::make_shared<HttpClient>();
66 auto pacman = std::make_shared<PackageManagerFake>(config.pacman, config.bootloader, storage, http);
67 KeyManager keys(storage, config.keymanagerConfig());
70 std::promise<bool> download_promise;
71 auto result = download_promise.get_future();
73 std::thread([&target, &fetcher, &download_promise, &token, pacman, &keys]() {
74 bool res = pacman->fetchTarget(target, fetcher, keys, progress_cb, &token);
75 download_promise.set_value(res);
78 std::unique_lock<std::mutex> lk(pause_m);
79 cv.wait(lk, [] {
return die; });
82 std::this_thread::sleep_for(std::chrono::seconds(pause_duration));
89 TEST(FetcherDeathTest, TestResumeAfterPause) {
91 config.storage.path = temp_dir.Path();
92 config.pacman.images_path = temp_dir.Path() /
"images";
94 Json::Value target_json;
95 target_json[
"hashes"][
"sha256"] =
"dd7bd1c37a3226e520b8d6939c30991b1c08772d5dab62b381c3a63541dc629a";
96 target_json[
"length"] = 100 * (1 << 20);
101 ASSERT_DEATH(try_and_die(target,
true),
"");
102 std::cout <<
"Fetcher died, retrying" << std::endl;
106 TEST(FetcherDeathTest, TestResumeAfterSigkill) {
108 config.storage.path = temp_dir.Path();
110 Json::Value target_json;
111 target_json[
"hashes"][
"sha256"] =
"dd7bd1c37a3226e520b8d6939c30991b1c08772d5dab62b381c3a63541dc629a";
112 target_json[
"length"] = 100 * (1 << 20);
117 ASSERT_DEATH(try_and_die(target,
false),
"");
118 std::cout <<
"Fetcher died, retrying" << std::endl;
123 int main(
int argc,
char** argv) {
124 ::testing::InitGoogleTest(&argc, argv);
127 logger_set_threshold(boost::log::trivial::trace);
129 std::string port = TestUtils::getFreePort();
130 server =
"http://127.0.0.1:" + port;
131 config.uptane.repo_server = server;
132 boost::process::child http_server_process(
"tests/fake_http_server/fake_test_server.py", port,
"-f");
133 TestUtils::waitForServer(server +
"/");
134 return RUN_ALL_TESTS();
136 #endif // __NO_MAIN__