1 files changed, 29 insertions, 0 deletions
diff --git a/test/test_restAPI.py b/test/test_restAPI.py
new file mode 100644
index 0000000..ac00996
--- /dev/null
+++ b/test/test_restAPI.py
@@ -0,0 +1,29 @@
+import unittest
+from unittest.mock import patch, MagicMock
+
+import dlpcvp
+
+class TestUrlopen(unittest.TestCase):
+    @patch('dlpcvp.urllib.request.urlopen')
+    def test_get_version_called_url(self, mock_obj):
+        result = dlpcvp.get_version_from_pypi('m2crypto')
+        url = dlpcvp.PyPI_base.format('m2crypto')
+        req_obj = mock_obj.call_args[0][0]
+        self.assertEqual(req_obj.full_url, url)
+        mock_obj.assert_called_once()
+
+    @unittest.skip('Not done yet')
+    @patch('dlpcvp.urllib.request.urlopen')
+    def test_get_version_returned(self, mock_obj):
+        mock_obj.read.return_value = {
+            'info': {
+                    "name": "M2CryptoTest",
+                    "version": '42'
+            }
+        }
+        mock_obj.info.return_value = 'AAA'
+        expected = ('M2CryptoTest', '0.30.1', 'AAA')
+
+        result = dlpcvp.get_version_from_pypi('m2crypto')
+        self.assertEqual(result, expected)
+        mock_obj.assert_called_once()