path: root/test/test_restAPI.py

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()