From 028e1f2b8f203e8560861beda60dadb178a2a5ab Mon Sep 17 00:00:00 2001
From: Tatsuhiro Tsujikawa <tatsuhiro.t@gmail.com>
Date: Sun, 12 Feb 2012 18:30:19 +0900
Subject: [PATCH] Added test case where DATA frame is backed off by higher
 priority frame.

---
 tests/main.c                 |  2 +
 tests/spdylay_session_test.c | 74 ++++++++++++++++++++++++++++++++++++
 tests/spdylay_session_test.h |  1 +
 3 files changed, 77 insertions(+)

diff --git a/tests/main.c b/tests/main.c
index 95fa3214..e0a542d7 100644
--- a/tests/main.c
+++ b/tests/main.c
@@ -113,6 +113,8 @@ int main(int argc, char* argv[])
                    test_spdylay_session_on_stream_close) ||
       !CU_add_test(pSuite, "session_max_concurrent_streams",
                    test_spdylay_session_max_concurrent_streams) ||
+      !CU_add_test(pSuite, "session_data_backoff_by_high_pri_frame",
+                   test_spdylay_session_data_backoff_by_high_pri_frame) ||
       !CU_add_test(pSuite, "frame_unpack_nv", test_spdylay_frame_unpack_nv) ||
       !CU_add_test(pSuite, "frame_count_nv_space",
                    test_spdylay_frame_count_nv_space) ||
diff --git a/tests/spdylay_session_test.c b/tests/spdylay_session_test.c
index 5abd3644..3bdd64fe 100644
--- a/tests/spdylay_session_test.c
+++ b/tests/spdylay_session_test.c
@@ -50,9 +50,12 @@ typedef struct {
   accumulator *acc;
   scripted_data_feed *df;
   int ctrl_recv_cb_called, invalid_ctrl_recv_cb_called;
+  int ctrl_send_cb_called;
+  spdylay_frame_type sent_frame_type;
   int stream_close_cb_called;
   size_t data_source_length;
   int32_t stream_id;
+  size_t block_count;
 } my_user_data;
 
 static void scripted_data_feed_init(scripted_data_feed *df,
@@ -124,6 +127,16 @@ static void on_invalid_ctrl_recv_callback(spdylay_session *session,
   ++ud->invalid_ctrl_recv_cb_called;
 }
 
+static void on_ctrl_send_callback(spdylay_session *session,
+                                  spdylay_frame_type type,
+                                  spdylay_frame *frame,
+                                  void *user_data)
+{
+  my_user_data *ud = (my_user_data*)user_data;
+  ++ud->ctrl_send_cb_called;
+  ud->sent_frame_type = type;
+}
+
 static ssize_t fixed_length_data_source_read_callback
 (spdylay_session *session, uint8_t *buf, size_t len, int *eof,
  spdylay_data_source *source, void *user_data)
@@ -1004,3 +1017,64 @@ void test_spdylay_session_max_concurrent_streams()
 
   spdylay_session_del(session);
 }
+
+static ssize_t block_count_send_callback(spdylay_session* session,
+                                         const uint8_t *data, size_t len,
+                                         int flags,
+                                         void *user_data)
+{
+  my_user_data *ud = (my_user_data*)user_data;
+  int r;
+  if(ud->block_count == 0) {
+    r = SPDYLAY_ERR_WOULDBLOCK;
+  } else {
+    --ud->block_count;
+    r = len;
+  }
+  return r;
+}
+
+void test_spdylay_session_data_backoff_by_high_pri_frame()
+{
+  spdylay_session *session;
+  spdylay_session_callbacks callbacks;
+  const char *nv[] = { NULL };
+  my_user_data ud;
+  spdylay_data_provider data_prd;
+  spdylay_stream *stream;
+
+  memset(&callbacks, 0, sizeof(spdylay_session_callbacks));
+  callbacks.send_callback = block_count_send_callback;
+  callbacks.on_ctrl_send_callback = on_ctrl_send_callback;
+  data_prd.read_callback = fixed_length_data_source_read_callback;
+
+  ud.ctrl_send_cb_called = 0;
+  ud.data_source_length = 16*1024;
+
+  spdylay_session_client_new(&session, &callbacks, &ud);
+  spdylay_submit_request(session, 3, nv, &data_prd, NULL);
+
+  ud.block_count = 2;
+  /* Sends SYN_STREAM + DATA[0] */
+  CU_ASSERT(0 == spdylay_session_send(session));
+  CU_ASSERT(SPDYLAY_SYN_STREAM == ud.sent_frame_type);
+  /* data for DATA[1] is read from data_prd but it is not sent */
+  CU_ASSERT(ud.data_source_length == 8*1024);
+
+  spdylay_submit_ping(session);
+  ud.block_count = 2;
+  /* Sends DATA[1] + PING, PING is interleaved in DATA sequence */
+  CU_ASSERT(0 == spdylay_session_send(session));
+  CU_ASSERT(SPDYLAY_PING == ud.sent_frame_type);
+  /* data for DATA[2] is read from data_prd but it is not sent */
+  CU_ASSERT(ud.data_source_length == 4*1024);
+
+  ud.block_count = 2;
+  /* Sends DATA[2..3] */
+  CU_ASSERT(0 == spdylay_session_send(session));
+
+  stream = spdylay_session_get_stream(session, 1);
+  CU_ASSERT(stream->shut_flags & SPDYLAY_SHUT_WR);
+
+  spdylay_session_del(session);
+}
diff --git a/tests/spdylay_session_test.h b/tests/spdylay_session_test.h
index 02ede6b6..083022e4 100644
--- a/tests/spdylay_session_test.h
+++ b/tests/spdylay_session_test.h
@@ -48,5 +48,6 @@ void test_spdylay_session_pop_next_ob_item();
 void test_spdylay_session_on_request_recv_callback();
 void test_spdylay_session_on_stream_close();
 void test_spdylay_session_max_concurrent_streams();
+void test_spdylay_session_data_backoff_by_high_pri_frame();
 
 #endif // SPDYLAY_SESSION_TEST_H