def create_agent(): agent = PyPiSpyAgent( test_settings.SERVER_NAME, test_settings.VENVS, test_settings.VENVS_DIR, test_settings.API_KEY, test_settings.API_URL, ) return agent
def test_run(self): old_call_pypispy_api = PyPiSpyAgent.call_pypispy_api PyPiSpyAgent.call_pypispy_api = mock.Mock() agent = PyPiSpyAgent(test_settings.SERVER_NAME, test_settings.VENVS, test_settings.VENVS_DIR, test_settings.API_KEY) agent.run() # test if called for each virtual environment self.assertEqual( agent.call_pypispy_api.call_count, len(test_settings.VENVS), msg="Call count doesn't match number of environments. \ Expected '%s' but got '%s'" % (len(test_settings.VENVS), agent.call_pypispy_api.call_count)) PyPiSpyAgent.call_pypispy_api = old_call_pypispy_api
def test_inspect_venv(self): agent = PyPiSpyAgent(test_settings.SERVER_NAME, test_settings.VENVS, test_settings.VENVS_DIR, test_settings.API_KEY) data = agent.inspect_venv(agent.venvs[0]) server_name = data.get('server_name') api_key = data.get('api_key') package_info = data.get('package_info') self.assertEqual(test_settings.SERVER_NAME, server_name, msg="Expected server name to be '%s' but go '%s'"\ % (test_settings.SERVER_NAME, server_name)) self.assertEqual(test_settings.API_KEY, api_key, msg="Expected api key to be '%s' but go '%s'"\ % (test_settings.API_KEY, api_key)) self.assertEqual(package_info, self.expected_package_list, msg="Got wrong package info")