from unittest import TestCase from senpy.testing import patch_requests from senpy.client import Client from senpy.models import Results, Plugins, Error from senpy.plugins import AnalysisPlugin class ModelsTest(TestCase): def setUp(self): self.host = '0.0.0.0' self.port = 5000 def test_client(self): endpoint = 'http://dummy/' client = Client(endpoint) with patch_requests(Results()) as (request, response): resp = client.analyse('hello') assert isinstance(resp, Results) request.assert_called_with( url=endpoint + '/', method='GET', params={'input': 'hello'}) with patch_requests(Error('Nothing')) as (request, response): try: client.analyse(input='hello', algorithm='NONEXISTENT') raise Exception('Exceptions should be raised. This is not golang') except Error: pass request.assert_called_with( url=endpoint + '/', method='GET', params={'input': 'hello', 'algorithm': 'NONEXISTENT'}) def test_plugins(self): endpoint = 'http://dummy/' client = Client(endpoint) plugins = Plugins() p1 = AnalysisPlugin({'name': 'AnalysisP1', 'version': 0, 'description': 'No'}) plugins.plugins = [p1, ] with patch_requests(plugins) as (request, response): response = client.plugins() assert isinstance(response, dict) assert len(response) == 1 assert "AnalysisP1" in response request.assert_called_with( url=endpoint + '/plugins', method='GET', params={})