def reload(self, napps=None):
"""Reload a NApp or all NApps.
Args:
napps (list): NApp list to be reloaded.
Raises:
requests.HTTPError: When there's a server error.
"""
client = NAppsClient(self._config)
client.reload_napps(napps)
class TestNAppsClient(unittest.TestCase):
"""Test the class NAppsClient."""
def setUp(self):
"""Execute steps before each tests."""
with tempfile.TemporaryDirectory() as tmp_dir:
config_file = '{}.kytosrc'.format(tmp_dir)
kytos_config = KytosConfig(config_file)
kytos_config.save_token('user', 'token')
self.napps_client = NAppsClient()
self.napps_client._config = kytos_config.config
self.napps_client._config.set('kytos', 'api', 'endpoint')
self.napps_client._config.set('napps', 'api', 'endpoint')
@staticmethod
def _expected_response(status_code):
"""Expected response mock."""
response = MagicMock()
response.content = '{"napps": []}'.encode()
response.status_code = status_code
return response
@patch('requests.get')
def test_get_napps(self, mock_request):
"""Test get_napps method."""
mock_request.return_value = self._expected_response(200)
self.napps_client.get_napps()
mock_request.assert_called_with('endpoint/napps/', json=[])
@patch('requests.get')
def test_get_napp(self, mock_request):
"""Test get_napp method."""
mock_request.return_value = self._expected_response(200)
self.napps_client.get_napp('username', 'name')
endpoint = 'endpoint/napps/username/name/'
mock_request.assert_called_with(endpoint, json=[])
@patch('requests.get')
def test_reload_napps__all(self, mock_request):
"""Test reload_napps method to all napps."""
mock_request.return_value = self._expected_response(200)
self.napps_client.reload_napps()
endpoint = 'endpoint/api/kytos/core/reload/all'
mock_request.assert_called_with(endpoint, json=[])
@patch('requests.get')
def test_reload_napps__any(self, mock_request):
"""Test reload_napps method to any napp."""
mock_request.return_value = self._expected_response(200)
napps = [('user', 'napp', 'version')]
self.napps_client.reload_napps(napps)
endpoint = 'endpoint/api/kytos/core/reload/user/napp'
mock_request.assert_called_with(endpoint, json=[])
@patch('requests.post')
@patch('requests.get')
@patch('configparser.ConfigParser.set')
@patch('configparser.ConfigParser.get', return_value='value')
@patch('configparser.ConfigParser.has_option', return_value=False)
@patch('kytos.utils.decorators.getpass', return_value='password')
@patch('builtins.input', return_value='username')
def test_upload_napp(self, *args):
"""Test upload_napp method."""
(_, _, _, _, _, mock_get, mock_post) = args
mock_get.return_value = self._expected_response(201)
mock_post.return_value = self._expected_response(201)
metadata = MagicMock()
self.napps_client.upload_napp(metadata, 'package')
mock_post.assert_called_with('value/napps/',
data=metadata,
files={'file': 'package'})
@patch('requests.delete')
@patch('requests.get')
@patch('configparser.ConfigParser.set')
@patch('configparser.ConfigParser.get', return_value='value')
@patch('configparser.ConfigParser.has_option', return_value=False)
@patch('kytos.utils.decorators.getpass', return_value='password')
@patch('builtins.input', return_value='username')
def test_delete(self, *args):
"""Test delete method."""
(_, _, _, _, _, mock_get, mock_delete) = args
mock_get.return_value = self._expected_response(201)
mock_delete.return_value = self._expected_response(201)
self.napps_client.delete('user', 'napp')
endpoint = 'value/napps/user/napp/'
token = self.napps_client._config.get('auth', 'token')
mock_delete.assert_called_with(endpoint, json={'token': token})