1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
3 * Data Differential YATL (i.e. libtest) library
5 * Copyright (C) 2012 Data Differential, http://datadifferential.com/
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions are
11 * * Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
14 * * Redistributions in binary form must reproduce the above
15 * copyright notice, this list of conditions and the following disclaimer
16 * in the documentation and/or other materials provided with the
19 * * The names of its contributors may not be used to endorse or
20 * promote products derived from this software without specific prior
23 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
24 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
25 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
26 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
27 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
28 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
29 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
30 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
31 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
32 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
33 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
38 #include <libtest/common.h>
40 #include <libtest/drizzled.h>
42 #include "util/instance.hpp"
43 #include "util/operation.hpp"
45 using namespace datadifferential
;
46 using namespace libtest
;
56 #include <sys/types.h>
60 #ifndef __INTEL_COMPILER
61 #pragma GCC diagnostic ignored "-Wold-style-cast"
64 #if defined(HAVE_LIBDRIZZLE) && HAVE_LIBDRIZZLE
65 #include <libdrizzle-1.0/drizzle_client.h>
68 using namespace libtest
;
71 bool ping_drizzled(const in_port_t _port
)
74 #if defined(HAVE_LIBDRIZZLE) && HAVE_LIBDRIZZLE
76 drizzle_st
*drizzle
= drizzle_create(NULL
);
85 if ((con
= drizzle_con_create(drizzle
, NULL
)) == NULL
)
87 drizzle_free(drizzle
);
91 drizzle_con_set_tcp(con
, "localhost", _port
);
92 drizzle_con_set_auth(con
, "root", 0);
97 if ((rc
= drizzle_con_connect(con
)) == DRIZZLE_RETURN_OK
)
99 drizzle_result_st
*result
= drizzle_ping(con
, NULL
, &rc
);
100 success
= bool(result
);
101 drizzle_result_free(result
);
106 else if (rc
!= DRIZZLE_RETURN_OK
)
108 Error
<< drizzle_error(drizzle
) << " localhost:" << _port
;
111 drizzle_con_free(con
);
112 drizzle_free(drizzle
);
120 } // namespace libtest
122 class Drizzle
: public libtest::Server
126 Drizzle(const std::string
& host_arg
, in_port_t port_arg
) :
127 libtest::Server(host_arg
, port_arg
, DRIZZLED_BINARY
, false)
135 while (_app
.check() and --limit
)
137 if (ping_drizzled(_port
))
141 libtest::dream(1, 0);
152 void log_file_option(Application
&, const std::string
&)
156 bool has_log_file_option() const
161 bool broken_pid_file()
171 bool has_syslog() const
176 bool has_port_option() const
181 void port_option(Application
& app
, in_port_t arg
)
186 snprintf(buffer
, sizeof(buffer
), "--drizzle-protocol.port=%d", int(arg
));
187 app
.add_option(buffer
);
191 bool build(size_t argc
, const char *argv
[]);
194 bool Drizzle::build(size_t argc
, const char *argv
[])
196 if (getuid() == 0 or geteuid() == 0)
198 add_option("--user=root");
201 add_option("--verbose=INSPECT");
203 add_option("--datadir=var/drizzle");
206 for (size_t x
= 0 ; x
< argc
; x
++)
216 libtest::Server
*build_drizzled(const char *hostname
, in_port_t try_port
)
218 return new Drizzle(hostname
, try_port
);