mock start scan response

diff --git a/tests/kds_s2000w_net_tests.c b/tests/kds_s2000w_net_tests.c
index 5b97ce9e10042bac3a7efff8e18a7fb75ec458a0..a7348b0434758f86616f176285951b04bb2a8ca3 100644 (file)
--- a/tests/kds_s2000w_net_tests.c
+++ b/tests/kds_s2000w_net_tests.c
@@ -8,6 +8,17 @@
 #include "../src/kds_s2000w_handler.h"
 #include "../src/kds_s2000w_image_converter.h"
 
+void mock_response(response* resp)
+{
+       response* mock_resp = mock_type(response*);
+       if (mock_resp->size > 0) {
+               resp->data = realloc(resp->data, mock_resp->size);
+               memcpy(resp->data, mock_resp->data, mock_resp->size);
+       }
+       resp->code = mock_resp->code;
+       resp->size = mock_resp->size;
+}
+
 void __wrap_load_config(program_config* config, const char* config_stream)
 {
        config->scanner_url = malloc(sizeof(char) * 28);
@@ -128,6 +139,7 @@ int __wrap_kds_s2000w_client_delete_image(int64_t sessionid, int img_number, res
 
 int __wrap_kds_s2000w_client_start_scan(int64_t sessionid, response* response)
 {
+       mock_response(response);
        return mock_type(int);
 }
 
@@ -579,6 +591,9 @@ void sane_kds_s2000w_net_start_cancel()
 
 void sane_kds_s2000w_net_start()
 {
+       response* resp = kds_s2000w_client_response_init();
+       resp->code = 200;
+       will_return(mock_response, resp);
        will_return(__wrap_kds_s2000w_client_start_scan, 0);
        will_return(__wrap_kds_s2000w_client_status_session, 0);
        will_return(__wrap_kds_s2000w_client_get_image, 0);
@@ -601,6 +616,8 @@ void sane_kds_s2000w_net_start()
 
        json_object_put(h->current_scanner_config);
        h->current_scanner_config = NULL;
+       kds_s2000w_client_response_free(resp);
+       resp = NULL;
        free_handler(h);
        h = NULL;
 }