From af3de262b69f44ba84009b9aba644b5f1cf3b1a7 Mon Sep 17 00:00:00 2001
From: Andreas Schneider <asn@cryptomilk.org>
Date: Fri, 11 Sep 2015 16:26:53 +0200
Subject: [PATCH] tests: Migrate torture_sftp_dir to cwrap test

Signed-off-by: Andreas Schneider <asn@cryptomilk.org>
---
 tests/client/CMakeLists.txt     |  7 +++-
 tests/client/torture_sftp_dir.c | 74 ++++++++++++++++++++-------------
 2 files changed, 52 insertions(+), 29 deletions(-)

diff --git a/tests/client/CMakeLists.txt b/tests/client/CMakeLists.txt
index e4289d1f..3598b430 100644
--- a/tests/client/CMakeLists.txt
+++ b/tests/client/CMakeLists.txt
@@ -4,7 +4,6 @@ find_package(socket_wrapper)
 
 if (WITH_SFTP)
     add_cmocka_test(torture_sftp_static torture_sftp_static.c ${TORTURE_LIBRARY})
-    add_cmocka_test(torture_sftp_dir torture_sftp_dir.c ${TORTURE_LIBRARY})
     add_cmocka_test(torture_sftp_read torture_sftp_read.c ${TORTURE_LIBRARY})
 endif (WITH_SFTP)
 
@@ -18,6 +17,12 @@ set(LIBSSH_CLIENT_TESTS
     torture_session
     torture_request_env)
 
+if (WITH_SFTP)
+    set(LIBSSH_CLIENT_TESTS
+        ${LIBSSH_CLIENT_TESTS}
+        torture_sftp_dir)
+endif (WITH_SFTP)
+
 foreach(_CLI_TEST ${LIBSSH_CLIENT_TESTS})
     add_cmocka_test(${_CLI_TEST} ${_CLI_TEST}.c ${TORTURE_LIBRARY})
 
diff --git a/tests/client/torture_sftp_dir.c b/tests/client/torture_sftp_dir.c
index e5b5369a..c4de2f60 100644
--- a/tests/client/torture_sftp_dir.c
+++ b/tests/client/torture_sftp_dir.c
@@ -3,44 +3,59 @@
 #include "torture.h"
 #include "sftp.c"
 
-static int setup(void **state) {
-    ssh_session session;
-    struct torture_sftp *t;
-    const char *host;
-    const char *user;
-    const char *password;
+#include <sys/types.h>
+#include <pwd.h>
 
-    host = getenv("TORTURE_HOST");
-    if (host == NULL) {
-        host = "localhost";
-    }
-
-    user = getenv("TORTURE_USER");
-    password = getenv("TORTURE_PASSWORD");
-
-    session = torture_ssh_session(host, NULL, user, password);
-    assert_false(session == NULL);
-    t = torture_sftp_session(session);
-    assert_false(t == NULL);
-
-    *state = t;
+static int sshd_setup(void **state)
+{
+    torture_setup_sshd_server(state);
 
     return 0;
 }
 
-static int teardown(void **state) {
-    struct torture_sftp *t = *state;
+static int sshd_teardown(void **state) {
+    torture_teardown_sshd_server(state);
 
-    assert_false(t == NULL);
+    return 0;
+}
 
-    torture_rmdirs(t->testdir);
-    torture_sftp_close(t);
+static int session_setup(void **state)
+{
+    struct torture_state *s = *state;
+    struct passwd *pwd;
+
+    pwd = getpwnam("bob");
+    assert_non_null(pwd);
+    setuid(pwd->pw_uid);
+
+    s->ssh.session = torture_ssh_session(TORTURE_SSH_SERVER,
+                                         NULL,
+                                         TORTURE_SSH_USER_ALICE,
+                                         NULL);
+    assert_non_null(s->ssh.session);
+
+    s->ssh.tsftp = torture_sftp_session(s->ssh.session);
+    assert_non_null(s->ssh.tsftp);
+
+    return 0;
+}
+
+static int session_teardown(void **state)
+{
+    struct torture_state *s = *state;
+
+    torture_rmdirs(s->ssh.tsftp->testdir);
+    torture_sftp_close(s->ssh.tsftp);
+    ssh_disconnect(s->ssh.session);
+    ssh_free(s->ssh.session);
 
     return 0;
 }
 
 static void torture_sftp_mkdir(void **state) {
-    struct torture_sftp *t = *state;
+    struct torture_state *s = *state;
+
+    struct torture_sftp *t = s->ssh.tsftp;
     char tmpdir[128] = {0};
     int rc;
 
@@ -66,13 +81,16 @@ static void torture_sftp_mkdir(void **state) {
 int torture_run_tests(void) {
     int rc;
     struct CMUnitTest tests[] = {
-        cmocka_unit_test_setup_teardown(torture_sftp_mkdir, setup, teardown)
+        cmocka_unit_test_setup_teardown(torture_sftp_mkdir,
+                                        session_setup,
+                                        session_teardown)
     };
 
     ssh_init();
 
     torture_filter_tests(tests);
-    rc = cmocka_run_group_tests(tests, NULL, NULL);
+    rc = cmocka_run_group_tests(tests, sshd_setup, sshd_teardown);
+
     ssh_finalize();
 
     return rc;