From 72058bfbcc25d85c5fa13a5aba7d1e5a3ef60670 Mon Sep 17 00:00:00 2001
From: Andreas Schneider <mail@cynapses.org>
Date: Sat, 13 Mar 2010 14:21:23 +0100
Subject: Added tests for ssh options.

---
 tests/unittests/CMakeLists.txt    |   1 +
 tests/unittests/torture_options.c | 158 ++++++++++++++++++++++++++++++++++++++
 2 files changed, 159 insertions(+)
 create mode 100644 tests/unittests/torture_options.c

(limited to 'tests')

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;
+}
+
-- 
cgit v1.2.3