From d20b0b5b8ac02596dd51209804f4f2f9d9602796 Mon Sep 17 00:00:00 2001
From: Badlop <badlop@process-one.net>
Date: Mon, 6 Jun 2022 22:51:27 +0200
Subject: [PATCH] Add tests for IPv4 and IPv6 parsing (#8)

---
 src/esip_codec.erl | 46 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)

diff --git a/src/esip_codec.erl b/src/esip_codec.erl
index e9e9596..d06d844 100644
--- a/src/esip_codec.erl
+++ b/src/esip_codec.erl
@@ -52,6 +52,10 @@
 -include("esip.hrl").
 -include("esip_lib.hrl").
 
+-ifdef(TEST).
+-include_lib("eunit/include/eunit.hrl").
+-endif.
+
 %%%===================================================================
 %%% API
 %%%===================================================================
@@ -1370,3 +1374,45 @@ test_loop(P, _, 0, T) ->
 test_loop(P, Msg, N, T) ->
     decode(Msg),
     test_loop(P, Msg, N-1, T).
+
+-ifdef(TEST).
+
+decode_via_test(V, H, P) ->
+    [U] = decode_via(V),
+    ?assertEqual(H, U#via.host),
+    ?assertEqual(P, U#via.port).
+
+decode_via_test() ->
+    decode_via_test(<<"SIP/2.0/UDP host4.example.com:5060;branch=z9hG4bKkdju43234">>,
+                    <<"host4.example.com">>, 5060),
+    decode_via_test(<<"SIP/2.0/UDP 192.0.2.15;;,;,,">>,
+                    <<"192.0.2.15">>, undefined),
+    decode_via_test(<<"SIP/2.0/UDP [::ffff:192.0.2.10]:19823;branch=z9hG4bKbh19">>,
+                    <<"[::ffff:192.0.2.10]">>, 19823),
+    decode_via_test(<<"SIP/2.0/UDP [2001:db8::9:1]:6050;branch=z9hG4bKas3-111">>,
+                    <<"[2001:db8::9:1]">>, 6050),
+    decode_via_test(<<"SIP/2.0/UDP [2001:db8::9:1];branch=z9hG4bKas3-111">>,
+                    <<"[2001:db8::9:1]">>, undefined),
+    decode_via_test(<<"SIP/2.0/UDP [2001:db8::20];branch=z9hG4bKas3-111">>,
+                    <<"[2001:db8::20]">>, undefined),
+    decode_via_test(<<"SIP/2.0/UDP [2001:db8::10:5070];branch=z9hG4bKas3-111">>,
+                    <<"[2001:db8::10:5070]">>, undefined),
+    decode_via_test(<<"SIP/2.0/UDP [2001:db8::10]:5070;branch=z9hG4bKas3-111">>,
+                    <<"[2001:db8::10]">>, 5070),
+    ok.
+
+decode_uri_host_test(HP, H, P) ->
+    U = decode_uri_host(HP, #uri{}),
+    ?assertEqual(H, U#uri.host),
+    ?assertEqual(P, U#uri.port).
+
+decode_uri_host_test() ->
+    decode_uri_host_test(<<"192.0.1.98:5222">>,
+                         <<"192.0.1.98">>, 5222),
+    decode_uri_host_test(<<"2001:0db8:85a3:0000:0000:8a2e:0370:7334:5222">>,
+                         <<"2001:0db8:85a3:0000:0000:8a2e:0370:7334">>, 5222),
+    decode_uri_host_test(<<"[2001:0db8:85a3:0000:0000:8a2e:0370:7334]:5222">>,
+                         <<"[2001:0db8:85a3:0000:0000:8a2e:0370:7334]">>, 5222),
+    ok.
+
+-endif.