http_ut.cpp 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368
  1. #include <library/cpp/testing/unittest/registar.h>
  2. #include <library/cpp/testing/unittest/tests_data.h>
  3. #include <library/cpp/actors/core/executor_pool_basic.h>
  4. #include <library/cpp/actors/core/scheduler_basic.h>
  5. #include <library/cpp/actors/testlib/test_runtime.h>
  6. #include <util/system/tempfile.h>
  7. #include "http.h"
  8. #include "http_proxy.h"
  9. enum EService : NActors::NLog::EComponent {
  10. MIN,
  11. Logger,
  12. MVP,
  13. MAX
  14. };
  15. namespace {
  16. template <typename HttpType>
  17. void EatWholeString(TIntrusivePtr<HttpType>& request, const TString& data) {
  18. request->EnsureEnoughSpaceAvailable(data.size());
  19. auto size = std::min(request->Avail(), data.size());
  20. memcpy(request->Pos(), data.data(), size);
  21. request->Advance(size);
  22. }
  23. template <typename HttpType>
  24. void EatPartialString(TIntrusivePtr<HttpType>& request, const TString& data) {
  25. for (char c : data) {
  26. request->EnsureEnoughSpaceAvailable(1);
  27. memcpy(request->Pos(), &c, 1);
  28. request->Advance(1);
  29. }
  30. }
  31. }
  32. Y_UNIT_TEST_SUITE(HttpProxy) {
  33. Y_UNIT_TEST(BasicParsing) {
  34. NHttp::THttpIncomingRequestPtr request = new NHttp::THttpIncomingRequest();
  35. EatWholeString(request, "GET /test HTTP/1.1\r\nHost: test\r\nSome-Header: 32344\r\n\r\n");
  36. UNIT_ASSERT_EQUAL(request->Stage, NHttp::THttpIncomingRequest::EParseStage::Done);
  37. UNIT_ASSERT_EQUAL(request->Method, "GET");
  38. UNIT_ASSERT_EQUAL(request->URL, "/test");
  39. UNIT_ASSERT_EQUAL(request->Protocol, "HTTP");
  40. UNIT_ASSERT_EQUAL(request->Version, "1.1");
  41. UNIT_ASSERT_EQUAL(request->Host, "test");
  42. UNIT_ASSERT_EQUAL(request->Headers, "Host: test\r\nSome-Header: 32344\r\n\r\n");
  43. }
  44. Y_UNIT_TEST(BasicParsingChunkedBodyRequest) {
  45. NHttp::THttpIncomingRequestPtr request = new NHttp::THttpIncomingRequest();
  46. EatWholeString(request, "POST /Url HTTP/1.1\r\nConnection: close\r\nTransfer-Encoding: chunked\r\n\r\n4\r\nthis\r\n4\r\n is \r\n5\r\ntest.\r\n0\r\n\r\n");
  47. UNIT_ASSERT_EQUAL(request->Stage, NHttp::THttpIncomingRequest::EParseStage::Done);
  48. UNIT_ASSERT_EQUAL(request->Method, "POST");
  49. UNIT_ASSERT_EQUAL(request->URL, "/Url");
  50. UNIT_ASSERT_EQUAL(request->Connection, "close");
  51. UNIT_ASSERT_EQUAL(request->Protocol, "HTTP");
  52. UNIT_ASSERT_EQUAL(request->Version, "1.1");
  53. UNIT_ASSERT_EQUAL(request->TransferEncoding, "chunked");
  54. UNIT_ASSERT_EQUAL(request->Body, "this is test.");
  55. }
  56. Y_UNIT_TEST(BasicParsingChunkedBodyResponse) {
  57. NHttp::THttpOutgoingRequestPtr request = nullptr; //new NHttp::THttpOutgoingRequest();
  58. NHttp::THttpIncomingResponsePtr response = new NHttp::THttpIncomingResponse(request);
  59. EatWholeString(response, "HTTP/1.1 200 OK\r\nConnection: close\r\nTransfer-Encoding: chunked\r\n\r\n4\r\nthis\r\n4\r\n is \r\n5\r\ntest.\r\n0\r\n\r\n");
  60. UNIT_ASSERT_EQUAL(response->Stage, NHttp::THttpIncomingResponse::EParseStage::Done);
  61. UNIT_ASSERT_EQUAL(response->Status, "200");
  62. UNIT_ASSERT_EQUAL(response->Connection, "close");
  63. UNIT_ASSERT_EQUAL(response->Protocol, "HTTP");
  64. UNIT_ASSERT_EQUAL(response->Version, "1.1");
  65. UNIT_ASSERT_EQUAL(response->TransferEncoding, "chunked");
  66. UNIT_ASSERT_EQUAL(response->Body, "this is test.");
  67. }
  68. Y_UNIT_TEST(InvalidParsingChunkedBody) {
  69. NHttp::THttpOutgoingRequestPtr request = nullptr; //new NHttp::THttpOutgoingRequest();
  70. NHttp::THttpIncomingResponsePtr response = new NHttp::THttpIncomingResponse(request);
  71. EatWholeString(response, "HTTP/1.1 200 OK\r\nConnection: close\r\nTransfer-Encoding: chunked\r\n\r\n5\r\nthis\r\n4\r\n is \r\n5\r\ntest.\r\n0\r\n\r\n");
  72. UNIT_ASSERT(response->IsError());
  73. }
  74. Y_UNIT_TEST(AdvancedParsingChunkedBody) {
  75. NHttp::THttpOutgoingRequestPtr request = nullptr; //new NHttp::THttpOutgoingRequest();
  76. NHttp::THttpIncomingResponsePtr response = new NHttp::THttpIncomingResponse(request);
  77. EatWholeString(response, "HTTP/1.1 200 OK\r\nConnection: close\r\nTransfer-Encoding: chunked\r\n\r\n6\r\nthis\r\n\r\n4\r\n is \r\n5\r\ntest.\r\n0\r\n\r\n");
  78. UNIT_ASSERT_EQUAL(response->Stage, NHttp::THttpIncomingResponse::EParseStage::Done);
  79. UNIT_ASSERT_EQUAL(response->Status, "200");
  80. UNIT_ASSERT_EQUAL(response->Connection, "close");
  81. UNIT_ASSERT_EQUAL(response->Protocol, "HTTP");
  82. UNIT_ASSERT_EQUAL(response->Version, "1.1");
  83. UNIT_ASSERT_EQUAL(response->TransferEncoding, "chunked");
  84. UNIT_ASSERT_EQUAL(response->Body, "this\r\n is test.");
  85. }
  86. Y_UNIT_TEST(CreateResponseWithCompressedBody) {
  87. NHttp::THttpIncomingRequestPtr request = nullptr;
  88. NHttp::THttpOutgoingResponsePtr response = new NHttp::THttpOutgoingResponse(request, "HTTP", "1.1", "200", "OK");
  89. response->Set<&NHttp::THttpResponse::ContentEncoding>("gzip");
  90. TString compressedBody = "compressed body";
  91. response->SetBody(compressedBody);
  92. UNIT_ASSERT_VALUES_EQUAL("gzip", response->ContentEncoding);
  93. UNIT_ASSERT_VALUES_EQUAL(ToString(compressedBody.size()), response->ContentLength);
  94. UNIT_ASSERT_VALUES_EQUAL(compressedBody, response->Body);
  95. }
  96. Y_UNIT_TEST(BasicPartialParsing) {
  97. NHttp::THttpIncomingRequestPtr request = new NHttp::THttpIncomingRequest();
  98. EatPartialString(request, "GET /test HTTP/1.1\r\nHost: test\r\nSome-Header: 32344\r\n\r\n");
  99. UNIT_ASSERT_EQUAL(request->Stage, NHttp::THttpIncomingRequest::EParseStage::Done);
  100. UNIT_ASSERT_EQUAL(request->Method, "GET");
  101. UNIT_ASSERT_EQUAL(request->URL, "/test");
  102. UNIT_ASSERT_EQUAL(request->Protocol, "HTTP");
  103. UNIT_ASSERT_EQUAL(request->Version, "1.1");
  104. UNIT_ASSERT_EQUAL(request->Host, "test");
  105. UNIT_ASSERT_EQUAL(request->Headers, "Host: test\r\nSome-Header: 32344\r\n\r\n");
  106. }
  107. Y_UNIT_TEST(BasicPartialParsingChunkedBody) {
  108. NHttp::THttpOutgoingRequestPtr request = nullptr; //new NHttp::THttpOutgoingRequest();
  109. NHttp::THttpIncomingResponsePtr response = new NHttp::THttpIncomingResponse(request);
  110. EatPartialString(response, "HTTP/1.1 200 OK\r\nConnection: close\r\nTransfer-Encoding: chunked\r\n\r\n4\r\nthis\r\n4\r\n is \r\n5\r\ntest.\r\n0\r\n\r\n");
  111. UNIT_ASSERT_EQUAL(response->Stage, NHttp::THttpIncomingResponse::EParseStage::Done);
  112. UNIT_ASSERT_EQUAL(response->Status, "200");
  113. UNIT_ASSERT_EQUAL(response->Connection, "close");
  114. UNIT_ASSERT_EQUAL(response->Protocol, "HTTP");
  115. UNIT_ASSERT_EQUAL(response->Version, "1.1");
  116. UNIT_ASSERT_EQUAL(response->TransferEncoding, "chunked");
  117. UNIT_ASSERT_EQUAL(response->Body, "this is test.");
  118. }
  119. Y_UNIT_TEST(AdvancedParsing) {
  120. NHttp::THttpIncomingRequestPtr request = new NHttp::THttpIncomingRequest();
  121. EatWholeString(request, "GE");
  122. EatWholeString(request, "T");
  123. EatWholeString(request, " ");
  124. EatWholeString(request, "/test");
  125. EatWholeString(request, " HTTP/1.1\r");
  126. EatWholeString(request, "\nHo");
  127. EatWholeString(request, "st: test");
  128. EatWholeString(request, "\r\n");
  129. EatWholeString(request, "Some-Header: 32344\r\n\r");
  130. EatWholeString(request, "\n");
  131. UNIT_ASSERT_EQUAL(request->Stage, NHttp::THttpIncomingRequest::EParseStage::Done);
  132. UNIT_ASSERT_EQUAL(request->Method, "GET");
  133. UNIT_ASSERT_EQUAL(request->URL, "/test");
  134. UNIT_ASSERT_EQUAL(request->Protocol, "HTTP");
  135. UNIT_ASSERT_EQUAL(request->Version, "1.1");
  136. UNIT_ASSERT_EQUAL(request->Host, "test");
  137. UNIT_ASSERT_EQUAL(request->Headers, "Host: test\r\nSome-Header: 32344\r\n\r\n");
  138. }
  139. Y_UNIT_TEST(AdvancedPartialParsing) {
  140. NHttp::THttpIncomingRequestPtr request = new NHttp::THttpIncomingRequest();
  141. EatPartialString(request, "GE");
  142. EatPartialString(request, "T");
  143. EatPartialString(request, " ");
  144. EatPartialString(request, "/test");
  145. EatPartialString(request, " HTTP/1.1\r");
  146. EatPartialString(request, "\nHo");
  147. EatPartialString(request, "st: test");
  148. EatPartialString(request, "\r\n");
  149. EatPartialString(request, "Some-Header: 32344\r\n\r");
  150. EatPartialString(request, "\n");
  151. UNIT_ASSERT_EQUAL(request->Stage, NHttp::THttpIncomingRequest::EParseStage::Done);
  152. UNIT_ASSERT_EQUAL(request->Method, "GET");
  153. UNIT_ASSERT_EQUAL(request->URL, "/test");
  154. UNIT_ASSERT_EQUAL(request->Protocol, "HTTP");
  155. UNIT_ASSERT_EQUAL(request->Version, "1.1");
  156. UNIT_ASSERT_EQUAL(request->Host, "test");
  157. UNIT_ASSERT_EQUAL(request->Headers, "Host: test\r\nSome-Header: 32344\r\n\r\n");
  158. }
  159. Y_UNIT_TEST(BasicRenderBodyWithHeadersAndCookies) {
  160. NHttp::THttpOutgoingRequestPtr request = NHttp::THttpOutgoingRequest::CreateRequestGet("http://www.yandex.ru/data/url");
  161. NHttp::THeadersBuilder headers;
  162. NHttp::TCookiesBuilder cookies;
  163. cookies.Set("cookie1", "123456");
  164. cookies.Set("cookie2", "45678");
  165. headers.Set("Cookie", cookies.Render());
  166. request->Set(headers);
  167. TString requestData;
  168. request->AsString(requestData);
  169. UNIT_ASSERT_VALUES_EQUAL(requestData, "GET /data/url HTTP/1.1\r\nHost: www.yandex.ru\r\nAccept: */*\r\nCookie: cookie1=123456; cookie2=45678;\r\n");
  170. }
  171. Y_UNIT_TEST(BasicRunning) {
  172. NActors::TTestActorRuntimeBase actorSystem;
  173. TPortManager portManager;
  174. TIpPort port = portManager.GetTcpPort();
  175. TAutoPtr<NActors::IEventHandle> handle;
  176. actorSystem.Initialize();
  177. NActors::IActor* proxy = NHttp::CreateHttpProxy();
  178. NActors::TActorId proxyId = actorSystem.Register(proxy);
  179. actorSystem.Send(new NActors::IEventHandle(proxyId, TActorId(), new NHttp::TEvHttpProxy::TEvAddListeningPort(port)), 0, true);
  180. actorSystem.DispatchEvents();
  181. NActors::TActorId serverId = actorSystem.AllocateEdgeActor();
  182. actorSystem.Send(new NActors::IEventHandle(proxyId, serverId, new NHttp::TEvHttpProxy::TEvRegisterHandler("/test", serverId)), 0, true);
  183. NActors::TActorId clientId = actorSystem.AllocateEdgeActor();
  184. NHttp::THttpOutgoingRequestPtr httpRequest = NHttp::THttpOutgoingRequest::CreateRequestGet("http://[::1]:" + ToString(port) + "/test");
  185. actorSystem.Send(new NActors::IEventHandle(proxyId, clientId, new NHttp::TEvHttpProxy::TEvHttpOutgoingRequest(httpRequest)), 0, true);
  186. NHttp::TEvHttpProxy::TEvHttpIncomingRequest* request = actorSystem.GrabEdgeEvent<NHttp::TEvHttpProxy::TEvHttpIncomingRequest>(handle);
  187. UNIT_ASSERT_EQUAL(request->Request->URL, "/test");
  188. NHttp::THttpOutgoingResponsePtr httpResponse = request->Request->CreateResponseString("HTTP/1.1 200 Found\r\nConnection: Close\r\nTransfer-Encoding: chunked\r\n\r\n6\r\npassed\r\n0\r\n\r\n");
  189. actorSystem.Send(new NActors::IEventHandle(handle->Sender, serverId, new NHttp::TEvHttpProxy::TEvHttpOutgoingResponse(httpResponse)), 0, true);
  190. NHttp::TEvHttpProxy::TEvHttpIncomingResponse* response = actorSystem.GrabEdgeEvent<NHttp::TEvHttpProxy::TEvHttpIncomingResponse>(handle);
  191. UNIT_ASSERT_EQUAL(response->Response->Status, "200");
  192. UNIT_ASSERT_EQUAL(response->Response->Body, "passed");
  193. }
  194. Y_UNIT_TEST(TlsRunning) {
  195. NActors::TTestActorRuntimeBase actorSystem;
  196. TPortManager portManager;
  197. TIpPort port = portManager.GetTcpPort();
  198. TAutoPtr<NActors::IEventHandle> handle;
  199. actorSystem.Initialize();
  200. TString certificateContent = R"___(-----BEGIN PRIVATE KEY-----
  201. MIIEvwIBADANBgkqhkiG9w0BAQEFAASCBKkwggSlAgEAAoIBAQCzRZjodO7Aqe1w
  202. RyOj6kG6g2nn8ZGAxfao4mLT0jDTbVksrhV/h2s3uldLkFo5WrNQ8WZe+iIbXeFL
  203. s8tO6hslzreo9sih2IHoRcH5KnS/6YTqVhRTJb1jE2dM8NwYbwTi+T2Pe0FrBPjI
  204. kgVO50gAtYl9C+fc715uZiSKW+rRlP5OoFTwxrOjiU27RPZjFYyWK9wTI1Es9uRr
  205. lbZbLl5cY6dK2J1AViRraaYKCWO26VbOPWLsY4OD3e+ZXIc3OMCz6Yb0wmRPeJ60
  206. bbbkGfI8O27kDdv69MAWHIm0yYMzKEnom1dce7rNQNDEqJfocsYIsg+EvayT1yQ9
  207. KTBegw7LAgMBAAECggEBAKaOCrotqYQmXArsjRhFFDwMy+BKdzyEr93INrlFl0dX
  208. WHpCYobRcbOc1G3H94tB0UdqgAnNqtJyLlb+++ydZAuEOu4oGc8EL+10ofq0jzOd
  209. 6Xct8kQt0/6wkFDTlii9PHUDy0X65ZRgUiNGRtg/2I2QG+SpowmI+trm2xwQueFs
  210. VaWrjc3cVvXx0b8Lu7hqZUv08kgC38stzuRk/n2T5VWSAr7Z4ZWQbO918Dv35HUw
  211. Wy/0jNUFP9CBCvFJ4l0OoH9nYhWFG+HXWzNdw6/Hca4jciRKo6esCiOZ9uWYv/ec
  212. /NvX9rgFg8G8/SrTisX10+Bbeq+R1RKwq/IG409TH4ECgYEA14L+3QsgNIUMeYAx
  213. jSCyk22R/tOHI1BM+GtKPUhnwHlAssrcPcxXMJovl6WL93VauYjym0wpCz9urSpA
  214. I2CqTsG8GYciA6Dr3mHgD6cK0jj9UPAU6EnZ5S0mjhPqKZqutu9QegzD2uESvuN8
  215. 36xezwQthzAf0nI/P3sJGjVXjikCgYEA1POm5xcV6SmM6HnIdadEebhzZIJ9TXQz
  216. ry3Jj3a7CKyD5C7fAdkHUTCjgT/2ElxPi9ABkZnC+d/cW9GtJFa0II5qO/agm3KQ
  217. ZXYiutu9A7xACHYFXRiJEjVUdGG9dKMVOHUEa8IHEgrrcUVM/suy/GgutywIfaXs
  218. y58IFP24K9MCgYEAk6zjz7wL+XEiNy+sxLQfKf7vB9sSwxQHakK6wHuY/L8Zomp3
  219. uLEJHfjJm/SIkK0N2g0JkXkCtv5kbKyC/rsCeK0wo52BpVLjzaLr0k34kE0U6B1b
  220. dkEE2pGx1bG3x4KDLj+Wuct9ecK5Aa0IqIyI+vo16GkFpUM8K9e3SQo8UOECgYEA
  221. sCZYAkILYtJ293p9giz5rIISGasDAUXE1vxWBXEeJ3+kneTTnZCrx9Im/ewtnWR0
  222. fF90XL9HFDDD88POqAd8eo2zfKR2l/89SGBfPBg2EtfuU9FkgGyiPciVcqvC7q9U
  223. B15saMKX3KnhtdGwbfeLt9RqCCTJZT4SUSDcq5hwdvcCgYAxY4Be8mNipj8Cgg22
  224. mVWSolA0TEzbtUcNk6iGodpi+Z0LKpsPC0YRqPRyh1K+rIltG1BVdmUBHcMlOYxl
  225. lWWvbJH6PkJWy4n2MF7PO45kjN3pPZg4hgH63JjZeAineBwEArUGb9zHnvzcdRvF
  226. wuQ2pZHL/HJ0laUSieHDJ5917w==
  227. -----END PRIVATE KEY-----
  228. -----BEGIN CERTIFICATE-----
  229. MIIDjTCCAnWgAwIBAgIURt5IBx0J3xgEaQvmyrFH2A+NkpMwDQYJKoZIhvcNAQEL
  230. BQAwVjELMAkGA1UEBhMCUlUxDzANBgNVBAgMBk1vc2NvdzEPMA0GA1UEBwwGTW9z
  231. Y293MQ8wDQYDVQQKDAZZYW5kZXgxFDASBgNVBAMMC3Rlc3Qtc2VydmVyMB4XDTE5
  232. MDkyMDE3MTQ0MVoXDTQ3MDIwNDE3MTQ0MVowVjELMAkGA1UEBhMCUlUxDzANBgNV
  233. BAgMBk1vc2NvdzEPMA0GA1UEBwwGTW9zY293MQ8wDQYDVQQKDAZZYW5kZXgxFDAS
  234. BgNVBAMMC3Rlc3Qtc2VydmVyMIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKC
  235. AQEAs0WY6HTuwKntcEcjo+pBuoNp5/GRgMX2qOJi09Iw021ZLK4Vf4drN7pXS5Ba
  236. OVqzUPFmXvoiG13hS7PLTuobJc63qPbIodiB6EXB+Sp0v+mE6lYUUyW9YxNnTPDc
  237. GG8E4vk9j3tBawT4yJIFTudIALWJfQvn3O9ebmYkilvq0ZT+TqBU8Mazo4lNu0T2
  238. YxWMlivcEyNRLPbka5W2Wy5eXGOnStidQFYka2mmCgljtulWzj1i7GODg93vmVyH
  239. NzjAs+mG9MJkT3ietG225BnyPDtu5A3b+vTAFhyJtMmDMyhJ6JtXXHu6zUDQxKiX
  240. 6HLGCLIPhL2sk9ckPSkwXoMOywIDAQABo1MwUTAdBgNVHQ4EFgQUDv/xuJ4CvCgG
  241. fPrZP3hRAt2+/LwwHwYDVR0jBBgwFoAUDv/xuJ4CvCgGfPrZP3hRAt2+/LwwDwYD
  242. VR0TAQH/BAUwAwEB/zANBgkqhkiG9w0BAQsFAAOCAQEAinKpMYaA2tjLpAnPVbjy
  243. /ZxSBhhB26RiQp3Re8XOKyhTWqgYE6kldYT0aXgK9x9mPC5obQannDDYxDc7lX+/
  244. qP/u1X81ZcDRo/f+qQ3iHfT6Ftt/4O3qLnt45MFM6Q7WabRm82x3KjZTqpF3QUdy
  245. tumWiuAP5DMd1IRDtnKjFHO721OsEsf6NLcqdX89bGeqXDvrkwg3/PNwTyW5E7cj
  246. feY8L2eWtg6AJUnIBu11wvfzkLiH3QKzHvO/SIZTGf5ihDsJ3aKEE9UNauTL3bVc
  247. CRA/5XcX13GJwHHj6LCoc3sL7mt8qV9HKY2AOZ88mpObzISZxgPpdKCfjsrdm63V
  248. 6g==
  249. -----END CERTIFICATE-----)___";
  250. TTempFileHandle certificateFile;
  251. certificateFile.Write(certificateContent.data(), certificateContent.size());
  252. NActors::IActor* proxy = NHttp::CreateHttpProxy();
  253. NActors::TActorId proxyId = actorSystem.Register(proxy);
  254. THolder<NHttp::TEvHttpProxy::TEvAddListeningPort> add = MakeHolder<NHttp::TEvHttpProxy::TEvAddListeningPort>(port);
  255. ///////// https configuration
  256. add->Secure = true;
  257. add->CertificateFile = certificateFile.Name();
  258. add->PrivateKeyFile = certificateFile.Name();
  259. /////////
  260. actorSystem.Send(new NActors::IEventHandle(proxyId, TActorId(), add.Release()), 0, true);
  261. actorSystem.DispatchEvents();
  262. NActors::TActorId serverId = actorSystem.AllocateEdgeActor();
  263. actorSystem.Send(new NActors::IEventHandle(proxyId, serverId, new NHttp::TEvHttpProxy::TEvRegisterHandler("/test", serverId)), 0, true);
  264. NActors::TActorId clientId = actorSystem.AllocateEdgeActor();
  265. NHttp::THttpOutgoingRequestPtr httpRequest = NHttp::THttpOutgoingRequest::CreateRequestGet("https://[::1]:" + ToString(port) + "/test");
  266. actorSystem.Send(new NActors::IEventHandle(proxyId, clientId, new NHttp::TEvHttpProxy::TEvHttpOutgoingRequest(httpRequest)), 0, true);
  267. NHttp::TEvHttpProxy::TEvHttpIncomingRequest* request = actorSystem.GrabEdgeEvent<NHttp::TEvHttpProxy::TEvHttpIncomingRequest>(handle);
  268. UNIT_ASSERT_EQUAL(request->Request->URL, "/test");
  269. NHttp::THttpOutgoingResponsePtr httpResponse = request->Request->CreateResponseString("HTTP/1.1 200 Found\r\nConnection: Close\r\nTransfer-Encoding: chunked\r\n\r\n6\r\npassed\r\n0\r\n\r\n");
  270. actorSystem.Send(new NActors::IEventHandle(handle->Sender, serverId, new NHttp::TEvHttpProxy::TEvHttpOutgoingResponse(httpResponse)), 0, true);
  271. NHttp::TEvHttpProxy::TEvHttpIncomingResponse* response = actorSystem.GrabEdgeEvent<NHttp::TEvHttpProxy::TEvHttpIncomingResponse>(handle);
  272. UNIT_ASSERT_EQUAL(response->Response->Status, "200");
  273. UNIT_ASSERT_EQUAL(response->Response->Body, "passed");
  274. }
  275. /*Y_UNIT_TEST(AdvancedRunning) {
  276. THolder<NActors::TActorSystemSetup> setup = MakeHolder<NActors::TActorSystemSetup>();
  277. setup->NodeId = 1;
  278. setup->ExecutorsCount = 1;
  279. setup->Executors = new TAutoPtr<NActors::IExecutorPool>[1];
  280. setup->Executors[0] = new NActors::TBasicExecutorPool(0, 2, 10);
  281. setup->Scheduler = new NActors::TBasicSchedulerThread(NActors::TSchedulerConfig(512, 100));
  282. NActors::TActorSystem actorSystem(setup);
  283. actorSystem.Start();
  284. NHttp::THttpProxy* incomingProxy = new NHttp::THttpProxy();
  285. NActors::TActorId incomingProxyId = actorSystem.Register(incomingProxy);
  286. actorSystem.Send(incomingProxyId, new NHttp::TEvHttpProxy::TEvAddListeningPort(13337));
  287. NHttp::THttpProxy* outgoingProxy = new NHttp::THttpProxy();
  288. NActors::TActorId outgoingProxyId = actorSystem.Register(outgoingProxy);
  289. THolder<NHttp::THttpStaticStringRequest> httpRequest = MakeHolder<NHttp::THttpStaticStringRequest>("GET /test HTTP/1.1\r\n\r\n");
  290. actorSystem.Send(outgoingProxyId, new NHttp::TEvHttpProxy::TEvHttpOutgoingRequest("[::]:13337", std::move(httpRequest)));
  291. Sleep(TDuration::Minutes(60));
  292. }*/
  293. Y_UNIT_TEST(TooLongHeader) {
  294. NActors::TTestActorRuntimeBase actorSystem;
  295. TPortManager portManager;
  296. TIpPort port = portManager.GetTcpPort();
  297. TAutoPtr<NActors::IEventHandle> handle;
  298. actorSystem.Initialize();
  299. NActors::IActor* proxy = NHttp::CreateHttpProxy();
  300. NActors::TActorId proxyId = actorSystem.Register(proxy);
  301. actorSystem.Send(new NActors::IEventHandle(proxyId, TActorId(), new NHttp::TEvHttpProxy::TEvAddListeningPort(port)), 0, true);
  302. actorSystem.DispatchEvents();
  303. NActors::TActorId serverId = actorSystem.AllocateEdgeActor();
  304. actorSystem.Send(new NActors::IEventHandle(proxyId, serverId, new NHttp::TEvHttpProxy::TEvRegisterHandler("/test", serverId)), 0, true);
  305. NActors::TActorId clientId = actorSystem.AllocateEdgeActor();
  306. NHttp::THttpOutgoingRequestPtr httpRequest = NHttp::THttpOutgoingRequest::CreateRequestGet("http://[::1]:" + ToString(port) + "/test");
  307. httpRequest->Set("Connection", "close");
  308. TString longHeader;
  309. longHeader.append(9000, 'X');
  310. httpRequest->Set(longHeader, "data");
  311. actorSystem.Send(new NActors::IEventHandle(proxyId, clientId, new NHttp::TEvHttpProxy::TEvHttpOutgoingRequest(httpRequest)), 0, true);
  312. NHttp::TEvHttpProxy::TEvHttpIncomingResponse* response = actorSystem.GrabEdgeEvent<NHttp::TEvHttpProxy::TEvHttpIncomingResponse>(handle);
  313. UNIT_ASSERT_EQUAL(response->Response->Status, "400");
  314. UNIT_ASSERT_EQUAL(response->Response->Body, "Invalid http header");
  315. }
  316. }