Add tests for #81

diff --git a/client_test.go b/client_test.go
index d10a7567b759346ce0df76ad23832e9e9e638e0a..6cefd18e669750965b76de9fdeb83f8c664f538b 100644 (file)
--- a/client_test.go
+++ b/client_test.go
@@ -897,3 +897,44 @@ func TestPrepareTrackerAnnounce(t *testing.T) {
        assert.EqualValues(t, "localhost:1234", host)
        assert.EqualValues(t, "http://127.0.0.1:1234/announce?herp", urlToUse)
 }
+
+// Check that when the listen port is 0, all the protocols listened on have
+// the same port, and it isn't zero.
+func TestClientDynamicListenPortAllProtocols(t *testing.T) {
+       cl, err := NewClient(&TestingConfig)
+       require.NoError(t, err)
+       defer cl.Close()
+       assert.NotEqual(t, 0, missinggo.AddrPort(cl.ListenAddr()))
+       assert.Equal(t, missinggo.AddrPort(cl.utpSock.Addr()), missinggo.AddrPort(cl.tcpListener.Addr()))
+}
+
+func TestClientDynamicListenTCPOnly(t *testing.T) {
+       cfg := TestingConfig
+       cfg.DisableUTP = true
+       cl, err := NewClient(&cfg)
+       require.NoError(t, err)
+       defer cl.Close()
+       assert.NotEqual(t, 0, missinggo.AddrPort(cl.ListenAddr()))
+       assert.Nil(t, cl.utpSock)
+}
+
+func TestClientDynamicListenUTPOnly(t *testing.T) {
+       cfg := TestingConfig
+       cfg.DisableTCP = true
+       cl, err := NewClient(&cfg)
+       require.NoError(t, err)
+       defer cl.Close()
+       assert.NotEqual(t, 0, missinggo.AddrPort(cl.ListenAddr()))
+       assert.Nil(t, cl.tcpListener)
+}
+
+func TestClientDynamicListenPortNoProtocols(t *testing.T) {
+       cfg := TestingConfig
+       cfg.DisableTCP = true
+       cfg.DisableUTP = true
+       cl, err := NewClient(&cfg)
+       require.NoError(t, err)
+       defer cl.Close()
+       t.Log(cl.listenAddr)
+       assert.Nil(t, cl.ListenAddr())
+}