From e581a0cf5d8d7257385abad6dbe9138be0631d0f Mon Sep 17 00:00:00 2001 From: Oliver Giles Date: Sat, 22 Jan 2022 08:07:41 +1300 Subject: [PATCH] tests: add setNumExecutors method --- test/laminar-fixture.h | 8 ++++++++ test/tempdir.h | 1 + 2 files changed, 9 insertions(+) diff --git a/test/laminar-fixture.h b/test/laminar-fixture.h index b08c74e..cc3c16b 100644 --- a/test/laminar-fixture.h +++ b/test/laminar-fixture.h @@ -98,6 +98,14 @@ public: return { res.getResult(), kj::mv(log) }; } + void setNumExecutors(int nexec) { + KJ_IF_MAYBE(f, tmp.fs->tryOpenFile(kj::Path{"cfg", "contexts", "default.conf"}, + kj::WriteMode::CREATE | kj::WriteMode::MODIFY | kj::WriteMode::CREATE_PARENT)) { + std::string content = "EXECUTORS=" + std::to_string(nexec); + (*f)->writeAll(content); + } + } + kj::String stripLaminarLogLines(const kj::String& str) { auto out = kj::heapString(str.size()); char *o = out.begin(); diff --git a/test/tempdir.h b/test/tempdir.h index ab9a6f0..d94319e 100644 --- a/test/tempdir.h +++ b/test/tempdir.h @@ -38,6 +38,7 @@ public: // set up empty directory structure fs->openSubdir(kj::Path{"cfg"}, kj::WriteMode::CREATE); fs->openSubdir(kj::Path{"cfg", "jobs"}, kj::WriteMode::CREATE); + fs->openSubdir(kj::Path{"cfg", "contexts"}, kj::WriteMode::CREATE); } void clean() { // rm -rf in config folder