mirror of
https://github.com/pocoproject/poco.git
synced 2026-01-12 00:04:54 +08:00
70bb3a40de
* feat(Foundation): PIDFile and ProcessRunner #4064 * feat(Thread): optional signal blocking on POSIX #2978 * fix(ProcessRunner):remove logger, code enhancement #4225 * feat(Foundation): add PIDFile and ProcessRunner Tests #4064 * fix(Foundation): failing ProcessRunner Test #4064 * fix(PIDFile): remove append argument #4064 * remove Windows TODO from ProcessRunner #4064 * feat(ProcessRunnerTest): add line to checkTimeout #4064 * fix(ProcessRunner): add done flag to run() #4064 * fix(ProcessRunnerTest): add missing pidFile argument #4064 * chore(ProcessRunner): remove comments #4064 * fix(ProcessRunner): add runCount flag #4064 * fix(test): SharedLibrary and Class tests paths * fix(ProcessRunner): thread sanitizer reported data races #4064 * fix(build): pass env var to testrunner #4064 * chore(PIDFile): remove ; in comments #4064 * feat(ProcessRunner): add Win argument format #4064 * fix(Tests): add ProcessRunnerTest to vcxproj #4064 * fix(Tests): change path to TestApp #4064 * feat(Tests): windows processrunner tests #4064 * fix(Tests): duplicate ProcessRunnerTest in TestSuite vcxproj #4064 * fix(CodeQL): sw declaration hides variable #4064 * fix test binaries path for cmake * fix(Build): missing include/PIDFile.h buildWin #4064 * fix(Build): add PocoFoundation depend in buildWin #4064 * feat(ProcessRunner): test process launching multiple threads #2976 --------- Co-authored-by: Pavle <pavle@debian-gnu-linux-11.localdomain> Co-authored-by: Alex Fabijanic <alex@pocoproject.org>
327 lines
7.2 KiB
C++
327 lines
7.2 KiB
C++
//
|
|
// ProcessTest.cpp
|
|
//
|
|
// Copyright (c) 2005-2006, Applied Informatics Software Engineering GmbH.
|
|
// and Contributors.
|
|
//
|
|
// SPDX-License-Identifier: BSL-1.0
|
|
//
|
|
|
|
|
|
#include "ProcessTest.h"
|
|
#include "CppUnit/TestCaller.h"
|
|
#include "CppUnit/TestSuite.h"
|
|
#include "Poco/Process.h"
|
|
#include "Poco/Pipe.h"
|
|
#include "Poco/PipeStream.h"
|
|
#include "Poco/Path.h"
|
|
#include "Poco/Format.h"
|
|
|
|
|
|
using namespace std::string_literals;
|
|
using Poco::Process;
|
|
using Poco::ProcessHandle;
|
|
using Poco::Pipe;
|
|
using Poco::Path;
|
|
using Poco::PipeInputStream;
|
|
using Poco::PipeOutputStream;
|
|
|
|
|
|
ProcessTest::ProcessTest(const std::string& name): CppUnit::TestCase(name)
|
|
{
|
|
}
|
|
|
|
|
|
ProcessTest::~ProcessTest()
|
|
{
|
|
}
|
|
|
|
|
|
void ProcessTest::testEscapeArgs()
|
|
{
|
|
#if defined(_WIN32)
|
|
assertTrue (Poco::ProcessImpl::mustEscapeArg("a b"));
|
|
assertFalse (Poco::ProcessImpl::mustEscapeArg("abc"));
|
|
assertFalse (Poco::ProcessImpl::mustEscapeArg("\"a b \""));
|
|
assertFalse (Poco::ProcessImpl::mustEscapeArg("\"abc\""));
|
|
assertTrue (Poco::ProcessImpl::mustEscapeArg("\"a b "));
|
|
assertFalse (Poco::ProcessImpl::mustEscapeArg("/arg=\"a b c\""));
|
|
|
|
assertEquals ("abc"s, Poco::ProcessImpl::escapeArg("abc"));
|
|
assertEquals ("\"a b c\""s, Poco::ProcessImpl::escapeArg("a b c"));
|
|
assertEquals ("\"a b \\\" c\""s, Poco::ProcessImpl::escapeArg("a b \" c"));
|
|
assertEquals ("/arg=\"a b c\""s, Poco::ProcessImpl::escapeArg("/arg=\"a b c\""));
|
|
#endif
|
|
}
|
|
|
|
|
|
void ProcessTest::testLaunch()
|
|
{
|
|
std::string name("TestApp");
|
|
std::string cmd;
|
|
#if defined(_DEBUG) && (POCO_OS != POCO_OS_ANDROID)
|
|
name += "d";
|
|
#endif
|
|
|
|
#if defined(POCO_OS_FAMILY_UNIX)
|
|
cmd += name;
|
|
#elif defined(_WIN32_WCE)
|
|
cmd = "\\";
|
|
cmd += name;
|
|
cmd += ".EXE";
|
|
#else
|
|
cmd = name;
|
|
#endif
|
|
|
|
std::vector<std::string> args;
|
|
args.push_back("arg1");
|
|
args.push_back("arg2");
|
|
args.push_back("arg3");
|
|
ProcessHandle ph = Process::launch(cmd, args);
|
|
int rc = ph.wait();
|
|
assertTrue (rc == 3);
|
|
}
|
|
|
|
|
|
void ProcessTest::testLaunchRedirectIn()
|
|
{
|
|
#if !defined(_WIN32_WCE)
|
|
std::string name("TestApp");
|
|
std::string cmd;
|
|
#if defined(_DEBUG) && (POCO_OS != POCO_OS_ANDROID)
|
|
name += "d";
|
|
#endif
|
|
|
|
#if defined(POCO_OS_FAMILY_UNIX)
|
|
cmd += name;
|
|
#else
|
|
cmd = name;
|
|
#endif
|
|
|
|
std::vector<std::string> args;
|
|
args.push_back("-count");
|
|
Pipe inPipe;
|
|
ProcessHandle ph = Process::launch(cmd, args, &inPipe, 0, 0);
|
|
PipeOutputStream ostr(inPipe);
|
|
ostr << std::string(100, 'x');
|
|
ostr.close();
|
|
int rc = ph.wait();
|
|
assertTrue (rc == 100);
|
|
#endif // !defined(_WIN32_WCE)
|
|
}
|
|
|
|
|
|
void ProcessTest::testLaunchRedirectOut()
|
|
{
|
|
#if !defined(_WIN32_WCE)
|
|
std::string name("TestApp");
|
|
std::string cmd;
|
|
#if defined(_DEBUG) && (POCO_OS != POCO_OS_ANDROID)
|
|
name += "d";
|
|
#endif
|
|
|
|
#if defined(POCO_OS_FAMILY_UNIX)
|
|
cmd += name;
|
|
#else
|
|
cmd = name;
|
|
#endif
|
|
|
|
std::vector<std::string> args;
|
|
args.push_back("-hello");
|
|
Pipe outPipe;
|
|
ProcessHandle ph = Process::launch(cmd, args, 0, &outPipe, 0);
|
|
PipeInputStream istr(outPipe);
|
|
std::string s;
|
|
int c = istr.get();
|
|
while (c != -1) { s += (char) c; c = istr.get(); }
|
|
assertTrue (s == "Hello, world!");
|
|
int rc = ph.wait();
|
|
assertTrue (rc == 1);
|
|
#endif // !defined(_WIN32_WCE)
|
|
}
|
|
|
|
|
|
void ProcessTest::testLaunchEnv()
|
|
{
|
|
#if !defined(_WIN32_WCE)
|
|
std::string name("TestApp");
|
|
std::string cmd;
|
|
#if defined(_DEBUG) && (POCO_OS != POCO_OS_ANDROID)
|
|
name += "d";
|
|
#endif
|
|
|
|
#if defined(POCO_OS_FAMILY_UNIX)
|
|
cmd += name;
|
|
#else
|
|
cmd = name;
|
|
#endif
|
|
|
|
std::vector<std::string> args;
|
|
args.push_back("-env");
|
|
Pipe outPipe;
|
|
Process::Env env;
|
|
env["TESTENV"] = "test";
|
|
ProcessHandle ph = Process::launch(cmd, args, 0, &outPipe, 0, env);
|
|
PipeInputStream istr(outPipe);
|
|
std::string s;
|
|
int c = istr.get();
|
|
while (c != -1) { s += (char) c; c = istr.get(); }
|
|
assertTrue (s == "test");
|
|
int rc = ph.wait();
|
|
assertTrue (rc == 0);
|
|
#endif // !defined(_WIN32_WCE)
|
|
}
|
|
|
|
|
|
void ProcessTest::testLaunchArgs()
|
|
{
|
|
#if defined (_WIN32) && !defined(_WIN32_WCE)
|
|
std::string name("TestApp");
|
|
#if defined(_DEBUG)
|
|
name += 'd';
|
|
#endif
|
|
std::string cmd = name;
|
|
|
|
std::vector<std::string> args;
|
|
args.push_back("-echo-args");
|
|
args.push_back("simple");
|
|
args.push_back("with space");
|
|
args.push_back("with\ttab");
|
|
args.push_back("with\vverticaltab");
|
|
// can't test newline here because TestApp -echo-args uses newline to separate the echoed args
|
|
//args.push_back("with\nnewline");
|
|
args.push_back("with \" quotes");
|
|
args.push_back("ends with \"quotes\"");
|
|
args.push_back("\"starts\" with quotes");
|
|
args.push_back("\"");
|
|
args.push_back("\\");
|
|
args.push_back("c:\\program files\\ends with backslash\\");
|
|
args.push_back("\"already quoted \\\" \\\\\"");
|
|
Pipe outPipe;
|
|
ProcessHandle ph = Process::launch(cmd, args, 0, &outPipe, 0);
|
|
PipeInputStream istr(outPipe);
|
|
std::string receivedArg;
|
|
int c = istr.get();
|
|
int argNumber = 1;
|
|
while (c != -1)
|
|
{
|
|
if ('\n' == c)
|
|
{
|
|
assertTrue (argNumber < args.size());
|
|
std::string expectedArg = args[argNumber];
|
|
if (expectedArg.npos != expectedArg.find("already quoted")) {
|
|
expectedArg = "already quoted \" \\";
|
|
}
|
|
assertTrue (receivedArg == expectedArg);
|
|
++argNumber;
|
|
receivedArg = "";
|
|
}
|
|
else if ('\r' != c)
|
|
{
|
|
receivedArg += (char)c;
|
|
}
|
|
c = istr.get();
|
|
}
|
|
assertTrue (argNumber == args.size());
|
|
int rc = ph.wait();
|
|
assertTrue (rc == args.size());
|
|
#endif // !defined(_WIN32_WCE)
|
|
}
|
|
|
|
|
|
void ProcessTest::testLaunchInvalidCommand()
|
|
{
|
|
std::string name("InvalidCmd");
|
|
std::string cmd;
|
|
#if defined(_DEBUG) && (POCO_OS != POCO_OS_ANDROID)
|
|
name += "d";
|
|
#endif
|
|
|
|
#if defined(POCO_OS_FAMILY_UNIX)
|
|
cmd += name;
|
|
#elif defined(_WIN32_WCE)
|
|
cmd = "\\";
|
|
cmd += name;
|
|
cmd += ".EXE";
|
|
#else
|
|
cmd = name;
|
|
#endif
|
|
|
|
std::vector<std::string> args;
|
|
args.push_back("arg1");
|
|
args.push_back("arg2");
|
|
args.push_back("arg3");
|
|
#if defined(POCO_OS_FAMILY_UNIX)
|
|
ProcessHandle ph = Process::launch(cmd, args);
|
|
int rc = ph.wait();
|
|
assertTrue (rc == 72);
|
|
#elif defined(POCO_OS_FAMILY_WINDOWS)
|
|
try
|
|
{
|
|
ProcessHandle ph = Process::launch(cmd, args);
|
|
int rc = ph.wait();
|
|
fail("must fail");
|
|
}
|
|
catch (...){}
|
|
#endif
|
|
}
|
|
|
|
|
|
void ProcessTest::testIsRunning()
|
|
{
|
|
#if !defined(_WIN32_WCE)
|
|
std::string name("TestApp");
|
|
std::string cmd;
|
|
#if defined(_DEBUG) && (POCO_OS != POCO_OS_ANDROID)
|
|
name += "d";
|
|
#endif
|
|
|
|
#if defined(POCO_OS_FAMILY_UNIX)
|
|
cmd += name;
|
|
#else
|
|
cmd = name;
|
|
#endif
|
|
|
|
std::vector<std::string> args;
|
|
args.push_back("-count");
|
|
Pipe inPipe;
|
|
ProcessHandle ph = Process::launch(cmd, args, &inPipe, 0, 0);
|
|
Process::PID id = ph.id();
|
|
assertTrue (Process::isRunning(ph));
|
|
assertTrue (Process::isRunning(id));
|
|
PipeOutputStream ostr(inPipe);
|
|
ostr << std::string(100, 'x');
|
|
ostr.close();
|
|
int POCO_UNUSED rc = ph.wait();
|
|
assertTrue (!Process::isRunning(ph));
|
|
assertTrue (!Process::isRunning(id));
|
|
#endif // !defined(_WIN32_WCE)
|
|
}
|
|
|
|
|
|
void ProcessTest::setUp()
|
|
{
|
|
}
|
|
|
|
|
|
void ProcessTest::tearDown()
|
|
{
|
|
}
|
|
|
|
|
|
CppUnit::Test* ProcessTest::suite()
|
|
{
|
|
CppUnit::TestSuite* pSuite = new CppUnit::TestSuite("ProcessTest");
|
|
|
|
CppUnit_addTest(pSuite, ProcessTest, testEscapeArgs);
|
|
CppUnit_addTest(pSuite, ProcessTest, testLaunch);
|
|
CppUnit_addTest(pSuite, ProcessTest, testLaunchRedirectIn);
|
|
CppUnit_addTest(pSuite, ProcessTest, testLaunchRedirectOut);
|
|
CppUnit_addTest(pSuite, ProcessTest, testLaunchEnv);
|
|
CppUnit_addTest(pSuite, ProcessTest, testLaunchArgs);
|
|
CppUnit_addTest(pSuite, ProcessTest, testLaunchInvalidCommand);
|
|
CppUnit_addTest(pSuite, ProcessTest, testIsRunning);
|
|
|
|
return pSuite;
|
|
}
|