add tests for users command

src/tests/__init__.py

@@ -0,0 +1 @@
+# -*- encoding: utf-8 -*-

src/tests/commands/__init__.py

@@ -0,0 +1 @@
+# -*- encoding: utf-8 -*-

src/tests/commands/test_users.py

@@ -0,0 +1,77 @@
+# -*- encoding: utf-8 -*-
+"""
+Tests users command
+"""
+import json
+from collections import namedtuple
+
+from src.commands.users import run
+from ..docker_hub_client import \
+    NoResultsUsersTestingDockerHubClient, WithUserResultsTestingDockerHubClient
+from ..helpers import generate_results
+
+
+Args = namedtuple('args', 'username page format all_pages')
+
+
+def test_no_users(capsys):
+    """ When there is no user profile found by API """
+    docker_hub_client = NoResultsUsersTestingDockerHubClient()
+    username = 'test'
+    run(
+        docker_hub_client,
+        Args(
+          username, page=1, format='json',
+          all_pages=False
+        )
+    )
+
+    captured = capsys.readouterr()
+    assert captured.out == f'Error fetching profile for: {username}\n'
+
+def test_no_users_and_all_pages(capsys):
+    """ When all_page applied during no user found by API """
+    docker_hub_client = NoResultsUsersTestingDockerHubClient()
+    username = 'test'
+    run(
+        docker_hub_client,
+        Args(
+            username, page=1, format='json',
+            all_pages=True
+        )
+    )
+
+    captured = capsys.readouterr()
+    assert captured.out == f'Error fetching profile for: {username}\n'
+
+def test_with_users(capsys):
+    """ When there is user returned API """
+    docker_hub_client = WithUserResultsTestingDockerHubClient()
+    username = 'test'
+    run(
+        docker_hub_client,
+        Args(
+            username, page=1, format='json',
+            all_pages=False
+        )
+    )
+    results = generate_results(1)
+
+    captured = capsys.readouterr()
+    assert json.loads(captured.out) == results
+
+def test_with_users_and_all_pages(capsys):
+    """ When all_pageas applied there is user returned API """
+    docker_hub_client = WithUserResultsTestingDockerHubClient()
+    username = 'test'
+    run(
+        docker_hub_client,
+        Args(
+            username, page=1, format='json',
+            all_pages=True
+        )
+    )
+    results = generate_results(1)
+
+    captured = capsys.readouterr()
+    assert json.loads(captured.out) == results

src/tests/docker_hub_client.py

@@ -2,7 +2,7 @@
 """
 Tests for Dockerhub API client
 """
-from ..libs.docker_hub_client import DockerHubClient
+from src.libs.docker_hub_client import DockerHubClient
 from .helpers import generate_results
 
 
@@ -20,6 +20,10 @@ def do_request(self, url, method='GET', data=None):
             content = self._fake_login()
         return {'content': content, 'code': 200}
 
+class NoResultsUsersTestingDockerHubClient(BaseTestingDockerHubClient):
+    """ When API returns no results for users endpoint """
+    def do_request(self, url, method='GET', data=None):
+        return {'content': None, 'code': 404}
 
 class WithResultsTestingDockerHubClient(BaseTestingDockerHubClient):
     """ When API returns results """
@@ -32,3 +36,11 @@ def do_request(self, url, method='GET', data=None):
         if 'login' in url:
             content = self._fake_login()
         return {'content': content, 'code': 200}
+
+class WithUserResultsTestingDockerHubClient(BaseTestingDockerHubClient):
+    """ When API returns user result """
+    def do_request(self, url, method='GET', data=None):
+        content = generate_results(1)[0]
+        if 'login' in url:
+            content = self._fake_login()
+        return {'content': content, 'code': 200}