Added test cases for aiosocks.helpers

tests/test_helpers.py

@@ -0,0 +1,85 @@
+import pytest
+import aiosocks
+
+
+def test_socks4_auth1():
+    with pytest.raises(ValueError):
+        aiosocks.Socks4Auth(None)
+
+
+def test_socks4_auth2():
+    auth = aiosocks.Socks4Auth('usr', encoding='ascii')
+    assert auth.login == b'usr'
+
+
+def test_socks4_auth3():
+    auth = aiosocks.Socks4Auth('usrё', encoding='utf-8')
+    assert auth.login == b'usr\xd1\x91'
+
+
+def test_socks5_auth1():
+    with pytest.raises(ValueError):
+        aiosocks.Socks5Auth(None, '')
+
+
+def test_socks5_auth2():
+    with pytest.raises(ValueError):
+        aiosocks.Socks5Auth('', None)
+
+
+def test_socks5_auth3():
+    auth = aiosocks.Socks5Auth('usr', 'pwd', encoding='ascii')
+    assert auth.login == b'usr'
+    assert auth.password == b'pwd'
+
+
+def test_socks5_auth4():
+    auth = aiosocks.Socks5Auth('usrё', 'pwdё', encoding='utf-8')
+    assert auth.login == b'usr\xd1\x91'
+    assert auth.password == b'pwd\xd1\x91'
+
+
+def test_socks4_addr1():
+    with pytest.raises(ValueError):
+        aiosocks.Socks4Addr(None)
+
+
+def test_socks4_addr2():
+    addr = aiosocks.Socks4Addr('localhost')
+    assert addr.host == 'localhost'
+    assert addr.port == 1080
+
+
+def test_socks4_addr3():
+    addr = aiosocks.Socks4Addr('localhost', 1)
+    assert addr.host == 'localhost'
+    assert addr.port == 1
+
+
+def test_socks4_addr4():
+    addr = aiosocks.Socks4Addr('localhost', None)
+    assert addr.host == 'localhost'
+    assert addr.port == 1080
+
+
+def test_socks5_addr1():
+    with pytest.raises(ValueError):
+        aiosocks.Socks5Addr(None)
+
+
+def test_socks5_addr2():
+    addr = aiosocks.Socks5Addr('localhost')
+    assert addr.host == 'localhost'
+    assert addr.port == 1080
+
+
+def test_socks5_addr3():
+    addr = aiosocks.Socks5Addr('localhost', 1)
+    assert addr.host == 'localhost'
+    assert addr.port == 1
+
+
+def test_socks5_addr4():
+    addr = aiosocks.Socks5Addr('localhost', None)
+    assert addr.host == 'localhost'
+    assert addr.port == 1080