fluentd222 1.16.2-x86_64-linux

Sign up to get free protection for your applications and to get access to all the features.
Files changed (562) hide show
  1. checksums.yaml +7 -0
  2. data/.deepsource.toml +13 -0
  3. data/.github/ISSUE_TEMPLATE/bug_report.yaml +71 -0
  4. data/.github/ISSUE_TEMPLATE/config.yml +5 -0
  5. data/.github/ISSUE_TEMPLATE/feature_request.yaml +39 -0
  6. data/.github/ISSUE_TEMPLATE.md +17 -0
  7. data/.github/PULL_REQUEST_TEMPLATE.md +14 -0
  8. data/.github/workflows/stale-actions.yml +24 -0
  9. data/.github/workflows/test-ruby-head.yaml +31 -0
  10. data/.github/workflows/test.yaml +32 -0
  11. data/.gitignore +30 -0
  12. data/ADOPTERS.md +5 -0
  13. data/AUTHORS +2 -0
  14. data/CHANGELOG.md +2720 -0
  15. data/CONTRIBUTING.md +45 -0
  16. data/GOVERNANCE.md +55 -0
  17. data/Gemfile +9 -0
  18. data/GithubWorkflow.md +78 -0
  19. data/LICENSE +202 -0
  20. data/MAINTAINERS.md +13 -0
  21. data/README.md +75 -0
  22. data/Rakefile +79 -0
  23. data/SECURITY.md +14 -0
  24. data/bin/fluent-binlog-reader +7 -0
  25. data/bin/fluent-ca-generate +6 -0
  26. data/bin/fluent-cap-ctl +7 -0
  27. data/bin/fluent-cat +5 -0
  28. data/bin/fluent-ctl +7 -0
  29. data/bin/fluent-debug +5 -0
  30. data/bin/fluent-gem +9 -0
  31. data/bin/fluent-plugin-config-format +5 -0
  32. data/bin/fluent-plugin-generate +5 -0
  33. data/bin/fluentd +15 -0
  34. data/code-of-conduct.md +3 -0
  35. data/docs/SECURITY_AUDIT.pdf +0 -0
  36. data/example/copy_roundrobin.conf +39 -0
  37. data/example/counter.conf +18 -0
  38. data/example/filter_stdout.conf +22 -0
  39. data/example/in_forward.conf +14 -0
  40. data/example/in_forward_client.conf +37 -0
  41. data/example/in_forward_shared_key.conf +15 -0
  42. data/example/in_forward_tls.conf +14 -0
  43. data/example/in_forward_users.conf +24 -0
  44. data/example/in_forward_workers.conf +21 -0
  45. data/example/in_http.conf +16 -0
  46. data/example/in_out_forward.conf +17 -0
  47. data/example/in_sample_blocks.conf +17 -0
  48. data/example/in_sample_with_compression.conf +23 -0
  49. data/example/in_syslog.conf +15 -0
  50. data/example/in_tail.conf +14 -0
  51. data/example/in_tcp.conf +13 -0
  52. data/example/in_udp.conf +13 -0
  53. data/example/logevents.conf +25 -0
  54. data/example/multi_filters.conf +61 -0
  55. data/example/out_copy.conf +20 -0
  56. data/example/out_exec_filter.conf +42 -0
  57. data/example/out_file.conf +13 -0
  58. data/example/out_forward.conf +35 -0
  59. data/example/out_forward_buf_file.conf +23 -0
  60. data/example/out_forward_client.conf +109 -0
  61. data/example/out_forward_heartbeat_none.conf +16 -0
  62. data/example/out_forward_sd.conf +17 -0
  63. data/example/out_forward_shared_key.conf +36 -0
  64. data/example/out_forward_tls.conf +18 -0
  65. data/example/out_forward_users.conf +65 -0
  66. data/example/out_null.conf +36 -0
  67. data/example/sd.yaml +8 -0
  68. data/example/secondary_file.conf +42 -0
  69. data/example/suppress_config_dump.conf +7 -0
  70. data/example/v0_12_filter.conf +78 -0
  71. data/example/v1_literal_example.conf +36 -0
  72. data/example/worker_section.conf +36 -0
  73. data/fluent.conf +139 -0
  74. data/fluentd.gemspec +54 -0
  75. data/lib/fluent/agent.rb +168 -0
  76. data/lib/fluent/capability.rb +87 -0
  77. data/lib/fluent/clock.rb +66 -0
  78. data/lib/fluent/command/binlog_reader.rb +244 -0
  79. data/lib/fluent/command/bundler_injection.rb +45 -0
  80. data/lib/fluent/command/ca_generate.rb +184 -0
  81. data/lib/fluent/command/cap_ctl.rb +174 -0
  82. data/lib/fluent/command/cat.rb +365 -0
  83. data/lib/fluent/command/ctl.rb +180 -0
  84. data/lib/fluent/command/debug.rb +103 -0
  85. data/lib/fluent/command/fluentd.rb +374 -0
  86. data/lib/fluent/command/plugin_config_formatter.rb +308 -0
  87. data/lib/fluent/command/plugin_generator.rb +365 -0
  88. data/lib/fluent/compat/call_super_mixin.rb +76 -0
  89. data/lib/fluent/compat/detach_process_mixin.rb +33 -0
  90. data/lib/fluent/compat/exec_util.rb +129 -0
  91. data/lib/fluent/compat/file_util.rb +54 -0
  92. data/lib/fluent/compat/filter.rb +68 -0
  93. data/lib/fluent/compat/formatter.rb +111 -0
  94. data/lib/fluent/compat/formatter_utils.rb +85 -0
  95. data/lib/fluent/compat/handle_tag_and_time_mixin.rb +62 -0
  96. data/lib/fluent/compat/handle_tag_name_mixin.rb +53 -0
  97. data/lib/fluent/compat/input.rb +49 -0
  98. data/lib/fluent/compat/output.rb +721 -0
  99. data/lib/fluent/compat/output_chain.rb +60 -0
  100. data/lib/fluent/compat/parser.rb +310 -0
  101. data/lib/fluent/compat/parser_utils.rb +40 -0
  102. data/lib/fluent/compat/propagate_default.rb +62 -0
  103. data/lib/fluent/compat/record_filter_mixin.rb +34 -0
  104. data/lib/fluent/compat/set_tag_key_mixin.rb +50 -0
  105. data/lib/fluent/compat/set_time_key_mixin.rb +69 -0
  106. data/lib/fluent/compat/socket_util.rb +165 -0
  107. data/lib/fluent/compat/string_util.rb +34 -0
  108. data/lib/fluent/compat/structured_format_mixin.rb +26 -0
  109. data/lib/fluent/compat/type_converter.rb +90 -0
  110. data/lib/fluent/config/basic_parser.rb +123 -0
  111. data/lib/fluent/config/configure_proxy.rb +424 -0
  112. data/lib/fluent/config/dsl.rb +152 -0
  113. data/lib/fluent/config/element.rb +265 -0
  114. data/lib/fluent/config/error.rb +44 -0
  115. data/lib/fluent/config/literal_parser.rb +286 -0
  116. data/lib/fluent/config/parser.rb +107 -0
  117. data/lib/fluent/config/section.rb +272 -0
  118. data/lib/fluent/config/types.rb +249 -0
  119. data/lib/fluent/config/v1_parser.rb +192 -0
  120. data/lib/fluent/config/yaml_parser/fluent_value.rb +47 -0
  121. data/lib/fluent/config/yaml_parser/loader.rb +108 -0
  122. data/lib/fluent/config/yaml_parser/parser.rb +166 -0
  123. data/lib/fluent/config/yaml_parser/section_builder.rb +107 -0
  124. data/lib/fluent/config/yaml_parser.rb +56 -0
  125. data/lib/fluent/config.rb +89 -0
  126. data/lib/fluent/configurable.rb +201 -0
  127. data/lib/fluent/counter/base_socket.rb +44 -0
  128. data/lib/fluent/counter/client.rb +297 -0
  129. data/lib/fluent/counter/error.rb +86 -0
  130. data/lib/fluent/counter/mutex_hash.rb +163 -0
  131. data/lib/fluent/counter/server.rb +273 -0
  132. data/lib/fluent/counter/store.rb +205 -0
  133. data/lib/fluent/counter/validator.rb +145 -0
  134. data/lib/fluent/counter.rb +23 -0
  135. data/lib/fluent/daemon.rb +13 -0
  136. data/lib/fluent/daemonizer.rb +88 -0
  137. data/lib/fluent/engine.rb +253 -0
  138. data/lib/fluent/env.rb +40 -0
  139. data/lib/fluent/error.rb +37 -0
  140. data/lib/fluent/event.rb +330 -0
  141. data/lib/fluent/event_router.rb +315 -0
  142. data/lib/fluent/ext_monitor_require.rb +28 -0
  143. data/lib/fluent/file_wrapper.rb +137 -0
  144. data/lib/fluent/filter.rb +21 -0
  145. data/lib/fluent/fluent_log_event_router.rb +139 -0
  146. data/lib/fluent/formatter.rb +23 -0
  147. data/lib/fluent/input.rb +21 -0
  148. data/lib/fluent/label.rb +46 -0
  149. data/lib/fluent/load.rb +34 -0
  150. data/lib/fluent/log/console_adapter.rb +66 -0
  151. data/lib/fluent/log.rb +752 -0
  152. data/lib/fluent/match.rb +187 -0
  153. data/lib/fluent/mixin.rb +31 -0
  154. data/lib/fluent/msgpack_factory.rb +111 -0
  155. data/lib/fluent/oj_options.rb +61 -0
  156. data/lib/fluent/output.rb +29 -0
  157. data/lib/fluent/output_chain.rb +23 -0
  158. data/lib/fluent/parser.rb +23 -0
  159. data/lib/fluent/plugin/bare_output.rb +104 -0
  160. data/lib/fluent/plugin/base.rb +214 -0
  161. data/lib/fluent/plugin/buf_file.rb +242 -0
  162. data/lib/fluent/plugin/buf_file_single.rb +254 -0
  163. data/lib/fluent/plugin/buf_memory.rb +34 -0
  164. data/lib/fluent/plugin/buffer/chunk.rb +240 -0
  165. data/lib/fluent/plugin/buffer/file_chunk.rb +413 -0
  166. data/lib/fluent/plugin/buffer/file_single_chunk.rb +310 -0
  167. data/lib/fluent/plugin/buffer/memory_chunk.rb +91 -0
  168. data/lib/fluent/plugin/buffer.rb +941 -0
  169. data/lib/fluent/plugin/compressable.rb +96 -0
  170. data/lib/fluent/plugin/exec_util.rb +22 -0
  171. data/lib/fluent/plugin/file_util.rb +22 -0
  172. data/lib/fluent/plugin/filter.rb +127 -0
  173. data/lib/fluent/plugin/filter_grep.rb +189 -0
  174. data/lib/fluent/plugin/filter_parser.rb +130 -0
  175. data/lib/fluent/plugin/filter_record_transformer.rb +324 -0
  176. data/lib/fluent/plugin/filter_stdout.rb +53 -0
  177. data/lib/fluent/plugin/formatter.rb +75 -0
  178. data/lib/fluent/plugin/formatter_csv.rb +78 -0
  179. data/lib/fluent/plugin/formatter_hash.rb +35 -0
  180. data/lib/fluent/plugin/formatter_json.rb +59 -0
  181. data/lib/fluent/plugin/formatter_ltsv.rb +44 -0
  182. data/lib/fluent/plugin/formatter_msgpack.rb +33 -0
  183. data/lib/fluent/plugin/formatter_out_file.rb +53 -0
  184. data/lib/fluent/plugin/formatter_single_value.rb +36 -0
  185. data/lib/fluent/plugin/formatter_stdout.rb +76 -0
  186. data/lib/fluent/plugin/formatter_tsv.rb +40 -0
  187. data/lib/fluent/plugin/in_debug_agent.rb +71 -0
  188. data/lib/fluent/plugin/in_dummy.rb +18 -0
  189. data/lib/fluent/plugin/in_exec.rb +110 -0
  190. data/lib/fluent/plugin/in_forward.rb +473 -0
  191. data/lib/fluent/plugin/in_gc_stat.rb +72 -0
  192. data/lib/fluent/plugin/in_http.rb +677 -0
  193. data/lib/fluent/plugin/in_monitor_agent.rb +412 -0
  194. data/lib/fluent/plugin/in_object_space.rb +93 -0
  195. data/lib/fluent/plugin/in_sample.rb +141 -0
  196. data/lib/fluent/plugin/in_syslog.rb +276 -0
  197. data/lib/fluent/plugin/in_tail/group_watch.rb +204 -0
  198. data/lib/fluent/plugin/in_tail/position_file.rb +269 -0
  199. data/lib/fluent/plugin/in_tail.rb +1299 -0
  200. data/lib/fluent/plugin/in_tcp.rb +226 -0
  201. data/lib/fluent/plugin/in_udp.rb +92 -0
  202. data/lib/fluent/plugin/in_unix.rb +195 -0
  203. data/lib/fluent/plugin/input.rb +75 -0
  204. data/lib/fluent/plugin/metrics.rb +119 -0
  205. data/lib/fluent/plugin/metrics_local.rb +96 -0
  206. data/lib/fluent/plugin/multi_output.rb +195 -0
  207. data/lib/fluent/plugin/out_copy.rb +120 -0
  208. data/lib/fluent/plugin/out_exec.rb +105 -0
  209. data/lib/fluent/plugin/out_exec_filter.rb +319 -0
  210. data/lib/fluent/plugin/out_file.rb +340 -0
  211. data/lib/fluent/plugin/out_forward/ack_handler.rb +176 -0
  212. data/lib/fluent/plugin/out_forward/connection_manager.rb +113 -0
  213. data/lib/fluent/plugin/out_forward/error.rb +28 -0
  214. data/lib/fluent/plugin/out_forward/failure_detector.rb +84 -0
  215. data/lib/fluent/plugin/out_forward/handshake_protocol.rb +125 -0
  216. data/lib/fluent/plugin/out_forward/load_balancer.rb +114 -0
  217. data/lib/fluent/plugin/out_forward/socket_cache.rb +142 -0
  218. data/lib/fluent/plugin/out_forward.rb +826 -0
  219. data/lib/fluent/plugin/out_http.rb +275 -0
  220. data/lib/fluent/plugin/out_null.rb +74 -0
  221. data/lib/fluent/plugin/out_relabel.rb +32 -0
  222. data/lib/fluent/plugin/out_roundrobin.rb +84 -0
  223. data/lib/fluent/plugin/out_secondary_file.rb +148 -0
  224. data/lib/fluent/plugin/out_stdout.rb +74 -0
  225. data/lib/fluent/plugin/out_stream.rb +130 -0
  226. data/lib/fluent/plugin/output.rb +1603 -0
  227. data/lib/fluent/plugin/owned_by_mixin.rb +41 -0
  228. data/lib/fluent/plugin/parser.rb +274 -0
  229. data/lib/fluent/plugin/parser_apache.rb +28 -0
  230. data/lib/fluent/plugin/parser_apache2.rb +88 -0
  231. data/lib/fluent/plugin/parser_apache_error.rb +26 -0
  232. data/lib/fluent/plugin/parser_csv.rb +114 -0
  233. data/lib/fluent/plugin/parser_json.rb +96 -0
  234. data/lib/fluent/plugin/parser_ltsv.rb +51 -0
  235. data/lib/fluent/plugin/parser_msgpack.rb +50 -0
  236. data/lib/fluent/plugin/parser_multiline.rb +152 -0
  237. data/lib/fluent/plugin/parser_nginx.rb +28 -0
  238. data/lib/fluent/plugin/parser_none.rb +36 -0
  239. data/lib/fluent/plugin/parser_regexp.rb +68 -0
  240. data/lib/fluent/plugin/parser_syslog.rb +496 -0
  241. data/lib/fluent/plugin/parser_tsv.rb +42 -0
  242. data/lib/fluent/plugin/sd_file.rb +156 -0
  243. data/lib/fluent/plugin/sd_srv.rb +135 -0
  244. data/lib/fluent/plugin/sd_static.rb +58 -0
  245. data/lib/fluent/plugin/service_discovery.rb +65 -0
  246. data/lib/fluent/plugin/socket_util.rb +22 -0
  247. data/lib/fluent/plugin/storage.rb +84 -0
  248. data/lib/fluent/plugin/storage_local.rb +162 -0
  249. data/lib/fluent/plugin/string_util.rb +22 -0
  250. data/lib/fluent/plugin.rb +206 -0
  251. data/lib/fluent/plugin_helper/cert_option.rb +191 -0
  252. data/lib/fluent/plugin_helper/child_process.rb +369 -0
  253. data/lib/fluent/plugin_helper/compat_parameters.rb +343 -0
  254. data/lib/fluent/plugin_helper/counter.rb +51 -0
  255. data/lib/fluent/plugin_helper/event_emitter.rb +100 -0
  256. data/lib/fluent/plugin_helper/event_loop.rb +170 -0
  257. data/lib/fluent/plugin_helper/extract.rb +104 -0
  258. data/lib/fluent/plugin_helper/formatter.rb +147 -0
  259. data/lib/fluent/plugin_helper/http_server/app.rb +79 -0
  260. data/lib/fluent/plugin_helper/http_server/compat/server.rb +92 -0
  261. data/lib/fluent/plugin_helper/http_server/compat/ssl_context_extractor.rb +52 -0
  262. data/lib/fluent/plugin_helper/http_server/compat/webrick_handler.rb +58 -0
  263. data/lib/fluent/plugin_helper/http_server/methods.rb +35 -0
  264. data/lib/fluent/plugin_helper/http_server/request.rb +42 -0
  265. data/lib/fluent/plugin_helper/http_server/router.rb +54 -0
  266. data/lib/fluent/plugin_helper/http_server/server.rb +94 -0
  267. data/lib/fluent/plugin_helper/http_server/ssl_context_builder.rb +41 -0
  268. data/lib/fluent/plugin_helper/http_server.rb +135 -0
  269. data/lib/fluent/plugin_helper/inject.rb +154 -0
  270. data/lib/fluent/plugin_helper/metrics.rb +129 -0
  271. data/lib/fluent/plugin_helper/parser.rb +147 -0
  272. data/lib/fluent/plugin_helper/record_accessor.rb +207 -0
  273. data/lib/fluent/plugin_helper/retry_state.rb +219 -0
  274. data/lib/fluent/plugin_helper/server.rb +828 -0
  275. data/lib/fluent/plugin_helper/service_discovery/manager.rb +146 -0
  276. data/lib/fluent/plugin_helper/service_discovery/round_robin_balancer.rb +43 -0
  277. data/lib/fluent/plugin_helper/service_discovery.rb +125 -0
  278. data/lib/fluent/plugin_helper/socket.rb +288 -0
  279. data/lib/fluent/plugin_helper/socket_option.rb +98 -0
  280. data/lib/fluent/plugin_helper/storage.rb +349 -0
  281. data/lib/fluent/plugin_helper/thread.rb +180 -0
  282. data/lib/fluent/plugin_helper/timer.rb +92 -0
  283. data/lib/fluent/plugin_helper.rb +75 -0
  284. data/lib/fluent/plugin_id.rb +93 -0
  285. data/lib/fluent/process.rb +22 -0
  286. data/lib/fluent/registry.rb +117 -0
  287. data/lib/fluent/root_agent.rb +372 -0
  288. data/lib/fluent/rpc.rb +95 -0
  289. data/lib/fluent/static_config_analysis.rb +194 -0
  290. data/lib/fluent/supervisor.rb +1076 -0
  291. data/lib/fluent/system_config.rb +189 -0
  292. data/lib/fluent/test/base.rb +78 -0
  293. data/lib/fluent/test/driver/base.rb +231 -0
  294. data/lib/fluent/test/driver/base_owned.rb +83 -0
  295. data/lib/fluent/test/driver/base_owner.rb +135 -0
  296. data/lib/fluent/test/driver/event_feeder.rb +98 -0
  297. data/lib/fluent/test/driver/filter.rb +61 -0
  298. data/lib/fluent/test/driver/formatter.rb +30 -0
  299. data/lib/fluent/test/driver/input.rb +31 -0
  300. data/lib/fluent/test/driver/multi_output.rb +53 -0
  301. data/lib/fluent/test/driver/output.rb +102 -0
  302. data/lib/fluent/test/driver/parser.rb +30 -0
  303. data/lib/fluent/test/driver/storage.rb +30 -0
  304. data/lib/fluent/test/driver/test_event_router.rb +45 -0
  305. data/lib/fluent/test/filter_test.rb +77 -0
  306. data/lib/fluent/test/formatter_test.rb +65 -0
  307. data/lib/fluent/test/helpers.rb +134 -0
  308. data/lib/fluent/test/input_test.rb +174 -0
  309. data/lib/fluent/test/log.rb +79 -0
  310. data/lib/fluent/test/output_test.rb +156 -0
  311. data/lib/fluent/test/parser_test.rb +70 -0
  312. data/lib/fluent/test/startup_shutdown.rb +44 -0
  313. data/lib/fluent/test.rb +58 -0
  314. data/lib/fluent/time.rb +512 -0
  315. data/lib/fluent/timezone.rb +171 -0
  316. data/lib/fluent/tls.rb +81 -0
  317. data/lib/fluent/unique_id.rb +39 -0
  318. data/lib/fluent/variable_store.rb +40 -0
  319. data/lib/fluent/version.rb +21 -0
  320. data/lib/fluent/win32api.rb +38 -0
  321. data/lib/fluent/winsvc.rb +100 -0
  322. data/templates/new_gem/Gemfile +3 -0
  323. data/templates/new_gem/README.md.erb +43 -0
  324. data/templates/new_gem/Rakefile +13 -0
  325. data/templates/new_gem/fluent-plugin.gemspec.erb +27 -0
  326. data/templates/new_gem/lib/fluent/plugin/filter.rb.erb +14 -0
  327. data/templates/new_gem/lib/fluent/plugin/formatter.rb.erb +14 -0
  328. data/templates/new_gem/lib/fluent/plugin/input.rb.erb +11 -0
  329. data/templates/new_gem/lib/fluent/plugin/output.rb.erb +11 -0
  330. data/templates/new_gem/lib/fluent/plugin/parser.rb.erb +15 -0
  331. data/templates/new_gem/lib/fluent/plugin/storage.rb.erb +40 -0
  332. data/templates/new_gem/test/helper.rb.erb +7 -0
  333. data/templates/new_gem/test/plugin/test_filter.rb.erb +18 -0
  334. data/templates/new_gem/test/plugin/test_formatter.rb.erb +18 -0
  335. data/templates/new_gem/test/plugin/test_input.rb.erb +18 -0
  336. data/templates/new_gem/test/plugin/test_output.rb.erb +18 -0
  337. data/templates/new_gem/test/plugin/test_parser.rb.erb +18 -0
  338. data/templates/new_gem/test/plugin/test_storage.rb.erb +18 -0
  339. data/templates/plugin_config_formatter/param.md-compact.erb +25 -0
  340. data/templates/plugin_config_formatter/param.md-table.erb +10 -0
  341. data/templates/plugin_config_formatter/param.md.erb +34 -0
  342. data/templates/plugin_config_formatter/section.md.erb +12 -0
  343. data/test/command/test_binlog_reader.rb +362 -0
  344. data/test/command/test_ca_generate.rb +70 -0
  345. data/test/command/test_cap_ctl.rb +100 -0
  346. data/test/command/test_cat.rb +128 -0
  347. data/test/command/test_ctl.rb +56 -0
  348. data/test/command/test_fluentd.rb +1291 -0
  349. data/test/command/test_plugin_config_formatter.rb +397 -0
  350. data/test/command/test_plugin_generator.rb +109 -0
  351. data/test/compat/test_calls_super.rb +166 -0
  352. data/test/compat/test_parser.rb +92 -0
  353. data/test/config/assertions.rb +42 -0
  354. data/test/config/test_config_parser.rb +551 -0
  355. data/test/config/test_configurable.rb +1784 -0
  356. data/test/config/test_configure_proxy.rb +604 -0
  357. data/test/config/test_dsl.rb +415 -0
  358. data/test/config/test_element.rb +518 -0
  359. data/test/config/test_literal_parser.rb +309 -0
  360. data/test/config/test_plugin_configuration.rb +56 -0
  361. data/test/config/test_section.rb +191 -0
  362. data/test/config/test_system_config.rb +195 -0
  363. data/test/config/test_types.rb +408 -0
  364. data/test/counter/test_client.rb +563 -0
  365. data/test/counter/test_error.rb +44 -0
  366. data/test/counter/test_mutex_hash.rb +179 -0
  367. data/test/counter/test_server.rb +589 -0
  368. data/test/counter/test_store.rb +258 -0
  369. data/test/counter/test_validator.rb +137 -0
  370. data/test/helper.rb +155 -0
  371. data/test/helpers/fuzzy_assert.rb +89 -0
  372. data/test/helpers/process_extenstion.rb +33 -0
  373. data/test/log/test_console_adapter.rb +110 -0
  374. data/test/plugin/data/2010/01/20100102-030405.log +0 -0
  375. data/test/plugin/data/2010/01/20100102-030406.log +0 -0
  376. data/test/plugin/data/2010/01/20100102.log +0 -0
  377. data/test/plugin/data/log/bar +0 -0
  378. data/test/plugin/data/log/foo/bar.log +0 -0
  379. data/test/plugin/data/log/foo/bar2 +0 -0
  380. data/test/plugin/data/log/test.log +0 -0
  381. data/test/plugin/data/sd_file/config +11 -0
  382. data/test/plugin/data/sd_file/config.json +17 -0
  383. data/test/plugin/data/sd_file/config.yaml +11 -0
  384. data/test/plugin/data/sd_file/config.yml +11 -0
  385. data/test/plugin/data/sd_file/invalid_config.yml +7 -0
  386. data/test/plugin/in_tail/test_fifo.rb +121 -0
  387. data/test/plugin/in_tail/test_io_handler.rb +150 -0
  388. data/test/plugin/in_tail/test_position_file.rb +346 -0
  389. data/test/plugin/out_forward/test_ack_handler.rb +140 -0
  390. data/test/plugin/out_forward/test_connection_manager.rb +145 -0
  391. data/test/plugin/out_forward/test_handshake_protocol.rb +112 -0
  392. data/test/plugin/out_forward/test_load_balancer.rb +106 -0
  393. data/test/plugin/out_forward/test_socket_cache.rb +174 -0
  394. data/test/plugin/test_bare_output.rb +131 -0
  395. data/test/plugin/test_base.rb +247 -0
  396. data/test/plugin/test_buf_file.rb +1314 -0
  397. data/test/plugin/test_buf_file_single.rb +898 -0
  398. data/test/plugin/test_buf_memory.rb +42 -0
  399. data/test/plugin/test_buffer.rb +1434 -0
  400. data/test/plugin/test_buffer_chunk.rb +209 -0
  401. data/test/plugin/test_buffer_file_chunk.rb +871 -0
  402. data/test/plugin/test_buffer_file_single_chunk.rb +611 -0
  403. data/test/plugin/test_buffer_memory_chunk.rb +339 -0
  404. data/test/plugin/test_compressable.rb +87 -0
  405. data/test/plugin/test_file_util.rb +96 -0
  406. data/test/plugin/test_filter.rb +368 -0
  407. data/test/plugin/test_filter_grep.rb +697 -0
  408. data/test/plugin/test_filter_parser.rb +731 -0
  409. data/test/plugin/test_filter_record_transformer.rb +577 -0
  410. data/test/plugin/test_filter_stdout.rb +207 -0
  411. data/test/plugin/test_formatter_csv.rb +136 -0
  412. data/test/plugin/test_formatter_hash.rb +38 -0
  413. data/test/plugin/test_formatter_json.rb +61 -0
  414. data/test/plugin/test_formatter_ltsv.rb +70 -0
  415. data/test/plugin/test_formatter_msgpack.rb +28 -0
  416. data/test/plugin/test_formatter_out_file.rb +116 -0
  417. data/test/plugin/test_formatter_single_value.rb +44 -0
  418. data/test/plugin/test_formatter_tsv.rb +76 -0
  419. data/test/plugin/test_in_debug_agent.rb +49 -0
  420. data/test/plugin/test_in_exec.rb +261 -0
  421. data/test/plugin/test_in_forward.rb +1178 -0
  422. data/test/plugin/test_in_gc_stat.rb +62 -0
  423. data/test/plugin/test_in_http.rb +1102 -0
  424. data/test/plugin/test_in_monitor_agent.rb +922 -0
  425. data/test/plugin/test_in_object_space.rb +66 -0
  426. data/test/plugin/test_in_sample.rb +190 -0
  427. data/test/plugin/test_in_syslog.rb +505 -0
  428. data/test/plugin/test_in_tail.rb +3125 -0
  429. data/test/plugin/test_in_tcp.rb +328 -0
  430. data/test/plugin/test_in_udp.rb +296 -0
  431. data/test/plugin/test_in_unix.rb +181 -0
  432. data/test/plugin/test_input.rb +137 -0
  433. data/test/plugin/test_metadata.rb +89 -0
  434. data/test/plugin/test_metrics.rb +294 -0
  435. data/test/plugin/test_metrics_local.rb +96 -0
  436. data/test/plugin/test_multi_output.rb +204 -0
  437. data/test/plugin/test_out_copy.rb +308 -0
  438. data/test/plugin/test_out_exec.rb +312 -0
  439. data/test/plugin/test_out_exec_filter.rb +606 -0
  440. data/test/plugin/test_out_file.rb +1038 -0
  441. data/test/plugin/test_out_forward.rb +1361 -0
  442. data/test/plugin/test_out_http.rb +429 -0
  443. data/test/plugin/test_out_null.rb +105 -0
  444. data/test/plugin/test_out_relabel.rb +28 -0
  445. data/test/plugin/test_out_roundrobin.rb +146 -0
  446. data/test/plugin/test_out_secondary_file.rb +458 -0
  447. data/test/plugin/test_out_stdout.rb +205 -0
  448. data/test/plugin/test_out_stream.rb +103 -0
  449. data/test/plugin/test_output.rb +1334 -0
  450. data/test/plugin/test_output_as_buffered.rb +2024 -0
  451. data/test/plugin/test_output_as_buffered_backup.rb +363 -0
  452. data/test/plugin/test_output_as_buffered_compress.rb +179 -0
  453. data/test/plugin/test_output_as_buffered_overflow.rb +250 -0
  454. data/test/plugin/test_output_as_buffered_retries.rb +966 -0
  455. data/test/plugin/test_output_as_buffered_secondary.rb +882 -0
  456. data/test/plugin/test_output_as_standard.rb +374 -0
  457. data/test/plugin/test_owned_by.rb +34 -0
  458. data/test/plugin/test_parser.rb +399 -0
  459. data/test/plugin/test_parser_apache.rb +42 -0
  460. data/test/plugin/test_parser_apache2.rb +47 -0
  461. data/test/plugin/test_parser_apache_error.rb +45 -0
  462. data/test/plugin/test_parser_csv.rb +200 -0
  463. data/test/plugin/test_parser_json.rb +138 -0
  464. data/test/plugin/test_parser_labeled_tsv.rb +160 -0
  465. data/test/plugin/test_parser_multiline.rb +111 -0
  466. data/test/plugin/test_parser_nginx.rb +88 -0
  467. data/test/plugin/test_parser_none.rb +52 -0
  468. data/test/plugin/test_parser_regexp.rb +284 -0
  469. data/test/plugin/test_parser_syslog.rb +650 -0
  470. data/test/plugin/test_parser_tsv.rb +122 -0
  471. data/test/plugin/test_sd_file.rb +228 -0
  472. data/test/plugin/test_sd_srv.rb +230 -0
  473. data/test/plugin/test_storage.rb +166 -0
  474. data/test/plugin/test_storage_local.rb +335 -0
  475. data/test/plugin/test_string_util.rb +26 -0
  476. data/test/plugin_helper/data/cert/cert-key.pem +27 -0
  477. data/test/plugin_helper/data/cert/cert-with-CRLF.pem +19 -0
  478. data/test/plugin_helper/data/cert/cert-with-no-newline.pem +19 -0
  479. data/test/plugin_helper/data/cert/cert.pem +19 -0
  480. data/test/plugin_helper/data/cert/cert_chains/ca-cert-key.pem +27 -0
  481. data/test/plugin_helper/data/cert/cert_chains/ca-cert.pem +20 -0
  482. data/test/plugin_helper/data/cert/cert_chains/cert-key.pem +27 -0
  483. data/test/plugin_helper/data/cert/cert_chains/cert.pem +40 -0
  484. data/test/plugin_helper/data/cert/empty.pem +0 -0
  485. data/test/plugin_helper/data/cert/generate_cert.rb +125 -0
  486. data/test/plugin_helper/data/cert/with_ca/ca-cert-key-pass.pem +30 -0
  487. data/test/plugin_helper/data/cert/with_ca/ca-cert-key.pem +27 -0
  488. data/test/plugin_helper/data/cert/with_ca/ca-cert-pass.pem +20 -0
  489. data/test/plugin_helper/data/cert/with_ca/ca-cert.pem +20 -0
  490. data/test/plugin_helper/data/cert/with_ca/cert-key-pass.pem +30 -0
  491. data/test/plugin_helper/data/cert/with_ca/cert-key.pem +27 -0
  492. data/test/plugin_helper/data/cert/with_ca/cert-pass.pem +21 -0
  493. data/test/plugin_helper/data/cert/with_ca/cert.pem +21 -0
  494. data/test/plugin_helper/data/cert/without_ca/cert-key-pass.pem +30 -0
  495. data/test/plugin_helper/data/cert/without_ca/cert-key.pem +27 -0
  496. data/test/plugin_helper/data/cert/without_ca/cert-pass.pem +20 -0
  497. data/test/plugin_helper/data/cert/without_ca/cert.pem +20 -0
  498. data/test/plugin_helper/http_server/test_app.rb +65 -0
  499. data/test/plugin_helper/http_server/test_route.rb +32 -0
  500. data/test/plugin_helper/service_discovery/test_manager.rb +93 -0
  501. data/test/plugin_helper/service_discovery/test_round_robin_balancer.rb +21 -0
  502. data/test/plugin_helper/test_cert_option.rb +25 -0
  503. data/test/plugin_helper/test_child_process.rb +862 -0
  504. data/test/plugin_helper/test_compat_parameters.rb +358 -0
  505. data/test/plugin_helper/test_event_emitter.rb +80 -0
  506. data/test/plugin_helper/test_event_loop.rb +52 -0
  507. data/test/plugin_helper/test_extract.rb +194 -0
  508. data/test/plugin_helper/test_formatter.rb +255 -0
  509. data/test/plugin_helper/test_http_server_helper.rb +372 -0
  510. data/test/plugin_helper/test_inject.rb +561 -0
  511. data/test/plugin_helper/test_metrics.rb +137 -0
  512. data/test/plugin_helper/test_parser.rb +264 -0
  513. data/test/plugin_helper/test_record_accessor.rb +238 -0
  514. data/test/plugin_helper/test_retry_state.rb +1006 -0
  515. data/test/plugin_helper/test_server.rb +1895 -0
  516. data/test/plugin_helper/test_service_discovery.rb +165 -0
  517. data/test/plugin_helper/test_socket.rb +146 -0
  518. data/test/plugin_helper/test_storage.rb +542 -0
  519. data/test/plugin_helper/test_thread.rb +164 -0
  520. data/test/plugin_helper/test_timer.rb +130 -0
  521. data/test/scripts/exec_script.rb +32 -0
  522. data/test/scripts/fluent/plugin/formatter1/formatter_test1.rb +7 -0
  523. data/test/scripts/fluent/plugin/formatter2/formatter_test2.rb +7 -0
  524. data/test/scripts/fluent/plugin/formatter_known.rb +8 -0
  525. data/test/scripts/fluent/plugin/out_test.rb +81 -0
  526. data/test/scripts/fluent/plugin/out_test2.rb +80 -0
  527. data/test/scripts/fluent/plugin/parser_known.rb +4 -0
  528. data/test/test_capability.rb +74 -0
  529. data/test/test_clock.rb +164 -0
  530. data/test/test_config.rb +369 -0
  531. data/test/test_configdsl.rb +148 -0
  532. data/test/test_daemonizer.rb +91 -0
  533. data/test/test_engine.rb +203 -0
  534. data/test/test_event.rb +531 -0
  535. data/test/test_event_router.rb +348 -0
  536. data/test/test_event_time.rb +199 -0
  537. data/test/test_file_wrapper.rb +53 -0
  538. data/test/test_filter.rb +121 -0
  539. data/test/test_fluent_log_event_router.rb +99 -0
  540. data/test/test_formatter.rb +369 -0
  541. data/test/test_input.rb +31 -0
  542. data/test/test_log.rb +1076 -0
  543. data/test/test_match.rb +148 -0
  544. data/test/test_mixin.rb +351 -0
  545. data/test/test_msgpack_factory.rb +50 -0
  546. data/test/test_oj_options.rb +55 -0
  547. data/test/test_output.rb +278 -0
  548. data/test/test_plugin.rb +251 -0
  549. data/test/test_plugin_classes.rb +370 -0
  550. data/test/test_plugin_helper.rb +81 -0
  551. data/test/test_plugin_id.rb +119 -0
  552. data/test/test_process.rb +14 -0
  553. data/test/test_root_agent.rb +951 -0
  554. data/test/test_static_config_analysis.rb +177 -0
  555. data/test/test_supervisor.rb +821 -0
  556. data/test/test_test_drivers.rb +136 -0
  557. data/test/test_time_formatter.rb +301 -0
  558. data/test/test_time_parser.rb +362 -0
  559. data/test/test_tls.rb +65 -0
  560. data/test/test_unique_id.rb +47 -0
  561. data/test/test_variable_store.rb +65 -0
  562. metadata +1183 -0
@@ -0,0 +1,1102 @@
1
+ require_relative '../helper'
2
+ require 'fluent/test/driver/input'
3
+ require 'fluent/plugin/in_http'
4
+ require 'net/http'
5
+ require 'timecop'
6
+
7
+ class HttpInputTest < Test::Unit::TestCase
8
+ class << self
9
+ def startup
10
+ @server = ServerEngine::SocketManager::Server.open
11
+ ENV['SERVERENGINE_SOCKETMANAGER_PATH'] = @server.path.to_s
12
+ end
13
+
14
+ def shutdown
15
+ @server.close
16
+ end
17
+ end
18
+
19
+ def setup
20
+ Fluent::Test.setup
21
+ @port = unused_port
22
+ end
23
+
24
+ def teardown
25
+ Timecop.return
26
+ @port = nil
27
+ end
28
+
29
+ def config
30
+ %[
31
+ port #{@port}
32
+ bind "127.0.0.1"
33
+ body_size_limit 10m
34
+ keepalive_timeout 5
35
+ respond_with_empty_img true
36
+ use_204_response false
37
+ ]
38
+ end
39
+
40
+ def create_driver(conf=config)
41
+ Fluent::Test::Driver::Input.new(Fluent::Plugin::HttpInput).configure(conf)
42
+ end
43
+
44
+ def test_configure
45
+ d = create_driver
46
+ assert_equal @port, d.instance.port
47
+ assert_equal '127.0.0.1', d.instance.bind
48
+ assert_equal 10*1024*1024, d.instance.body_size_limit
49
+ assert_equal 5, d.instance.keepalive_timeout
50
+ assert_equal false, d.instance.add_http_headers
51
+ assert_equal false, d.instance.add_query_params
52
+ end
53
+
54
+ def test_time
55
+ d = create_driver
56
+ time = event_time("2011-01-02 13:14:15.123 UTC")
57
+ Timecop.freeze(Time.at(time))
58
+
59
+ events = [
60
+ ["tag1", time, {"a" => 1}],
61
+ ["tag2", time, {"a" => 2}],
62
+ ]
63
+ res_codes = []
64
+
65
+ d.run(expect_records: 2) do
66
+ events.each do |tag, _time, record|
67
+ res = post("/#{tag}", {"json"=>record.to_json})
68
+ res_codes << res.code
69
+ end
70
+ end
71
+
72
+ assert_equal ["200", "200"], res_codes
73
+ assert_equal events, d.events
74
+ assert_equal_event_time time, d.events[0][1]
75
+ assert_equal_event_time time, d.events[1][1]
76
+ end
77
+
78
+ def test_time_as_float
79
+ d = create_driver
80
+ time = event_time("2011-01-02 13:14:15.123 UTC")
81
+ float_time = time.to_f
82
+
83
+ events = [
84
+ ["tag1", time, {"a"=>1}],
85
+ ]
86
+ res_codes = []
87
+
88
+ d.run(expect_records: 1) do
89
+ events.each do |tag, t, record|
90
+ res = post("/#{tag}", {"json"=>record.to_json, "time"=>float_time.to_s})
91
+ res_codes << res.code
92
+ end
93
+ end
94
+ assert_equal ["200"], res_codes
95
+ assert_equal events, d.events
96
+ assert_equal_event_time time, d.events[0][1]
97
+ end
98
+
99
+ def test_json
100
+ d = create_driver
101
+ time = event_time("2011-01-02 13:14:15 UTC")
102
+ time_i = time.to_i
103
+
104
+ events = [
105
+ ["tag1", time_i, {"a"=>1}],
106
+ ["tag2", time_i, {"a"=>2}],
107
+ ]
108
+ res_codes = []
109
+
110
+ d.run(expect_records: 2) do
111
+ events.each do |tag, t, record|
112
+ res = post("/#{tag}", {"json"=>record.to_json, "time"=>t.to_s})
113
+ res_codes << res.code
114
+ end
115
+ end
116
+ assert_equal ["200", "200"], res_codes
117
+ assert_equal events, d.events
118
+ assert_equal_event_time time, d.events[0][1]
119
+ assert_equal_event_time time, d.events[1][1]
120
+ end
121
+
122
+ data('json' => ['json', :to_json],
123
+ 'msgpack' => ['msgpack', :to_msgpack])
124
+ def test_default_with_time_format(data)
125
+ param, method_name = data
126
+ d = create_driver(config + %[
127
+ <parse>
128
+ keep_time_key
129
+ time_format %iso8601
130
+ </parse>
131
+ ])
132
+
133
+ time = event_time("2020-06-10T01:14:27+00:00")
134
+ events = [
135
+ ["tag1", time, {"a" => 1, "time" => '2020-06-10T01:14:27+00:00'}],
136
+ ["tag2", time, {"a" => 2, "time" => '2020-06-10T01:14:27+00:00'}],
137
+ ]
138
+ res_codes = []
139
+
140
+ d.run(expect_records: 2) do
141
+ events.each do |tag, t, record|
142
+ res = post("/#{tag}", {param => record.__send__(method_name)})
143
+ res_codes << res.code
144
+ end
145
+ end
146
+ assert_equal ["200", "200"], res_codes
147
+ assert_equal events, d.events
148
+ assert_equal_event_time time, d.events[0][1]
149
+ assert_equal_event_time time, d.events[1][1]
150
+ end
151
+
152
+ def test_multi_json
153
+ d = create_driver
154
+ time = event_time("2011-01-02 13:14:15 UTC")
155
+ time_i = time.to_i
156
+
157
+ records = [{"a"=>1},{"a"=>2}]
158
+ events = [
159
+ ["tag1", time_i, records[0]],
160
+ ["tag1", time_i, records[1]],
161
+ ]
162
+ tag = "tag1"
163
+ res_codes = []
164
+ d.run(expect_records: 2, timeout: 5) do
165
+ res = post("/#{tag}", {"json"=>records.to_json, "time"=>time_i.to_s})
166
+ res_codes << res.code
167
+ end
168
+ assert_equal ["200"], res_codes
169
+ assert_equal events, d.events
170
+ assert_equal_event_time time, d.events[0][1]
171
+ assert_equal_event_time time, d.events[1][1]
172
+ end
173
+
174
+ def test_multi_json_with_time_field
175
+ d = create_driver
176
+ time = event_time("2011-01-02 13:14:15 UTC")
177
+ time_i = time.to_i
178
+ time_f = time.to_f
179
+
180
+ records = [{"a" => 1, 'time' => time_i},{"a" => 2, 'time' => time_f}]
181
+ events = [
182
+ ["tag1", time, {'a' => 1}],
183
+ ["tag1", time, {'a' => 2}],
184
+ ]
185
+ tag = "tag1"
186
+ res_codes = []
187
+ d.run(expect_records: 2, timeout: 5) do
188
+ res = post("/#{tag}", {"json" => records.to_json})
189
+ res_codes << res.code
190
+ end
191
+ assert_equal ["200"], res_codes
192
+ assert_equal events, d.events
193
+ assert_instance_of Fluent::EventTime, d.events[0][1]
194
+ assert_instance_of Fluent::EventTime, d.events[1][1]
195
+ assert_equal_event_time time, d.events[0][1]
196
+ assert_equal_event_time time, d.events[1][1]
197
+ end
198
+
199
+ data('json' => ['json', :to_json],
200
+ 'msgpack' => ['msgpack', :to_msgpack])
201
+ def test_default_multi_with_time_format(data)
202
+ param, method_name = data
203
+ d = create_driver(config + %[
204
+ <parse>
205
+ keep_time_key
206
+ time_format %iso8601
207
+ </parse>
208
+ ])
209
+ time = event_time("2020-06-10T01:14:27+00:00")
210
+ events = [
211
+ ["tag1", time, {'a' => 1, 'time' => "2020-06-10T01:14:27+00:00"}],
212
+ ["tag1", time, {'a' => 2, 'time' => "2020-06-10T01:14:27+00:00"}],
213
+ ]
214
+ tag = "tag1"
215
+ res_codes = []
216
+ d.run(expect_records: 2, timeout: 5) do
217
+ res = post("/#{tag}", {param => events.map { |e| e[2] }.__send__(method_name)})
218
+ res_codes << res.code
219
+ end
220
+ assert_equal ["200"], res_codes
221
+ assert_equal events, d.events
222
+ assert_equal_event_time time, d.events[0][1]
223
+ assert_equal_event_time time, d.events[1][1]
224
+ end
225
+
226
+ def test_multi_json_with_nonexistent_time_key
227
+ d = create_driver(config + %[
228
+ <parse>
229
+ time_key missing
230
+ </parse>
231
+ ])
232
+ time = event_time("2011-01-02 13:14:15 UTC")
233
+ time_i = time.to_i
234
+ time_f = time.to_f
235
+
236
+ records = [{"a" => 1, 'time' => time_i},{"a" => 2, 'time' => time_f}]
237
+ tag = "tag1"
238
+ res_codes = []
239
+ d.run(expect_records: 2, timeout: 5) do
240
+ res = post("/#{tag}", {"json" => records.to_json})
241
+ res_codes << res.code
242
+ end
243
+ assert_equal ["200"], res_codes
244
+ assert_equal 2, d.events.size
245
+ assert_not_equal time_i, d.events[0][1].sec # current time is used because "missing" field doesn't exist
246
+ assert_not_equal time_i, d.events[1][1].sec
247
+ end
248
+
249
+ def test_json_with_add_remote_addr
250
+ d = create_driver(config + "add_remote_addr true")
251
+ time = event_time("2011-01-02 13:14:15 UTC")
252
+ time_i = time.to_i
253
+
254
+ events = [
255
+ ["tag1", time, {"REMOTE_ADDR"=>"127.0.0.1", "a"=>1}],
256
+ ["tag2", time, {"REMOTE_ADDR"=>"127.0.0.1", "a"=>2}],
257
+ ]
258
+ res_codes = []
259
+ d.run(expect_records: 2) do
260
+ events.each do |tag, _t, record|
261
+ res = post("/#{tag}", {"json"=>record.to_json, "time"=>time_i.to_s})
262
+ res_codes << res.code
263
+ end
264
+ end
265
+ assert_equal ["200", "200"], res_codes
266
+ assert_equal events, d.events
267
+ assert_equal_event_time time, d.events[0][1]
268
+ assert_equal_event_time time, d.events[1][1]
269
+ end
270
+
271
+ def test_exact_match_for_expect
272
+ d = create_driver(config)
273
+ records = [{ "a" => 1}, { "a" => 2 }]
274
+ tag = "tag1"
275
+ res_codes = []
276
+
277
+ d.run(expect_records: 0, timeout: 5) do
278
+ res = post("/#{tag}", { "json" => records.to_json }, { 'Expect' => 'something' })
279
+ res_codes << res.code
280
+ end
281
+ assert_equal ["417"], res_codes
282
+ end
283
+
284
+ def test_exact_match_for_expect_with_other_header
285
+ d = create_driver(config)
286
+
287
+ records = [{ "a" => 1}, { "a" => 2 }]
288
+ tag = "tag1"
289
+ res_codes = []
290
+
291
+ d.run(expect_records: 2, timeout: 5) do
292
+ res = post("/#{tag}", { "json" => records.to_json, 'x-envoy-expected-rq-timeout-ms' => 4 })
293
+ res_codes << res.code
294
+ end
295
+ assert_equal ["200"], res_codes
296
+
297
+ assert_equal "tag1", d.events[0][0]
298
+ assert_equal 1, d.events[0][2]["a"]
299
+ assert_equal "tag1", d.events[1][0]
300
+ assert_equal 2, d.events[1][2]["a"]
301
+ end
302
+
303
+ def test_multi_json_with_add_remote_addr
304
+ d = create_driver(config + "add_remote_addr true")
305
+ time = event_time("2011-01-02 13:14:15 UTC")
306
+ time_i = time.to_i
307
+
308
+ records = [{"a"=>1},{"a"=>2}]
309
+ tag = "tag1"
310
+ res_codes = []
311
+
312
+ d.run(expect_records: 2, timeout: 5) do
313
+ res = post("/#{tag}", {"json"=>records.to_json, "time"=>time_i.to_s})
314
+ res_codes << res.code
315
+ end
316
+ assert_equal ["200"], res_codes
317
+
318
+ assert_equal "tag1", d.events[0][0]
319
+ assert_equal_event_time time, d.events[0][1]
320
+ assert_equal 1, d.events[0][2]["a"]
321
+ assert{ d.events[0][2].has_key?("REMOTE_ADDR") && d.events[0][2]["REMOTE_ADDR"] =~ /^\d{1,4}(\.\d{1,4}){3}$/ }
322
+
323
+ assert_equal "tag1", d.events[1][0]
324
+ assert_equal_event_time time, d.events[1][1]
325
+ assert_equal 2, d.events[1][2]["a"]
326
+ end
327
+
328
+ def test_json_with_add_remote_addr_given_x_forwarded_for
329
+ d = create_driver(config + "add_remote_addr true")
330
+ time = event_time("2011-01-02 13:14:15 UTC")
331
+ time_i = time.to_i
332
+
333
+ events = [
334
+ ["tag1", time, {"a"=>1}],
335
+ ["tag2", time, {"a"=>2}],
336
+ ]
337
+ res_codes = []
338
+
339
+ d.run(expect_records: 2) do
340
+ events.each do |tag, _t, record|
341
+ res = post("/#{tag}", {"json"=>record.to_json, "time"=>time_i.to_s}, {"X-Forwarded-For"=>"129.78.138.66, 127.0.0.1"})
342
+ res_codes << res.code
343
+ end
344
+ end
345
+ assert_equal ["200", "200"], res_codes
346
+
347
+ assert_equal "tag1", d.events[0][0]
348
+ assert_equal_event_time time, d.events[0][1]
349
+ assert_equal({"REMOTE_ADDR"=>"129.78.138.66", "a"=>1}, d.events[0][2])
350
+
351
+ assert_equal "tag2", d.events[1][0]
352
+ assert_equal_event_time time, d.events[1][1]
353
+ assert_equal({"REMOTE_ADDR"=>"129.78.138.66", "a"=>2}, d.events[1][2])
354
+ end
355
+
356
+ def test_multi_json_with_add_remote_addr_given_x_forwarded_for
357
+ d = create_driver(config + "add_remote_addr true")
358
+
359
+ tag = "tag1"
360
+ time = event_time("2011-01-02 13:14:15 UTC")
361
+ time_i = time.to_i
362
+ records = [{"a"=>1},{"a"=>2}]
363
+ events = [
364
+ [tag, time, {"REMOTE_ADDR"=>"129.78.138.66", "a"=>1}],
365
+ [tag, time, {"REMOTE_ADDR"=>"129.78.138.66", "a"=>2}],
366
+ ]
367
+ res_codes = []
368
+
369
+ d.run(expect_records: 2, timeout: 5) do
370
+ res = post("/#{tag}", {"json"=>records.to_json, "time"=>time_i.to_s}, {"X-Forwarded-For"=>"129.78.138.66, 127.0.0.1"})
371
+ res_codes << res.code
372
+ end
373
+ assert_equal ["200"], res_codes
374
+ assert_equal events, d.events
375
+ assert_equal_event_time time, d.events[0][1]
376
+ assert_equal_event_time time, d.events[1][1]
377
+ end
378
+
379
+ def test_add_remote_addr_given_multi_x_forwarded_for
380
+ d = create_driver(config + "add_remote_addr true")
381
+
382
+ tag = "tag1"
383
+ time = event_time("2011-01-02 13:14:15 UTC")
384
+ time_i = time.to_i
385
+ record = {"a" => 1}
386
+ event = ["tag1", time, {"REMOTE_ADDR" => "129.78.138.66", "a" => 1}]
387
+ res_code = nil
388
+
389
+ d.run(expect_records: 1, timeout: 5) do
390
+ res = post("/#{tag}", {"json" => record.to_json, "time" => time_i.to_s}) { |http, req|
391
+ # net/http can't send multiple headers so overwrite it.
392
+ def req.each_capitalized
393
+ block_given? or return enum_for(__method__) { @header.size }
394
+ @header.each do |k, vs|
395
+ vs.each { |v|
396
+ yield capitalize(k), v
397
+ }
398
+ end
399
+ end
400
+ req.add_field("X-Forwarded-For", "129.78.138.66, 127.0.0.1")
401
+ req.add_field("X-Forwarded-For", "8.8.8.8")
402
+ }
403
+ res_code = res.code
404
+ end
405
+ assert_equal "200", res_code
406
+ assert_equal event, d.events.first
407
+ assert_equal_event_time time, d.events.first[1]
408
+ end
409
+
410
+ def test_multi_json_with_add_http_headers
411
+ d = create_driver(config + "add_http_headers true")
412
+ time = event_time("2011-01-02 13:14:15 UTC")
413
+ time_i = time.to_i
414
+ records = [{"a"=>1},{"a"=>2}]
415
+ tag = "tag1"
416
+ res_codes = []
417
+
418
+ d.run(expect_records: 2, timeout: 5) do
419
+ res = post("/#{tag}", {"json"=>records.to_json, "time"=>time_i.to_s})
420
+ res_codes << res.code
421
+ end
422
+ assert_equal ["200"], res_codes
423
+
424
+ assert_equal "tag1", d.events[0][0]
425
+ assert_equal_event_time time, d.events[0][1]
426
+ assert_equal 1, d.events[0][2]["a"]
427
+
428
+ assert_equal "tag1", d.events[1][0]
429
+ assert_equal_event_time time, d.events[1][1]
430
+ assert_equal 2, d.events[1][2]["a"]
431
+
432
+ assert include_http_header?(d.events[0][2])
433
+ assert include_http_header?(d.events[1][2])
434
+ end
435
+
436
+ def test_json_with_add_http_headers
437
+ d = create_driver(config + "add_http_headers true")
438
+ time = event_time("2011-01-02 13:14:15 UTC")
439
+ time_i = time.to_i
440
+ events = [
441
+ ["tag1", time, {"a"=>1}],
442
+ ["tag2", time, {"a"=>2}],
443
+ ]
444
+ res_codes = []
445
+
446
+ d.run(expect_records: 2) do
447
+ events.each do |tag, t, record|
448
+ res = post("/#{tag}", {"json"=>record.to_json, "time"=>time_i.to_s})
449
+ res_codes << res.code
450
+ end
451
+ end
452
+ assert_equal ["200", "200"], res_codes
453
+
454
+ assert_equal "tag1", d.events[0][0]
455
+ assert_equal_event_time time, d.events[0][1]
456
+ assert_equal 1, d.events[0][2]["a"]
457
+
458
+ assert_equal "tag2", d.events[1][0]
459
+ assert_equal_event_time time, d.events[1][1]
460
+ assert_equal 2, d.events[1][2]["a"]
461
+
462
+ assert include_http_header?(d.events[0][2])
463
+ assert include_http_header?(d.events[1][2])
464
+ end
465
+
466
+ def test_multi_json_with_custom_parser
467
+ d = create_driver(config + %[
468
+ <parse>
469
+ @type json
470
+ keep_time_key true
471
+ time_key foo
472
+ time_format %iso8601
473
+ </parse>
474
+ ])
475
+
476
+ time = event_time("2011-01-02 13:14:15 UTC")
477
+ time_s = Time.at(time).iso8601
478
+
479
+ records = [{"foo"=>time_s,"bar"=>"test1"},{"foo"=>time_s,"bar"=>"test2"}]
480
+ tag = "tag1"
481
+ res_codes = []
482
+
483
+ d.run(expect_records: 2, timeout: 5) do
484
+ res = post("/#{tag}", records.to_json, {"Content-Type"=>"application/octet-stream"})
485
+ res_codes << res.code
486
+ end
487
+ assert_equal ["200"], res_codes
488
+
489
+ assert_equal "tag1", d.events[0][0]
490
+ assert_equal_event_time time, d.events[0][1]
491
+ assert_equal d.events[0][2], records[0]
492
+
493
+ assert_equal "tag1", d.events[1][0]
494
+ assert_equal_event_time time, d.events[1][1]
495
+ assert_equal d.events[1][2], records[1]
496
+ end
497
+
498
+ def test_application_json
499
+ d = create_driver
500
+ time = event_time("2011-01-02 13:14:15 UTC")
501
+ time_i = time.to_i
502
+ events = [
503
+ ["tag1", time, {"a"=>1}],
504
+ ["tag2", time, {"a"=>2}],
505
+ ]
506
+ res_codes = []
507
+
508
+ d.run(expect_records: 2) do
509
+ events.each do |tag, t, record|
510
+ res = post("/#{tag}?time=#{time_i.to_s}", record.to_json, {"Content-Type"=>"application/json; charset=utf-8"})
511
+ res_codes << res.code
512
+ end
513
+ end
514
+ assert_equal ["200", "200"], res_codes
515
+ assert_equal events, d.events
516
+ assert_equal_event_time time, d.events[0][1]
517
+ assert_equal_event_time time, d.events[1][1]
518
+ end
519
+
520
+ def test_application_msgpack
521
+ d = create_driver
522
+ time = event_time("2011-01-02 13:14:15 UTC")
523
+ time_i = time.to_i
524
+ events = [
525
+ ["tag1", time, {"a"=>1}],
526
+ ["tag2", time, {"a"=>2}],
527
+ ]
528
+ res_codes = []
529
+
530
+ d.run(expect_records: 2) do
531
+ events.each do |tag, t, record|
532
+ res = post("/#{tag}?time=#{time_i.to_s}", record.to_msgpack, {"Content-Type"=>"application/msgpack"})
533
+ res_codes << res.code
534
+ end
535
+ end
536
+ assert_equal ["200", "200"], res_codes
537
+ assert_equal events, d.events
538
+ assert_equal_event_time time, d.events[0][1]
539
+ assert_equal_event_time time, d.events[1][1]
540
+ end
541
+
542
+ def test_application_ndjson
543
+ d = create_driver
544
+ events = [
545
+ ["tag1", 1643935663, "{\"a\":1}\n{\"b\":2}"],
546
+ ["tag2", 1643935664, "{\"a\":3}\r\n{\"b\":4}"]
547
+ ]
548
+
549
+ expected = [
550
+ ["tag1", 1643935663, {"a"=>1}],
551
+ ["tag1", 1643935663, {"b"=>2}],
552
+ ["tag2", 1643935664, {"a"=>3}],
553
+ ["tag2", 1643935664, {"b"=>4}]
554
+ ]
555
+
556
+ d.run(expect_records: 1) do
557
+ events.each do |tag, time, record|
558
+ res = post("/#{tag}?time=#{time}", record, {"Content-Type"=>"application/x-ndjson"})
559
+ assert_equal("200", res.code)
560
+ end
561
+ end
562
+ assert_equal(expected, d.events)
563
+ end
564
+
565
+ def test_msgpack
566
+ d = create_driver
567
+ time = event_time("2011-01-02 13:14:15 UTC")
568
+ time_i = time.to_i
569
+
570
+ events = [
571
+ ["tag1", time, {"a"=>1}],
572
+ ["tag2", time, {"a"=>2}],
573
+ ]
574
+ res_codes = []
575
+
576
+ d.run(expect_records: 2) do
577
+ events.each do |tag, t, record|
578
+ res = post("/#{tag}", {"msgpack"=>record.to_msgpack, "time"=>time_i.to_s})
579
+ res_codes << res.code
580
+ end
581
+ end
582
+ assert_equal ["200", "200"], res_codes
583
+ assert_equal events, d.events
584
+ assert_equal_event_time time, d.events[0][1]
585
+ assert_equal_event_time time, d.events[1][1]
586
+ end
587
+
588
+ def test_multi_msgpack
589
+ d = create_driver
590
+
591
+ time = event_time("2011-01-02 13:14:15 UTC")
592
+ time_i = time.to_i
593
+
594
+ records = [{"a"=>1},{"a"=>2}]
595
+ events = [
596
+ ["tag1", time, records[0]],
597
+ ["tag1", time, records[1]],
598
+ ]
599
+ tag = "tag1"
600
+ res_codes = []
601
+ d.run(expect_records: 2) do
602
+ res = post("/#{tag}", {"msgpack"=>records.to_msgpack, "time"=>time_i.to_s})
603
+ res_codes << res.code
604
+ end
605
+ assert_equal ["200"], res_codes
606
+ assert_equal events, d.events
607
+ assert_equal_event_time time, d.events[0][1]
608
+ assert_equal_event_time time, d.events[1][1]
609
+ end
610
+
611
+ def test_with_regexp
612
+ d = create_driver(config + %[
613
+ format /^(?<field_1>\\d+):(?<field_2>\\w+)$/
614
+ types field_1:integer
615
+ ])
616
+
617
+ time = event_time("2011-01-02 13:14:15 UTC")
618
+ time_i = time.to_i
619
+ events = [
620
+ ["tag1", time, {"field_1" => 1, "field_2" => 'str'}],
621
+ ["tag2", time, {"field_1" => 2, "field_2" => 'str'}],
622
+ ]
623
+ res_codes = []
624
+
625
+ d.run(expect_records: 2) do
626
+ events.each do |tag, t, record|
627
+ body = record.map { |k, v|
628
+ v.to_s
629
+ }.join(':')
630
+ res = post("/#{tag}?time=#{time_i.to_s}", body, {'Content-Type' => 'application/octet-stream'})
631
+ res_codes << res.code
632
+ end
633
+ end
634
+ assert_equal ["200", "200"], res_codes
635
+ assert_equal events, d.events
636
+ assert_equal_event_time time, d.events[0][1]
637
+ assert_equal_event_time time, d.events[1][1]
638
+ end
639
+
640
+ def test_with_csv
641
+ require 'csv'
642
+
643
+ d = create_driver(config + %[
644
+ format csv
645
+ keys foo,bar
646
+ ])
647
+ time = event_time("2011-01-02 13:14:15 UTC")
648
+ time_i = time.to_i
649
+ events = [
650
+ ["tag1", time, {"foo" => "1", "bar" => 'st"r'}],
651
+ ["tag2", time, {"foo" => "2", "bar" => 'str'}],
652
+ ]
653
+ res_codes = []
654
+
655
+ d.run(expect_records: 2) do
656
+ events.each do |tag, t, record|
657
+ body = record.map { |k, v| v }.to_csv
658
+ res = post("/#{tag}?time=#{time_i.to_s}", body, {'Content-Type' => 'text/comma-separated-values'})
659
+ res_codes << res.code
660
+ end
661
+ end
662
+ assert_equal ["200", "200"], res_codes
663
+ assert_equal events, d.events
664
+ assert_equal_event_time time, d.events[0][1]
665
+ assert_equal_event_time time, d.events[1][1]
666
+ end
667
+
668
+ def test_response_with_empty_img
669
+ d = create_driver(config)
670
+ assert_equal true, d.instance.respond_with_empty_img
671
+
672
+ time = event_time("2011-01-02 13:14:15 UTC")
673
+ time_i = time.to_i
674
+ events = [
675
+ ["tag1", time, {"a"=>1}],
676
+ ["tag2", time, {"a"=>2}],
677
+ ]
678
+ res_codes = []
679
+ res_bodies = []
680
+
681
+ d.run do
682
+ events.each do |tag, _t, record|
683
+ res = post("/#{tag}", {"json"=>record.to_json, "time"=>time_i.to_s})
684
+ res_codes << res.code
685
+ # Ruby returns ASCII-8 encoded string for GIF.
686
+ res_bodies << res.body.force_encoding("UTF-8")
687
+ end
688
+ end
689
+ assert_equal ["200", "200"], res_codes
690
+ assert_equal [Fluent::Plugin::HttpInput::EMPTY_GIF_IMAGE, Fluent::Plugin::HttpInput::EMPTY_GIF_IMAGE], res_bodies
691
+ assert_equal events, d.events
692
+ assert_equal_event_time time, d.events[0][1]
693
+ assert_equal_event_time time, d.events[1][1]
694
+ end
695
+
696
+ def test_response_without_empty_img
697
+ d = create_driver(config + "respond_with_empty_img false")
698
+ assert_equal false, d.instance.respond_with_empty_img
699
+
700
+ time = event_time("2011-01-02 13:14:15 UTC")
701
+ time_i = time.to_i
702
+ events = [
703
+ ["tag1", time, {"a"=>1}],
704
+ ["tag2", time, {"a"=>2}],
705
+ ]
706
+ res_codes = []
707
+ res_bodies = []
708
+
709
+ d.run do
710
+ events.each do |tag, _t, record|
711
+ res = post("/#{tag}", {"json"=>record.to_json, "time"=>time_i.to_s})
712
+ res_codes << res.code
713
+ end
714
+ end
715
+ assert_equal ["200", "200"], res_codes
716
+ assert_equal [], res_bodies
717
+ assert_equal events, d.events
718
+ assert_equal_event_time time, d.events[0][1]
719
+ assert_equal_event_time time, d.events[1][1]
720
+ end
721
+
722
+ def test_response_use_204_response
723
+ d = create_driver(config + %[
724
+ respond_with_empty_img false
725
+ use_204_response true
726
+ ])
727
+ assert_equal true, d.instance.use_204_response
728
+
729
+ time = event_time("2011-01-02 13:14:15 UTC")
730
+ time_i = time.to_i
731
+ events = [
732
+ ["tag1", time, {"a"=>1}],
733
+ ["tag2", time, {"a"=>2}],
734
+ ]
735
+ res_codes = []
736
+ res_bodies = []
737
+
738
+ d.run do
739
+ events.each do |tag, _t, record|
740
+ res = post("/#{tag}", {"json"=>record.to_json, "time"=>time_i.to_s})
741
+ res_codes << res.code
742
+ end
743
+ end
744
+ assert_equal ["204", "204"], res_codes
745
+ assert_equal [], res_bodies
746
+ assert_equal events, d.events
747
+ assert_equal_event_time time, d.events[0][1]
748
+ assert_equal_event_time time, d.events[1][1]
749
+ end
750
+
751
+ def test_cors_allowed
752
+ d = create_driver(config + "cors_allow_origins [\"http://foo.com\"]")
753
+ assert_equal ["http://foo.com"], d.instance.cors_allow_origins
754
+
755
+ time = event_time("2011-01-02 13:14:15 UTC")
756
+ time_i = time.to_i
757
+ events = [
758
+ ["tag1", time, {"a"=>1}],
759
+ ["tag2", time, {"a"=>2}],
760
+ ]
761
+ res_codes = []
762
+ res_headers = []
763
+
764
+ d.run do
765
+ events.each do |tag, _t, record|
766
+ res = post("/#{tag}", {"json"=>record.to_json, "time"=>time_i.to_s}, {"Origin"=>"http://foo.com"})
767
+ res_codes << res.code
768
+ res_headers << res["Access-Control-Allow-Origin"]
769
+ end
770
+ end
771
+ assert_equal ["200", "200"], res_codes
772
+ assert_equal ["http://foo.com", "http://foo.com"], res_headers
773
+ assert_equal events, d.events
774
+ assert_equal_event_time time, d.events[0][1]
775
+ assert_equal_event_time time, d.events[1][1]
776
+ end
777
+
778
+ def test_cors_allowed_wildcard
779
+ d = create_driver(config + 'cors_allow_origins ["*"]')
780
+
781
+ time = event_time("2011-01-02 13:14:15 UTC")
782
+ events = [
783
+ ["tag1", time, {"a"=>1}],
784
+ ]
785
+
786
+ d.run do
787
+ events.each do |tag, time, record|
788
+ headers = {"Origin" => "http://foo.com"}
789
+
790
+ res = post("/#{tag}", {"json" => record.to_json, "time" => time.to_i}, headers)
791
+
792
+ assert_equal "200", res.code
793
+ assert_equal "*", res["Access-Control-Allow-Origin"]
794
+ end
795
+ end
796
+ end
797
+
798
+ def test_get_request
799
+ d = create_driver(config)
800
+
801
+ d.run do
802
+ res = get("/cors.test", {}, {})
803
+ assert_equal "200", res.code
804
+ end
805
+ end
806
+
807
+ def test_cors_preflight
808
+ d = create_driver(config + 'cors_allow_origins ["*"]')
809
+
810
+ d.run do
811
+ header = {
812
+ "Origin" => "http://foo.com",
813
+ "Access-Control-Request-Method" => "POST",
814
+ "Access-Control-Request-Headers" => "Content-Type",
815
+ }
816
+ res = options("/cors.test", {}, header)
817
+
818
+ assert_equal "200", res.code
819
+ assert_equal "*", res["Access-Control-Allow-Origin"]
820
+ assert_equal "POST", res["Access-Control-Allow-Methods"]
821
+ end
822
+ end
823
+
824
+ def test_cors_allowed_wildcard_for_subdomain
825
+ d = create_driver(config + 'cors_allow_origins ["http://*.foo.com"]')
826
+
827
+ time = event_time("2011-01-02 13:14:15 UTC")
828
+ events = [
829
+ ["tag1", time, {"a"=>1}],
830
+ ]
831
+
832
+ d.run do
833
+ events.each do |tag, time, record|
834
+ headers = {"Origin" => "http://subdomain.foo.com"}
835
+
836
+ res = post("/#{tag}", {"json" => record.to_json, "time" => time.to_i}, headers)
837
+
838
+ assert_equal "200", res.code
839
+ assert_equal "http://subdomain.foo.com", res["Access-Control-Allow-Origin"]
840
+ end
841
+ end
842
+ end
843
+
844
+ def test_cors_allowed_exclude_empty_string
845
+ d = create_driver(config + 'cors_allow_origins ["", "http://*.foo.com"]')
846
+
847
+ time = event_time("2011-01-02 13:14:15 UTC")
848
+ events = [
849
+ ["tag1", time, {"a"=>1}],
850
+ ]
851
+
852
+ d.run do
853
+ events.each do |tag, time, record|
854
+ headers = {"Origin" => "http://subdomain.foo.com"}
855
+
856
+ res = post("/#{tag}", {"json" => record.to_json, "time" => time.to_i}, headers)
857
+
858
+ assert_equal "200", res.code
859
+ assert_equal "http://subdomain.foo.com", res["Access-Control-Allow-Origin"]
860
+ end
861
+ end
862
+ end
863
+
864
+ def test_cors_allowed_wildcard_preflight_for_subdomain
865
+ d = create_driver(config + 'cors_allow_origins ["http://*.foo.com"]')
866
+
867
+ d.run do
868
+ header = {
869
+ "Origin" => "http://subdomain.foo.com",
870
+ "Access-Control-Request-Method" => "POST",
871
+ "Access-Control-Request-Headers" => "Content-Type",
872
+ }
873
+ res = options("/cors.test", {}, header)
874
+
875
+ assert_equal "200", res.code
876
+ assert_equal "http://subdomain.foo.com", res["Access-Control-Allow-Origin"]
877
+ assert_equal "POST", res["Access-Control-Allow-Methods"]
878
+ end
879
+ end
880
+
881
+ def test_cors_allow_credentials
882
+ d = create_driver(config + %[
883
+ cors_allow_origins ["http://foo.com"]
884
+ cors_allow_credentials
885
+ ])
886
+ assert_equal true, d.instance.cors_allow_credentials
887
+
888
+ time = event_time("2011-01-02 13:14:15 UTC")
889
+ event = ["tag1", time, {"a"=>1}]
890
+ res_code = nil
891
+ res_header = nil
892
+
893
+ d.run do
894
+ res = post("/#{event[0]}", {"json"=>event[2].to_json, "time"=>time.to_i.to_s}, {"Origin"=>"http://foo.com"})
895
+ res_code = res.code
896
+ res_header = res["Access-Control-Allow-Credentials"]
897
+ end
898
+ assert_equal(
899
+ {
900
+ response_code: "200",
901
+ allow_credentials_header: "true",
902
+ events: [event]
903
+ },
904
+ {
905
+ response_code: res_code,
906
+ allow_credentials_header: res_header,
907
+ events: d.events
908
+ }
909
+ )
910
+ end
911
+
912
+ def test_cors_allow_credentials_for_wildcard_origins
913
+ assert_raise(Fluent::ConfigError) do
914
+ create_driver(config + %[
915
+ cors_allow_origins ["*"]
916
+ cors_allow_credentials
917
+ ])
918
+ end
919
+ end
920
+
921
+ def test_content_encoding_gzip
922
+ d = create_driver
923
+
924
+ time = event_time("2011-01-02 13:14:15 UTC")
925
+ events = [
926
+ ["tag1", time, {"a"=>1}],
927
+ ["tag2", time, {"a"=>2}],
928
+ ]
929
+ res_codes = []
930
+
931
+ d.run do
932
+ events.each do |tag, time, record|
933
+ header = {'Content-Type'=>'application/json', 'Content-Encoding'=>'gzip'}
934
+ res = post("/#{tag}?time=#{time}", compress_gzip(record.to_json), header)
935
+ res_codes << res.code
936
+ end
937
+ end
938
+ assert_equal ["200", "200"], res_codes
939
+ assert_equal events, d.events
940
+ assert_equal_event_time time, d.events[0][1]
941
+ assert_equal_event_time time, d.events[1][1]
942
+ end
943
+
944
+ def test_content_encoding_deflate
945
+ d = create_driver
946
+
947
+ time = event_time("2011-01-02 13:14:15 UTC")
948
+ events = [
949
+ ["tag1", time, {"a"=>1}],
950
+ ["tag2", time, {"a"=>2}],
951
+ ]
952
+ res_codes = []
953
+
954
+ d.run do
955
+ events.each do |tag, time, record|
956
+ header = {'Content-Type'=>'application/msgpack', 'Content-Encoding'=>'deflate'}
957
+ res = post("/#{tag}?time=#{time}", Zlib.deflate(record.to_msgpack), header)
958
+ res_codes << res.code
959
+ end
960
+ end
961
+ assert_equal ["200", "200"], res_codes
962
+ assert_equal events, d.events
963
+ assert_equal_event_time time, d.events[0][1]
964
+ assert_equal_event_time time, d.events[1][1]
965
+ end
966
+
967
+ def test_cors_disallowed
968
+ d = create_driver(config + "cors_allow_origins [\"http://foo.com\"]")
969
+ assert_equal ["http://foo.com"], d.instance.cors_allow_origins
970
+
971
+ time = event_time("2011-01-02 13:14:15 UTC")
972
+ time_i = time.to_i
973
+ res_codes = []
974
+
975
+ d.end_if{ res_codes.size == 2 }
976
+ d.run do
977
+ res = post("/tag1", {"json"=>{"a"=>1}.to_json, "time"=>time_i.to_s}, {"Origin"=>"http://bar.com"})
978
+ res_codes << res.code
979
+ res = post("/tag2", {"json"=>{"a"=>1}.to_json, "time"=>time_i.to_s}, {"Origin"=>"http://bar.com"})
980
+ res_codes << res.code
981
+ end
982
+ assert_equal ["403", "403"], res_codes
983
+ end
984
+
985
+ def test_add_query_params
986
+ d = create_driver(config + "add_query_params true")
987
+ assert_equal true, d.instance.add_query_params
988
+
989
+ time = event_time("2011-01-02 13:14:15 UTC")
990
+ time_i = time.to_i
991
+ events = [
992
+ ["tag1", time, {"a"=>1, "QUERY_A"=>"b"}],
993
+ ["tag2", time, {"a"=>2, "QUERY_A"=>"b"}],
994
+ ]
995
+ res_codes = []
996
+ res_bodies = []
997
+
998
+ d.run do
999
+ events.each do |tag, _t, record|
1000
+ res = post("/#{tag}?a=b", {"json"=>record.to_json, "time"=>time_i.to_s})
1001
+ res_codes << res.code
1002
+ end
1003
+ end
1004
+ assert_equal ["200", "200"], res_codes
1005
+ assert_equal [], res_bodies
1006
+ assert_equal events, d.events
1007
+ end
1008
+
1009
+ $test_in_http_connection_object_ids = []
1010
+ $test_in_http_content_types = []
1011
+ $test_in_http_content_types_flag = false
1012
+ module ContentTypeHook
1013
+ def initialize(*args)
1014
+ @io_handler = nil
1015
+ super
1016
+ end
1017
+ def on_headers_complete(headers)
1018
+ super
1019
+ if $test_in_http_content_types_flag
1020
+ $test_in_http_content_types << self.content_type
1021
+ end
1022
+ end
1023
+
1024
+ def on_message_begin
1025
+ super
1026
+ if $test_in_http_content_types_flag
1027
+ $test_in_http_connection_object_ids << @io_handler.object_id
1028
+ end
1029
+ end
1030
+ end
1031
+
1032
+ class Fluent::Plugin::HttpInput::Handler
1033
+ prepend ContentTypeHook
1034
+ end
1035
+
1036
+ def test_if_content_type_is_initialized_properly
1037
+ # This test is to check if Fluent::HttpInput::Handler's @content_type is initialized properly.
1038
+ # Especially when in Keep-Alive and the second request has no 'Content-Type'.
1039
+
1040
+ begin
1041
+ d = create_driver
1042
+
1043
+ $test_in_http_content_types_flag = true
1044
+ d.run do
1045
+ # Send two requests the second one has no Content-Type in Keep-Alive
1046
+ Net::HTTP.start("127.0.0.1", @port) do |http|
1047
+ req = Net::HTTP::Post.new("/foodb/bartbl", {"connection" => "keepalive", "Content-Type" => "application/json"})
1048
+ http.request(req)
1049
+ req = Net::HTTP::Get.new("/foodb/bartbl", {"connection" => "keepalive"})
1050
+ http.request(req)
1051
+ end
1052
+
1053
+ end
1054
+ ensure
1055
+ $test_in_http_content_types_flag = false
1056
+ end
1057
+ assert_equal(['application/json', ''], $test_in_http_content_types)
1058
+ # Asserting keepalive
1059
+ assert_equal $test_in_http_connection_object_ids[0], $test_in_http_connection_object_ids[1]
1060
+ end
1061
+
1062
+ def get(path, params, header = {})
1063
+ http = Net::HTTP.new("127.0.0.1", @port)
1064
+ req = Net::HTTP::Get.new(path, header)
1065
+ http.request(req)
1066
+ end
1067
+
1068
+ def options(path, params, header = {})
1069
+ http = Net::HTTP.new("127.0.0.1", @port)
1070
+ req = Net::HTTP::Options.new(path, header)
1071
+ http.request(req)
1072
+ end
1073
+
1074
+ def post(path, params, header = {}, &block)
1075
+ http = Net::HTTP.new("127.0.0.1", @port)
1076
+ req = Net::HTTP::Post.new(path, header)
1077
+ block.call(http, req) if block
1078
+ if params.is_a?(String)
1079
+ unless header.has_key?('Content-Type')
1080
+ header['Content-Type'] = 'application/octet-stream'
1081
+ end
1082
+ req.body = params
1083
+ else
1084
+ unless header.has_key?('Content-Type')
1085
+ header['Content-Type'] = 'application/x-www-form-urlencoded'
1086
+ end
1087
+ req.set_form_data(params)
1088
+ end
1089
+ http.request(req)
1090
+ end
1091
+
1092
+ def compress_gzip(data)
1093
+ io = StringIO.new
1094
+ io.binmode
1095
+ Zlib::GzipWriter.wrap(io) { |gz| gz.write data }
1096
+ return io.string
1097
+ end
1098
+
1099
+ def include_http_header?(record)
1100
+ record.keys.find { |header| header.start_with?('HTTP_') }
1101
+ end
1102
+ end