src/gromacs/commandline/tests/cmdlineparser.cpp

   1 /*
   2  *
   3  *                This source code is part of
   4  *
   5  *                 G   R   O   M   A   C   S
   6  *
   7  *          GROningen MAchine for Chemical Simulations
   8  *
   9  * Written by David van der Spoel, Erik Lindahl, Berk Hess, and others.
  10  * Copyright (c) 1991-2000, University of Groningen, The Netherlands.
  11  * Copyright (c) 2001-2009, The GROMACS development team,
  12  * check out http://www.gromacs.org for more information.
  13
  14  * This program is free software; you can redistribute it and/or
  15  * modify it under the terms of the GNU General Public License
  16  * as published by the Free Software Foundation; either version 2
  17  * of the License, or (at your option) any later version.
  18  *
  19  * If you want to redistribute modifications, please consider that
  20  * scientific software is very special. Version control is crucial -
  21  * bugs must be traceable. We will be happy to consider code for
  22  * inclusion in the official distribution, but derived work must not
  23  * be called official GROMACS. Details are found in the README & COPYING
  24  * files - if they are missing, get the official version at www.gromacs.org.
  25  *
  26  * To help us fund GROMACS development, we humbly ask that you cite
  27  * the papers on the package - you can find them in the top README file.
  28  *
  29  * For more info, check our website at http://www.gromacs.org
  30  */
  31 /*! \internal \file
  32  * \brief
  33  * Tests gmx::CommandLineParser.
  34  *
  35  * These tests exercise a large fraction of the code, so they may
  36  * catch errors in other parts than just in command-line parsing.
  37  *
  38  * \author Teemu Murtola <teemu.murtola@cbr.su.se>
  39  * \ingroup module_commandline
  40  */
  41 #include <cstdlib>
  42 #include <cstring>
  43 #include <vector>
  44
  45 #include <gtest/gtest.h>
  46
  47 #include "gromacs/commandline/cmdlineparser.h"
  48 #include "gromacs/options/basicoptions.h"
  49 #include "gromacs/options/options.h"
  50
  51 namespace
  52 {
  53
  54 class CommandLineParserTest : public ::testing::Test
  55 {
  56     public:
  57         CommandLineParserTest();
  58         ~CommandLineParserTest();
  59
  60         void createArguments(const char *cmdline[]);
  61
  62         gmx::Options            _options;
  63         gmx::CommandLineParser  _parser;
  64         bool                    _flag;
  65         std::vector<int>        _ivalues;
  66         std::vector<double>     _dvalues;
  67         int                     _argc;
  68         char                  **_argv;
  69 };
  70
  71 CommandLineParserTest::CommandLineParserTest()
  72     : _options(NULL, NULL), _parser(&_options),
  73       _flag(false),
  74       _argc(0), _argv(NULL)
  75 {
  76     using gmx::BooleanOption;
  77     using gmx::IntegerOption;
  78     using gmx::DoubleOption;
  79     _options.addOption(BooleanOption("flag").store(&_flag));
  80     _options.addOption(IntegerOption("mvi").storeVector(&_ivalues).multiValue());
  81     _options.addOption(DoubleOption("mvd").storeVector(&_dvalues).allowMultiple());
  82 }
  83
  84 CommandLineParserTest::~CommandLineParserTest()
  85 {
  86     if (_argv != NULL)
  87     {
  88         for (int i = 0; i < _argc; ++i)
  89         {
  90             free(_argv[i]);
  91         }
  92     }
  93     delete [] _argv;
  94 }
  95
  96 void CommandLineParserTest::createArguments(const char *cmdline[])
  97 {
  98     _argc = 0;
  99     while (cmdline[_argc] != NULL) ++_argc;
 100     ++_argc;
 101
 102     _argv = new char *[_argc];
 103     _argv[0] = strdup("test");
 104     for (int i = 1; i < _argc; ++i)
 105     {
 106         _argv[i] = strdup(cmdline[i - 1]);
 107     }
 108 }
 109
 110 TEST_F(CommandLineParserTest, HandlesSingleValues)
 111 {
 112     const char *cmdline[] = {"-flag", "yes", "-mvi", "2", "-mvd", "2.7", NULL};
 113     createArguments(cmdline);
 114     ASSERT_NO_THROW(_parser.parse(&_argc, _argv));
 115     ASSERT_NO_THROW(_options.finish());
 116
 117     EXPECT_TRUE(_flag);
 118     ASSERT_EQ(1U, _ivalues.size());
 119     EXPECT_EQ(2, _ivalues[0]);
 120     ASSERT_EQ(1U, _dvalues.size());
 121     EXPECT_DOUBLE_EQ(2.7, _dvalues[0]);
 122 }
 123
 124 TEST_F(CommandLineParserTest, HandlesNegativeNumbers)
 125 {
 126     const char *cmdline[] = {"-mvi", "1", "-2", "-mvd", "-2.7", NULL};
 127     createArguments(cmdline);
 128     ASSERT_NO_THROW(_parser.parse(&_argc, _argv));
 129     ASSERT_NO_THROW(_options.finish());
 130
 131     ASSERT_EQ(2U, _ivalues.size());
 132     EXPECT_EQ(1, _ivalues[0]);
 133     EXPECT_EQ(-2, _ivalues[1]);
 134     ASSERT_EQ(1U, _dvalues.size());
 135     EXPECT_DOUBLE_EQ(-2.7, _dvalues[0]);
 136 }
 137
 138 } // namespace