Fake host

heroku · Mar 7, 2018 · 749def8 · 749def8
1 parent fcf991e
commit 749def8
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 4 deletions.
diff --git a/src/logplex_api_v3_channel_logs.erl b/src/logplex_api_v3_channel_logs.erl
@@ -86,7 +86,7 @@ prepare_logs([], Acc) ->
 prepare_logs([Msg | Logs], Acc) ->
     prepare_logs(Logs, [prepare_msg(Msg) | Acc]).
 
-prepare_msg(Msg) ->
-    ParsedMsg = logplex_syslog_utils:from_msg(Msg),
-    RFC5424Msg = logplex_syslog_utils:rfc5424(ParsedMsg),
+prepare_msg(RawMsg) ->
+    {Facility, Severity, Time, Source, Process, Msg} = logplex_syslog_utils:from_msg(RawMsg),
+    RFC5424Msg = logplex_syslog_utils:rfc5424(Facility, Severity, Time, "host", Source, Process, undefined, Msg),
     logplex_syslog_utils:frame([RFC5424Msg, $\n]).
diff --git a/test/logplex_api_v3_SUITE.erl b/test/logplex_api_v3_SUITE.erl
@@ -55,6 +55,7 @@ groups() ->
       [channel_logs_service_unavailable
       , channel_logs_not_authorized
       , fetch_channel_logs
+      , channel_logs_format
       ]},
      {sessions,
       [sessions_service_unavailable
@@ -587,10 +588,26 @@ fetch_channel_logs(Config0) ->
     [begin
          ?assertEqual(match, re:run(Line, Expected, [{capture, none}])),
          NBin = list_to_binary(integer_to_list(N)),
-         ?assertMatch(<<"61 <", NBin:1/binary, _/binary>>, Line)
+         ?assertMatch(<<"64 <", NBin:1/binary, _/binary>>, Line)
      end || {{N, Expected}, Line} <- lists:zip(ExpectedLogMsgs, Lines)],
     Config.
 
+channel_logs_format(Config0) ->
+    Config = create_channel_with_tokens(Config0),
+    Channel = ?config(channel, Config),
+    [{TokenName, TokenId} | _] = ?config(tokens, Config),
+    InputMsg = <<"<158>1 2018-03-02T23:22:34.901697+00:00 [email protected] heroku router - - at=info code=H81 desc=\"Blank app\" method=GET path=\"/\" host=direwolf-d99daa9eda.staging.herokuappdev.com request_id=56a11b1c-1f68-4658-80e4-8213a09a81c2 fwd=\"54.163.94.153\" dyno= connect= service= status=502 bytes= protocol=https">>,
+    ExpectedMsg = "290 <158>1 2018-03-02T23:22:34.901697+00:00 host heroku router - at=info code=H81 desc=\"Blank app\" method=GET path=\"/\" host=direwolf-d99daa9eda.staging.herokuappdev.com request_id=56a11b1c-1f68-4658-80e4-8213a09a81c2 fwd=\"54.163.94.153\" dyno= connect= service= status=502 bytes= protocol=https\n",
+    logplex_message:process_msg({msg, InputMsg}, list_to_binary(Channel), TokenId, TokenName),
+    Props = stream_channel_logs(Channel, Config),
+    Headers = proplists:get_value(headers, Props),
+    ?assertEqual("application/logplex-1", proplists:get_value("content-type", Headers)),
+    ?assertEqual("1", proplists:get_value("logplex-msg-count", Headers)),
+    ?assertEqual("chunked", proplists:get_value("transfer-encoding", Headers)),
+    ?assertEqual("close", proplists:get_value("connection", Headers)),
+	?assertMatch(ExpectedMsg, proplists:get_value(body, Props)).
+
+
 
 %% -----------------------------------------------------------------------------
 %% sessions