tests/exec/exec.cpp

   1 ///////////////////////////////////////////////////////////////////////////////
   2 // Name:        tests/exec/exec.cpp
   3 // Purpose:     test wxExecute()
   4 // Author:      Francesco Montorsi
   5 //              (based on console sample TestExecute() function)
   6 // Created:     2009-01-10
   7 // RCS-ID:      $Id$
   8 // Copyright:   (c) 2009 Francesco Montorsi
   9 ///////////////////////////////////////////////////////////////////////////////
  10
  11 // ----------------------------------------------------------------------------
  12 // headers
  13 // ----------------------------------------------------------------------------
  14
  15 #include "testprec.h"
  16
  17 #ifdef __BORLANDC__
  18     #pragma hdrstop
  19 #endif
  20
  21 #include "wx/utils.h"
  22 #include "wx/process.h"
  23 #include "wx/sstream.h"
  24
  25 #ifdef __UNIX__
  26     #define COMMAND "echo hi"
  27     #define ASYNC_COMMAND "xclock"
  28     #define SHELL_COMMAND "echo hi from shell>/dev/null"
  29     #define COMMAND_NO_OUTPUT "echo -n"
  30 #elif defined(__WXMSW__)
  31     #define COMMAND "cmd.exe /c \"echo hi\""
  32     #define ASYNC_COMMAND "notepad"
  33     #define SHELL_COMMAND "echo hi > nul:"
  34     #define COMMAND_NO_OUTPUT COMMAND " > nul:"
  35 #else
  36     #error "no command to exec"
  37 #endif // OS
  38
  39 // ----------------------------------------------------------------------------
  40 // test class
  41 // ----------------------------------------------------------------------------
  42
  43 class ExecTestCase : public CppUnit::TestCase
  44 {
  45 public:
  46     ExecTestCase() { }
  47
  48 private:
  49     CPPUNIT_TEST_SUITE( ExecTestCase );
  50         CPPUNIT_TEST( TestShell );
  51         CPPUNIT_TEST( TestExecute );
  52         CPPUNIT_TEST( TestProcess );
  53     CPPUNIT_TEST_SUITE_END();
  54
  55     void TestShell();
  56     void TestExecute();
  57     void TestProcess();
  58
  59     DECLARE_NO_COPY_CLASS(ExecTestCase)
  60 };
  61
  62 // register in the unnamed registry so that these tests are run by default
  63 CPPUNIT_TEST_SUITE_REGISTRATION( ExecTestCase );
  64
  65 // also include in its own registry so that these tests can be run alone
  66 CPPUNIT_TEST_SUITE_NAMED_REGISTRATION( ExecTestCase, "ExecTestCase" );
  67
  68
  69 void ExecTestCase::TestShell()
  70 {
  71     CPPUNIT_ASSERT( wxShell(SHELL_COMMAND) );
  72 }
  73
  74 void ExecTestCase::TestExecute()
  75 {
  76     // test sync exec (with a command not producing any output to avoid
  77     // interfering with the test):
  78     CPPUNIT_ASSERT( wxExecute(COMMAND_NO_OUTPUT, wxEXEC_SYNC) == 0 );
  79
  80     // test asynch exec
  81     long pid = wxExecute(ASYNC_COMMAND, wxEXEC_ASYNC);
  82     CPPUNIT_ASSERT( pid != 0 );
  83
  84     // NOTE: under Windows the first wxKill() invocation with wxSIGTERM
  85     //       may fail if the system is fast and the ASYNC_COMMAND app
  86     //       doesn't manage to create its HWND before our wxKill() is
  87     //       executed; in that case we "fall back" to the second invocation
  88     //       with wxSIGKILL (which should always succeed)
  89     CPPUNIT_ASSERT( wxKill(pid, wxSIGTERM) == 0 ||
  90                     wxKill(pid, wxSIGKILL) == 0 );
  91
  92     // test running COMMAND again, but this time with redirection:
  93     wxArrayString stdout_arr;
  94     CPPUNIT_ASSERT_EQUAL( 0, wxExecute(COMMAND, stdout_arr, wxEXEC_SYNC) );
  95     CPPUNIT_ASSERT_EQUAL( "hi", stdout_arr[0] );
  96 }
  97
  98 void ExecTestCase::TestProcess()
  99 {
 100     // test wxExecute with wxProcess
 101     wxProcess *proc = new wxProcess;
 102     long pid = wxExecute(ASYNC_COMMAND, wxEXEC_ASYNC, proc);
 103     CPPUNIT_ASSERT( proc->GetPid() == pid && pid != 0 );
 104
 105     // we're not going to process the wxEVT_END_PROCESS event,
 106     // so the proc instance will auto-delete itself after we kill
 107     // the asynch process:
 108     CPPUNIT_ASSERT( wxKill(pid, wxSIGTERM) == 0 ||
 109                     wxKill(pid, wxSIGKILL) == 0 );
 110
 111
 112     // test wxExecute with wxProcess and REDIRECTION
 113     wxProcess *proc2 = new wxProcess;
 114     proc2->Redirect();
 115     CPPUNIT_ASSERT( wxExecute(COMMAND, wxEXEC_SYNC, proc2) == 0 );
 116
 117     wxStringOutputStream stdout_stream;
 118     CPPUNIT_ASSERT( proc2->GetInputStream() );
 119     CPPUNIT_ASSERT_EQUAL( wxSTREAM_EOF,
 120         proc2->GetInputStream()->Read(stdout_stream).GetLastError() );
 121
 122     wxString str(stdout_stream.GetString());
 123     CPPUNIT_ASSERT_EQUAL( "hi", str.Trim() );
 124 }
 125