tests: include gnutls_ext_s/get_data into tests of mini-extension

diff --git a/tests/mini-extension.c b/tests/mini-extension.c
index 3a7e3e23638d9746e5bebfe50239c5d28227e535..80095edfd8be69491ffe9c641e8b46fdde6bb71c 100644 (file)
--- a/tests/mini-extension.c
+++ b/tests/mini-extension.c
@@ -82,7 +82,9 @@ static int ext_recv_client_params(gnutls_session_t session, const unsigned char
                fail("ext_recv_client_params: Invalid input buffer data\n");
 
        TLSEXT_TYPE_client_received = 1;
-       
+
+       gnutls_ext_set_data(session, TLSEXT_TYPE_SAMPLE, session);
+
        return 0; //Success
 }
 
@@ -118,6 +120,7 @@ static void client(int sd)
        int ret;
        gnutls_session_t session;
        gnutls_certificate_credentials_t clientx509cred;
+       void *p;
 
        global_init();
        gnutls_global_set_log_function(tls_log_func);
@@ -161,6 +164,15 @@ static void client(int sd)
        if (TLSEXT_TYPE_client_sent != 1 || TLSEXT_TYPE_client_received != 1)
                fail("client: extension not properly sent/received\n");
 
+       ret = gnutls_ext_get_data(session, TLSEXT_TYPE_SAMPLE, &p);
+       if (ret < 0) {
+               fail("gnutls_ext_get_data: %s\n", gnutls_strerror(ret));
+       }
+
+       if (p != session) {
+               fail("client: gnutls_ext_get_data failed\n");
+       }
+
        gnutls_bye(session, GNUTLS_SHUT_RDWR);
 
 end: