kwonhosts: Add functions to check if servers public key is known

Signed-off-by: Andreas Schneider <asn@cryptomilk.org>

1 files changed, 20 insertions, 0 deletions

diff --git a/tests/client/torture_knownhosts_verify.c b/tests/client/torture_knownhosts_verify.c
index a266e92a..323b8e2b 100644
--- a/tests/client/torture_knownhosts_verify.c
+++ b/tests/client/torture_knownhosts_verify.c
@@ -103,12 +103,32 @@ static void torture_knownhosts_export(void **state)
     SAFE_FREE(entry);
 }
 
+static void torture_knownhosts_write_and_verify(void **state)
+{
+    struct torture_state *s = *state;
+    ssh_session session = s->ssh.session;
+    enum ssh_known_hosts_e found;
+    int rc;
+
+    rc = ssh_connect(session);
+    assert_int_equal(rc, SSH_OK);
+
+    rc = ssh_session_update_known_hosts(session);
+    assert_int_equal(rc, SSH_OK);
+
+    found = ssh_session_is_known_server(session);
+    assert_int_equal(found, SSH_KNOWN_HOSTS_OK);
+}
+
 int torture_run_tests(void) {
     int rc;
     struct CMUnitTest tests[] = {
         cmocka_unit_test_setup_teardown(torture_knownhosts_export,
                                         session_setup,
                                         session_teardown),
+        cmocka_unit_test_setup_teardown(torture_knownhosts_write_and_verify,
+                                        session_setup,
+                                        session_teardown),
     };
 
     ssh_init();