Move state record to .hrl file for easier embeding in whitebox tests

include/escalus_tcp.hrl

@@ -0,0 +1,32 @@
+%%==============================================================================
+%% Copyright 2015 Erlang Solutions Ltd.
+%%
+%% Licensed under the Apache License, Version 2.0 (the "License");
+%% you may not use this file except in compliance with the License.
+%% You may obtain a copy of the License at
+%%
+%% http://www.apache.org/licenses/LICENSE-2.0
+%%
+%% Unless required by applicable law or agreed to in writing, software
+%% distributed under the License is distributed on an "AS IS" BASIS,
+%% WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+%% See the License for the specific language governing permissions and
+%% limitations under the License.
+%%==============================================================================
+
+%% Stream management automation
+%%               :: {Auto-ack?,                H,         counting Hs?}.
+-type sm_state() :: {boolean(), non_neg_integer(), 'active'|'inactive'}.
+
+-record(state, {owner,
+                socket,
+                parser,
+                filter_pred,
+                ssl = false,
+                compress = false,
+                event_client,
+                on_reply,
+                on_request,
+                active = true,
+                sm_state = {true, 0, inactive} :: sm_state(),
+                replies = []}).

src/escalus_tcp.erl

@@ -38,25 +38,13 @@
          set_active/2,
          recv/1]).
 
+-ifdef(EUNIT_TEST).
+-compile(export_all).
+-endif.
+
 -define(WAIT_FOR_SOCKET_CLOSE_TIMEOUT, 200).
 -define(SERVER, ?MODULE).
-
-%% Stream management automation
-%%               :: {Auto-ack?,                H,         counting Hs?}.
--type sm_state() :: {boolean(), non_neg_integer(), 'active'|'inactive'}.
-
--record(state, {owner,
-                socket,
-                parser,
-                filter_pred,
-                ssl = false,
-                compress = false,
-                event_client,
-                on_reply,
-                on_request,
-                active = true,
-                sm_state = {true, 0, inactive} :: sm_state(),
-                replies = []}).
+-include("escalus_tcp.hrl").
 
 %%%===================================================================
 %%% API