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_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 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_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")
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")
