unit: Support for SNEP REJECT request

2 files changed, 21 insertions, 7 deletions

diff --git a/src/snep.c b/src/snep.c
index 4db47a5..e1bbe81 100644
--- a/src/snep.c
+++ b/src/snep.c
@@ -425,6 +425,25 @@ static near_bool_t snep_core_process_request(int client_fd,
 		}
 		break;
 
+	case NEAR_SNEP_REQ_REJECT:
+		DBG("NEAR_SNEP_REQ_REJECT");
+		if (snep_data->req->fragments == NULL) {
+			near_error("error: NEAR_SNEP_REQ_REJECT but no fragment");
+			ret = FALSE;
+		}
+		else {
+			ret = TRUE;
+		}
+
+		g_slist_free_full(snep_data->req->fragments,
+						free_snep_core_fragment);
+		g_slist_free(snep_data->req->fragments);
+
+		g_hash_table_remove(snep_client_hash,
+						GINT_TO_POINTER(client_fd));
+
+		break;
+
 	case NEAR_SNEP_REQ_CONTINUE:
 		/*
 		 * NEAR_SNEP_REQ_CONTINUE indicates that we have to send the
diff --git a/unit/test-snep-read.c b/unit/test-snep-read.c
index c008376..9ce2c1c 100644
--- a/unit/test-snep-read.c
+++ b/unit/test-snep-read.c
@@ -772,13 +772,8 @@ static void test_snep_read_get_req_frags_client_resp(gpointer context,
 	g_free(req);
 
 	if (client_resp == NEAR_SNEP_REQ_REJECT) {
-		/*
-		 * TODO server shall not send any response:
-		 *      g_assert(ret);
-		 *      test_snep_read_no_response();
-		 */
-
-		TEST_SNEP_LOG("EXPECTED FAIL: REJECT not handled by SNEP\n");
+		g_assert(ret);
+		test_snep_read_no_response();
 	} else if (client_resp == NEAR_SNEP_REQ_CONTINUE) {
 		g_assert(ret);