1
0
mirror of https://github.com/samba-team/samba.git synced 2024-12-22 13:34:15 +03:00

ctdb-tests: Add test for ctdb_io.c

BUG: https://bugzilla.samba.org/show_bug.cgi?id=13791

Signed-off-by: Christof Schmitt <cs@samba.org>
Reviewed-by: Martin Schwenke <martin@meltin.net>

Autobuild-User(master): Martin Schwenke <martins@samba.org>
Autobuild-Date(master): Fri Feb 22 03:51:37 CET 2019 on sn-devel-144
This commit is contained in:
Christof Schmitt 2019-02-19 13:59:05 -07:00 committed by Martin Schwenke
parent fa8e69ac95
commit 92a9052437
3 changed files with 210 additions and 0 deletions

View File

@ -0,0 +1,8 @@
#!/bin/sh
. "${TEST_SCRIPTS_DIR}/unit.sh"
ok_null
unit_test ctdb_io_test 1
unit_test ctdb_io_test 2

View File

@ -0,0 +1,196 @@
/*
ctdb_io tests
Copyright (C) Christof Schmitt 2019
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, see <http://www.gnu.org/licenses/>.
*/
#include "replace.h"
#include "system/filesys.h"
#include <assert.h>
#include "common/ctdb_io.c"
void ctdb_set_error(struct ctdb_context *ctdb, const char *fmt, ...)
{
va_list ap;
va_start(ap, fmt);
vprintf(fmt, ap);
assert(false);
}
static void test_setup(ctdb_queue_cb_fn_t cb,
int *pfd,
struct ctdb_context **pctdb)
{
int pipefd[2], ret;
struct ctdb_context *ctdb;
struct ctdb_queue *queue;
ret = pipe(pipefd);
assert(ret == 0);
ctdb = talloc_zero(NULL, struct ctdb_context);
assert(ctdb != NULL);
ctdb->ev = tevent_context_init(NULL);
queue = ctdb_queue_setup(ctdb, ctdb, pipefd[0], 0, cb,
NULL, "test queue");
assert(queue != NULL);
*pctdb = ctdb;
*pfd = pipefd[1];
}
static const size_t test1_req_len = 8;
static const char *test1_req = "abcdefgh";
static void test1_callback(uint8_t *data, size_t length, void *private_data)
{
uint32_t len;
len = *(uint32_t *)data;
assert(len == sizeof(uint32_t) + test1_req_len);
assert(length == sizeof(uint32_t) + test1_req_len);
assert(memcmp(data + sizeof(len), test1_req, test1_req_len) == 0);
}
static void test1(void)
{
struct ctdb_context *ctdb;
int fd, ret;
uint32_t pkt_size;
test_setup(test1_callback, &fd, &ctdb);
pkt_size = sizeof(uint32_t) + test1_req_len;
ret = write(fd, &pkt_size, sizeof(pkt_size));
assert(ret == sizeof(pkt_size));
ret = write(fd, test1_req, test1_req_len);
assert(ret == test1_req_len);
tevent_loop_once(ctdb->ev);
TALLOC_FREE(ctdb);
}
static const size_t test2_req_len[] = { 900, 24, 600 };
static int test2_cb_num = 0;
static void test2_callback(uint8_t *data, size_t length, void *private_data)
{
uint32_t len;
len = *(uint32_t *)data;
assert(len == sizeof(uint32_t) + test2_req_len[test2_cb_num]);
assert(length == sizeof(uint32_t) + test2_req_len[test2_cb_num]);
test2_cb_num++;
}
static void test2(void)
{
struct ctdb_context *ctdb;
int fd, ret, i;
uint32_t pkt_size;
char req[1024] = { 0 };
for (i = 0; i < sizeof(req); i++) {
req[i] = i % CHAR_MAX;
}
test_setup(test2_callback, &fd, &ctdb);
/*
* request 0
*/
pkt_size = sizeof(uint32_t) + test2_req_len[0];
ret = write(fd, &pkt_size, sizeof(pkt_size));
assert(ret == sizeof(pkt_size));
ret = write(fd, req, test2_req_len[0]);
assert(ret == test2_req_len[0]);
/*
* request 1
*/
pkt_size = sizeof(uint32_t) + test2_req_len[1];
ret = write(fd, &pkt_size, sizeof(pkt_size));
assert(ret == sizeof(pkt_size));
/*
* Omit the last byte to avoid buffer processing.
*/
ret = write(fd, req, test2_req_len[1] - 1);
assert(ret == test2_req_len[1] - 1);
tevent_loop_once(ctdb->ev);
/*
* Write the missing byte now.
*/
ret = write(fd, &req[test2_req_len[1] - 1], 1);
assert(ret == 1);
/*
* request 2
*/
pkt_size = sizeof(uint32_t) + test2_req_len[2];
ret = write(fd, &pkt_size, sizeof(pkt_size));
assert(ret == sizeof(pkt_size));
ret = write(fd, req, test2_req_len[2]);
assert(ret == test2_req_len[2]);
tevent_loop_once(ctdb->ev);
tevent_loop_once(ctdb->ev);
assert(test2_cb_num == 2);
TALLOC_FREE(ctdb);
}
int main(int argc, const char **argv)
{
int num;
if (argc != 2) {
fprintf(stderr, "%s <testnum>\n", argv[0]);
exit(1);
}
num = atoi(argv[1]);
switch (num) {
case 1:
test1();
break;
case 2:
test2();
break;
default:
fprintf(stderr, "Unknown test number %s\n", argv[1]);
}
return 0;
}

View File

@ -943,6 +943,12 @@ def build(bld):
LIBASYNC_REQ samba-util sys_rw''',
install_path='${CTDB_TEST_LIBEXECDIR}')
bld.SAMBA_BINARY('ctdb_io_test',
source='tests/src/ctdb_io_test.c',
deps='''talloc tevent tdb samba-util sys_rw''',
install_path='${CTDB_TEST_LIBEXECDIR}')
bld.SAMBA_SUBSYSTEM('ctdb-protocol-tests-basic',
source=bld.SUBDIR('tests/src',
'protocol_common_basic.c'),