diff options
| -rw-r--r-- | tests/unittests/CMakeLists.txt | 1 | ||||
| -rw-r--r-- | tests/unittests/torture_options.c | 158 |
2 files changed, 159 insertions, 0 deletions
diff --git a/tests/unittests/CMakeLists.txt b/tests/unittests/CMakeLists.txt index b836ed8b..2d0a8607 100644 --- a/tests/unittests/CMakeLists.txt +++ b/tests/unittests/CMakeLists.txt @@ -3,3 +3,4 @@ project(unittests C) add_check_test(torture_list torture_list.c ${TORTURE_LIBRARY}) add_check_test(torture_misc torture_misc.c ${TORTURE_LIBRARY}) add_check_test(torture_keyfiles torture_keyfiles.c ${TORTURE_LIBRARY}) +add_check_test(torture_options torture_options.c ${TORTURE_LIBRARY}) diff --git a/tests/unittests/torture_options.c b/tests/unittests/torture_options.c new file mode 100644 index 00000000..380896fc --- /dev/null +++ b/tests/unittests/torture_options.c @@ -0,0 +1,158 @@ +#define LIBSSH_STATIC + +#include "torture.h" +#include "options.c" + +ssh_session session; + +static void setup(void) { + session = ssh_new(); +} + +static void teardown(void) { + ssh_free(session); +} + +START_TEST (torture_options_set_host) +{ + int rc; + + rc = ssh_options_set(session, SSH_OPTIONS_HOST, "localhost"); + ck_assert(rc == 0); + ck_assert_str_eq(session->host, "localhost"); + + rc = ssh_options_set(session, SSH_OPTIONS_HOST, "guru@meditation"); + ck_assert(rc == 0); + ck_assert_str_eq(session->host, "meditation"); + ck_assert_str_eq(session->username, "guru"); +} +END_TEST + +START_TEST (torture_options_set_port) +{ + int rc; + unsigned int port = 42; + + rc = ssh_options_set(session, SSH_OPTIONS_PORT, &port); + ck_assert(rc == 0); + ck_assert(session->port == port); + + rc = ssh_options_set(session, SSH_OPTIONS_PORT_STR, "23"); + ck_assert(rc == 0); + ck_assert(session->port == 23); + + rc = ssh_options_set(session, SSH_OPTIONS_PORT_STR, "five"); + ck_assert(rc == 0); + ck_assert(session->port == 0); + + rc = ssh_options_set(session, SSH_OPTIONS_PORT, NULL); + ck_assert(rc == 0); + ck_assert(session->port == 22); +} +END_TEST + +START_TEST (torture_options_set_fd) +{ + socket_t fd = 42; + int rc; + + rc = ssh_options_set(session, SSH_OPTIONS_FD, &fd); + ck_assert(rc == 0); + ck_assert(session->fd == fd); + + rc = ssh_options_set(session, SSH_OPTIONS_FD, NULL); + ck_assert(rc == 0); + ck_assert(session->fd == -1); +} +END_TEST + +START_TEST (torture_options_set_user) +{ + int rc; +#ifndef _WIN32 + struct passwd pwd; + struct passwd *pwdbuf; + char buf[NSS_BUFLEN_PASSWD]; +#endif + + rc = ssh_options_set(session, SSH_OPTIONS_USER, "guru"); + ck_assert(rc == 0); + ck_assert_str_eq(session->username, "guru"); + + + rc = ssh_options_set(session, SSH_OPTIONS_USER, NULL); + ck_assert(rc == 0); + +#ifndef _WIN32 + /* get local username */ + rc = getpwuid_r(getuid(), &pwd, buf, NSS_BUFLEN_PASSWD, &pwdbuf); + ck_assert(rc == 0); + + ck_assert_str_eq(session->username, pwd.pw_name); +#endif +} +END_TEST + +/* TODO */ +#if 0 +START_TEST (torture_options_set_sshdir) +{ +} +END_TEST +#endif + +START_TEST (torture_options_set_identity) +{ + int rc; + + rc = ssh_options_set(session, SSH_OPTIONS_ADD_IDENTITY, "identity1"); + ck_assert(rc == 0); + ck_assert_str_eq(session->identity->root->data, "identity1"); + + rc = ssh_options_set(session, SSH_OPTIONS_IDENTITY, "identity2"); + ck_assert(rc == 0); + ck_assert_str_eq(session->identity->root->data, "identity2"); + ck_assert_str_eq(session->identity->root->next->data, "identity1"); +} +END_TEST + +static Suite *torture_make_suite(void) { + Suite *s = suite_create("libssh_keyfiles"); + + torture_create_case_fixture(s, "torture_options_set_host", + torture_options_set_host, setup, teardown); + torture_create_case_fixture(s, "torture_options_set_port", + torture_options_set_port, setup, teardown); + torture_create_case_fixture(s, "torture_options_set_fd", + torture_options_set_fd, setup, teardown); + torture_create_case_fixture(s, "torture_options_set_user", + torture_options_set_user, setup, teardown); + torture_create_case_fixture(s, "torture_options_set_identity", + torture_options_set_identity, setup, teardown); + + return s; +} + +int main(int argc, char **argv) { + Suite *s = NULL; + SRunner *sr = NULL; + struct argument_s arguments; + int nf; + + ZERO_STRUCT(arguments); + + torture_cmdline_parse(argc, argv, &arguments); + + s = torture_make_suite(); + + sr = srunner_create(s); + if (arguments.nofork) { + srunner_set_fork_status(sr, CK_NOFORK); + } + srunner_run_all(sr, CK_VERBOSE); + nf = srunner_ntests_failed(sr); + srunner_free(sr); + + return (nf == 0) ? EXIT_SUCCESS : EXIT_FAILURE; +} + |