uniconf/t/uniretrygen.t.cc

   1 #include "wvtest.h"
   2 #include "uniconfroot.h"
   3 #include "unitempgen.h"
   4 #include "uniretrygen.h"
   5 #include "uniclientgen.h"
   6 #include "uniconfgen-sanitytest.h"
   7 #include "wvfileutils.h"
   8
   9 #include <sys/types.h>
  10 #include <sys/wait.h>
  11 #include <signal.h>
  12
  13
  14 struct UniRetryGenTester
  15 {
  16     WvString socket;
  17     WvString ini;
  18     pid_t uniconfd_pid;
  19     UniRetryGenTester() :
  20         socket("/tmp/uniretrygen-uniconfd-%s", getpid()),
  21         ini("/tmp/uniretrygen-uniconfd.ini-%s", getpid()),
  22         uniconfd_pid(0)
  23     { }
  24 };
  25
  26
  27 void wait_for_daemon(UniRetryGenTester t)
  28 {
  29     UniConfRoot another_cfg(WvString("retry:unix:%s", t.socket));
  30
  31     for (;;)
  32     {
  33         another_cfg.xset("wait", "pong");
  34         if (another_cfg.xget("wait") == "pong") break;
  35         fprintf(stderr, "still waiting for connection.\n");
  36         wvdelay(100);
  37     }
  38
  39     fprintf(stderr, "managed to connect\n");
  40 }
  41
  42
  43 WVTEST_MAIN("UniRetryGen Sanity Test")
  44 {
  45     UniRetryGen *gen = new UniRetryGen("temp:");
  46     UniConfGenSanityTester::sanity_test(gen, "retry:temp:");
  47     WVRELEASE(gen);
  48 }
  49
  50 WVTEST_MAIN("UniRetryGen: uniconfd")
  51 {
  52     signal(SIGPIPE, SIG_IGN);
  53
  54     UniRetryGenTester t;
  55
  56     unlink(t.ini);
  57
  58     UniConfRoot cfg(WvString("retry:unix:%s 100", t.socket));
  59     cfg["/key"].setme("value");
  60     WVPASS(!cfg["/key"].exists());
  61
  62     {
  63         UniConfTestDaemon daemon(t.socket, WvString("ini:%s", t.ini));
  64         wait_for_daemon(t);
  65
  66         wvdelay(100); // guarantee that the retry interval has expired
  67         cfg["/key"].setme("value");
  68         WVPASSEQ(cfg["/key"].getme(), "value");
  69
  70         cfg.commit();
  71     }
  72
  73     // no delay necessary: we disconnect right away
  74     WVPASS(!cfg["/key"].exists());
  75
  76     {
  77         UniConfTestDaemon daemon(t.socket, WvString("ini:%s", t.ini));
  78         wait_for_daemon(t);
  79
  80         wvdelay(100); // guarantee that the retry interval has expired
  81         WVPASSEQ(cfg["/key"].getme(), "value");
  82
  83         cfg.commit();
  84     }
  85
  86     WVPASS(!cfg["/key"].exists());
  87 }
  88
  89 bool reconnected = false;
  90 void reconnect_cb(UniRetryGen &uni)
  91     { reconnected = true; }
  92
  93 WVTEST_MAIN("UniRetryGen: reconnect callback")
  94 {
  95     signal(SIGPIPE, SIG_IGN);
  96     UniRetryGenTester t;
  97
  98     unlink(t.ini);
  99
 100     UniConfRoot cfg;
 101     cfg.mountgen(new UniRetryGen(WvString("unix:%s", t.socket),
 102                 UniRetryGen::ReconnectCallback(reconnect_cb), 100));
 103
 104     reconnected = false;
 105
 106     UniConfTestDaemon daemon(t.socket, WvString("ini:%s", t.ini));
 107     wait_for_daemon(t);
 108
 109     wvdelay(100); // guarantee that the retry interval has expired
 110     cfg.getme(); // Do something to reconnect
 111     WVPASS(reconnected);
 112 }
 113
 114 WVTEST_MAIN("UniRetryGen: immediate reconnect")
 115 {
 116     signal(SIGPIPE, SIG_IGN);
 117
 118     UniRetryGenTester t;
 119
 120     unlink(t.ini);
 121
 122     // Need to set the reconnect delay to 0 to read immediately
 123     UniConfRoot cfg(WvString("retry:unix:%s 0", t.socket));
 124
 125     {
 126         UniConfTestDaemon daemon(t.socket, WvString("ini:%s", t.ini));
 127         wait_for_daemon(t);
 128
 129         cfg["/key"].setme("value");
 130         WVPASSEQ(cfg["/key"].getme(), "value");
 131
 132         cfg.commit();
 133     }
 134
 135     // don't check anything before restarting so cfg doesn't know that
 136     // uniconfd has disconnected.
 137     {
 138         UniConfTestDaemon daemon(t.socket, WvString("ini:%s", t.ini));
 139         wait_for_daemon(t);
 140
 141         cfg.getme(); // Do something to reconnect
 142         WVPASSEQ(cfg["/key"].getme(), "value");
 143     }
 144
 145     WVPASS(!cfg["/key"].exists());
 146 }
 147
 148
 149 WVTEST_MAIN("UniRetryGen: mount point exists")
 150 {
 151     // bug 9769
 152     UniConfRoot uniconf("temp:");
 153
 154     WVPASS(uniconf["foo"].mount("retry:unix:/tmp/foobar"));
 155
 156     WVPASS(uniconf["foo"].exists());
 157     WVPASSEQ(uniconf["foo"].xget("", WvString::null), "");
 158
 159     WVFAIL(uniconf["foo/bar"].exists());
 160     WVPASSEQ(uniconf["foo"].xget(""), WvString::null);
 161 }
 162