Branch data     Line data    Source code

       1                 :            : /* SPDX-License-Identifier: LGPL-2.1+ */
       2                 :            : 
       3                 :            : #include <getopt.h>
       4                 :            : 
       5                 :            : #include "macro.h"
       6                 :            : #include "strv.h"
       7                 :            : #include "verbs.h"
       8                 :            : 
       9                 :         12 : static int noop_dispatcher(int argc, char *argv[], void *userdata) {
      10                 :         12 :         return 0;
      11                 :            : }
      12                 :            : 
      13                 :            : #define test_dispatch_one(argv, verbs, expected) \
      14                 :            :         optind = 0; \
      15                 :            :         assert_se(dispatch_verb(strv_length(argv), argv, verbs, NULL) == expected);
      16                 :            : 
      17                 :          4 : static void test_verbs(void) {
      18                 :            :         static const Verb verbs[] = {
      19                 :            :                 { "help",        VERB_ANY, VERB_ANY, 0,            noop_dispatcher },
      20                 :            :                 { "list-images", VERB_ANY, 1,        0,            noop_dispatcher },
      21                 :            :                 { "list",        VERB_ANY, 2,        VERB_DEFAULT, noop_dispatcher },
      22                 :            :                 { "status",      2,        VERB_ANY, 0,            noop_dispatcher },
      23                 :            :                 { "show",        VERB_ANY, VERB_ANY, 0,            noop_dispatcher },
      24                 :            :                 { "terminate",   2,        VERB_ANY, 0,            noop_dispatcher },
      25                 :            :                 { "login",       2,        2,        0,            noop_dispatcher },
      26                 :            :                 { "copy-to",     3,        4,        0,            noop_dispatcher },
      27                 :            :                 {}
      28                 :            :         };
      29                 :            : 
      30                 :            :         /* not found */
      31         [ -  + ]:          4 :         test_dispatch_one(STRV_MAKE("command-not-found"), verbs, -EINVAL);
      32                 :            : 
      33                 :            :         /* found */
      34         [ -  + ]:          4 :         test_dispatch_one(STRV_MAKE("show"), verbs, 0);
      35                 :            : 
      36                 :            :         /* found, too few args */
      37         [ -  + ]:          4 :         test_dispatch_one(STRV_MAKE("copy-to", "foo"), verbs, -EINVAL);
      38                 :            : 
      39                 :            :         /* found, meets min args */
      40         [ -  + ]:          4 :         test_dispatch_one(STRV_MAKE("status", "foo", "bar"), verbs, 0);
      41                 :            : 
      42                 :            :         /* found, too many args */
      43         [ -  + ]:          4 :         test_dispatch_one(STRV_MAKE("copy-to", "foo", "bar", "baz", "quux", "qaax"), verbs, -EINVAL);
      44                 :            : 
      45                 :            :         /* no verb, but a default is set */
      46         [ -  + ]:          4 :         test_dispatch_one(STRV_MAKE_EMPTY, verbs, 0);
      47                 :          4 : }
      48                 :            : 
      49                 :          4 : static void test_verbs_no_default(void) {
      50                 :            :         static const Verb verbs[] = {
      51                 :            :                 { "help", VERB_ANY, VERB_ANY, 0, noop_dispatcher },
      52                 :            :                 {},
      53                 :            :         };
      54                 :            : 
      55         [ -  + ]:          4 :         test_dispatch_one(STRV_MAKE(NULL), verbs, -EINVAL);
      56                 :          4 : }
      57                 :            : 
      58                 :          4 : int main(int argc, char *argv[]) {
      59                 :          4 :         test_verbs();
      60                 :          4 :         test_verbs_no_default();
      61                 :            : 
      62                 :          4 :         return 0;
      63                 :            : }