contrib/atf/atf-c++/detail/application_test.cpp

   1 //
   2 // Automated Testing Framework (atf)
   3 //
   4 // Copyright (c) 2009 The NetBSD Foundation, Inc.
   5 // All rights reserved.
   6 //
   7 // Redistribution and use in source and binary forms, with or without
   8 // modification, are permitted provided that the following conditions
   9 // are met:
  10 // 1. Redistributions of source code must retain the above copyright
  11 //    notice, this list of conditions and the following disclaimer.
  12 // 2. Redistributions in binary form must reproduce the above copyright
  13 //    notice, this list of conditions and the following disclaimer in the
  14 //    documentation and/or other materials provided with the distribution.
  15 //
  16 // THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND
  17 // CONTRIBUTORS ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
  18 // INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
  19 // MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  20 // IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS BE LIABLE FOR ANY
  21 // DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  22 // DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
  23 // GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  24 // INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
  25 // IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
  26 // OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
  27 // IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  28 //
  29
  30 extern "C" {
  31 #include <unistd.h>
  32 }
  33
  34 #include "application.hpp"
  35
  36 #include "../macros.hpp"
  37
  38 class getopt_app : public atf::application::app {
  39 public:
  40     getopt_app(void) : app("description", "manpage", "other") {}
  41
  42     int main(void)
  43     {
  44         // Provide an option that is unknown to the application driver and
  45         // one that is, together with an argument that would be swallowed by
  46         // the test program option if it were recognized.
  47         int argc = 4;
  48         char arg1[] = "progname";
  49         char arg2[] = "-Z";
  50         char arg3[] = "-s";
  51         char arg4[] = "foo";
  52         char *const argv[] = { arg1, arg2, arg3, arg4, NULL };
  53
  54         int ch;
  55         bool zflag;
  56
  57         // Given that this obviously is an application, and that we used the
  58         // same driver to start, we can test getopt(3) right here without doing
  59         // any fancy stuff.
  60         zflag = false;
  61         while ((ch = ::getopt(argc, argv, ":Z")) != -1) {
  62             switch (ch) {
  63             case 'Z':
  64                 zflag = true;
  65                 break;
  66
  67             case '?':
  68             default:
  69                 if (optopt != 's')
  70                     ATF_FAIL("Unexpected unknown option found");
  71             }
  72         }
  73
  74         ATF_REQUIRE(zflag);
  75         ATF_REQUIRE_EQ(1, argc - optind);
  76         ATF_REQUIRE_EQ(std::string("foo"), argv[optind]);
  77
  78         return 0;
  79     }
  80 };
  81
  82 ATF_TEST_CASE_WITHOUT_HEAD(getopt);
  83 ATF_TEST_CASE_BODY(getopt)
  84 {
  85     int argc = 1;
  86     char arg1[] = "progname";
  87     char *const argv[] = { arg1, NULL };
  88     ATF_REQUIRE_EQ(0, getopt_app().run(argc, argv));
  89 }
  90
  91 ATF_INIT_TEST_CASES(tcs)
  92 {
  93     ATF_ADD_TEST_CASE(tcs, getopt);
  94 }