SettingUpClientServer (#7)

Reviewed-on: #7

tests/test_client.py

@@ -0,0 +1,98 @@
+import json
+
+from fastapi import HTTPException
+from starlette import status
+
+from client import SheerkaClient, parse_arguments
+from mockserver import MockServer
+
+
+def test_i_can_start_with_a_default_hostname():
+    parsed = parse_arguments([])
+
+    assert parsed.hostname == "http://localhost"
+    assert parsed.port == 56356
+
+
+def test_i_can_override_hostname_and_port():
+    parsed = parse_arguments(["new_host", "--port", "1515"])
+
+    assert parsed.hostname == "new_host"
+    assert parsed.port == 1515
+
+    parsed = parse_arguments(["new_host", "-p", "1515"])
+
+    assert parsed.hostname == "new_host"
+    assert parsed.port == 1515
+
+
+def test_i_can_provide_user_and_password():
+    parsed = parse_arguments(["--username", "my_user", "--password", "my_password"])
+    assert parsed.username == "my_user"
+    assert parsed.password == "my_password"
+
+    parsed = parse_arguments(["-u", "my_user", "-P", "my_password"])
+    assert parsed.username == "my_user"
+    assert parsed.password == "my_password"
+
+
+def test_i_can_manage_when_no_server():
+    client = SheerkaClient("http://localhost", 80)
+    res = client.check_url()
+
+    assert res.status is False
+    assert res.message == "Connection refused."
+
+
+def test_i_can_manage_when_resource_is_not_found():
+    with MockServer([]):
+        client = SheerkaClient("http://localhost", 5000)
+        res = client.check_url()
+
+        assert not res.status
+        assert res.message == '{"detail":"Not Found"}'
+
+
+def test_i_can_connect_to_a_server():
+    with MockServer([{
+        "path": "/",
+        "response": "Hello world"
+    }]):
+        client = SheerkaClient("http://localhost", 5000)
+        res = client.check_url()
+        assert res.status
+        assert res.message == '"Hello world"'
+
+
+def test_i_can_authenticate_with_valid_credentials():
+    with MockServer([{
+        "path": "/",
+        "response": "Hello world"
+    }, {
+        "method": "post",
+        "path": "/token",
+        "response": {"access_token": "xxxx", "token_type": "bearer"}
+    }]):
+        client = SheerkaClient("http://localhost", 5000)
+        res = client.connect("valid_username", "valid_password")
+        assert res.status
+        assert res.message == "Connected as valid_username"
+
+
+def test_i_can_manage_when_authentication_fails():
+    with MockServer([{
+        "path": "/",
+        "response": "Hello world"
+    }, {
+        "method": "post",
+        "path": "/token",
+        "exception": HTTPException(
+            status_code=status.HTTP_401_UNAUTHORIZED,
+            detail="Incorrect username or password",
+            headers={"WWW-Authenticate": "Bearer"},
+        )
+    }]):
+        client = SheerkaClient("http://localhost", 5000)
+        res = client.connect("username", "wrong_password")
+        assert not res.status
+        assert res.message == 'Incorrect username or password'