From e09a85680540029df22001073f73b77eec67773d Mon Sep 17 00:00:00 2001
From: Cameron Bytheway <bytheway.cameron@gmail.com>
Date: Fri, 24 Jun 2016 19:34:11 -0600
Subject: send resp_headers with websocket responses

---
 test/ws_SUITE_data/ws_subprotocol.erl | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)
 create mode 100644 test/ws_SUITE_data/ws_subprotocol.erl

(limited to 'test/ws_SUITE_data')

diff --git a/test/ws_SUITE_data/ws_subprotocol.erl b/test/ws_SUITE_data/ws_subprotocol.erl
new file mode 100644
index 0000000..a042dd2
--- /dev/null
+++ b/test/ws_SUITE_data/ws_subprotocol.erl
@@ -0,0 +1,18 @@
+%% Feel free to use, reuse and abuse the code in this file.
+
+-module(ws_subprotocol).
+
+-export([init/2]).
+-export([websocket_handle/3]).
+-export([websocket_info/3]).
+
+init(Req, Opts) ->
+	[Protocol | _] = cowboy_req:parse_header(<<"sec-websocket-protocol">>, Req),
+	Req2 = cowboy_req:set_resp_header(<<"sec-websocket-protocol">>, Protocol, Req),
+	{cowboy_websocket, Req2, Opts, 1000}.
+
+websocket_handle(_Frame, Req, State) ->
+	{ok, Req, State}.
+
+websocket_info(_Info, Req, State) ->
+	{ok, Req, State}.
-- 
cgit v1.2.3