123 lines
3.1 KiB
C
123 lines
3.1 KiB
C
#include <cmocka.h>
|
|
#include <db/db.h>
|
|
#include <errno.h>
|
|
#include <stdarg.h>
|
|
#include <stddef.h>
|
|
#include <stdint.h>
|
|
#include <string.h>
|
|
#include <sys/socket.h>
|
|
|
|
#include <client.h>
|
|
#include <file_p.h>
|
|
#include <main.h>
|
|
#include <server.h>
|
|
|
|
int _errno = 0;
|
|
int _exit_code = 0;
|
|
extern void *print_str_arr;
|
|
|
|
int __wrap_socket(int __domain, int __type, int __protocol) {
|
|
errno = _errno;
|
|
return mock_type(int);
|
|
}
|
|
|
|
int __wrap_setsockopt(int __fd, int __level, int __optname,
|
|
const void *__optval, socklen_t __optlen) {
|
|
errno = _errno;
|
|
return mock_type(int);
|
|
}
|
|
|
|
int __wrap_bind(int __fd, __CONST_SOCKADDR_ARG __addr, socklen_t __len) {
|
|
errno = _errno;
|
|
return mock_type(int);
|
|
}
|
|
|
|
int __wrap_listen(int __fd, int __n) {
|
|
errno = _errno;
|
|
return mock_type(int);
|
|
}
|
|
|
|
void __wrap_exit(int __status) { _exit_code = __status; }
|
|
|
|
void test__start_server_success(void **state) {
|
|
will_return(__wrap_socket, 1);
|
|
will_return(__wrap_setsockopt, 0);
|
|
will_return(__wrap_bind, 0);
|
|
will_return(__wrap_listen, 0);
|
|
int ret = start_server();
|
|
assert_int_equal(ret, 1);
|
|
}
|
|
|
|
void test__start_server_err_socket(void **state) {
|
|
_errno = EACCES;
|
|
will_return(__wrap_socket, -1);
|
|
|
|
will_return(__wrap_setsockopt, 0);
|
|
will_return(__wrap_bind, 0);
|
|
will_return(__wrap_listen, 0);
|
|
|
|
const char *fmt_args[] = {"starting, socket: %s\n", strerror(_errno), NULL};
|
|
print_str_arr = fmt_args;
|
|
start_server();
|
|
assert_int_equal(_exit_code, 3);
|
|
}
|
|
|
|
void test__start_server_err_setsockopt(void **state) {
|
|
_errno = EBADF;
|
|
will_return(__wrap_socket, 1);
|
|
will_return(__wrap_setsockopt, -1);
|
|
will_return(__wrap_bind, 0);
|
|
will_return(__wrap_listen, 0);
|
|
|
|
const char *fmt_args[] = {"starting, setsockopt: %s\n", strerror(_errno),
|
|
NULL};
|
|
print_str_arr = fmt_args;
|
|
start_server();
|
|
assert_int_equal(_exit_code, 4);
|
|
}
|
|
|
|
void test__start_server_err_bind(void **state) {
|
|
_errno = EBADF;
|
|
will_return(__wrap_socket, 1);
|
|
will_return(__wrap_setsockopt,0);
|
|
will_return(__wrap_bind, -1);
|
|
will_return(__wrap_listen, 0);
|
|
|
|
const char *fmt_args[] = {"starting, socket bind: %s\n", strerror(_errno),
|
|
NULL};
|
|
print_str_arr = fmt_args;
|
|
start_server();
|
|
assert_int_equal(_exit_code, 5);
|
|
}
|
|
|
|
void test__start_server_err_listen(void **state) {
|
|
_errno = EBADF;
|
|
will_return(__wrap_socket, 1);
|
|
will_return(__wrap_setsockopt,0);
|
|
will_return(__wrap_bind, 0);
|
|
will_return(__wrap_listen, -1);
|
|
|
|
const char *fmt_args[] = {"starting, socket listen: %s\n", strerror(_errno),
|
|
NULL};
|
|
print_str_arr = fmt_args;
|
|
start_server();
|
|
assert_int_equal(_exit_code, 6);
|
|
}
|
|
|
|
int setup(void **state) {
|
|
print_str_arr = NULL;
|
|
return 0;
|
|
}
|
|
int tear_down(void **state) { return 0; }
|
|
|
|
int main(int argc, char **argv) {
|
|
const struct CMUnitTest tests[] = {
|
|
cmocka_unit_test(test__start_server_success),
|
|
cmocka_unit_test(test__start_server_err_socket),
|
|
cmocka_unit_test(test__start_server_err_bind),
|
|
cmocka_unit_test(test__start_server_err_listen),
|
|
};
|
|
|
|
return cmocka_run_group_tests(tests, setup, tear_down);
|
|
}
|