def test_endpoint(self, mock_config, mock_temp, mock_stfrtime):
f = Flare()
f._ask_for_email = lambda: None
f._open_tarfile()
f._tar.close()
with self.assertRaises(Exception) as cm:
f.upload()
self.assertEqual(str(cm.exception), "Your request is incorrect: Invalid inputs: 'API key unknown'")
def test_endpoint(self, mock_config, mock_temp, mock_stfrtime):
if os.environ['FLARE_BROKEN']:
raise unittest.case.SkipTest('Flare broken, acknowledged')
f = Flare()
f._ask_for_email = lambda: None
f._open_tarfile()
f._tar.close()
with self.assertRaises(Exception) as cm:
f.upload()
self.assertEqual(str(cm.exception), "Your request is incorrect: Invalid inputs: 'API key unknown'")
def test_endpoint(self, mock_config, mock_temp, mock_stfrtime, mock_os_remove):
if os.environ.get('FLARE_BROKEN'):
raise unittest.case.SkipTest('Flare broken, acknowledged')
f = Flare()
f._ask_for_email = lambda: None
f._open_tarfile()
f._tar.close()
with self.assertRaises(Exception) as cm:
f.upload()
self.assertEqual(str(cm.exception), "403 Client Error: Forbidden for url: https://app.datadoghq.com/support/flare?api_key=APIKEY")
def test_endpoint(self, mock_config, mock_temp, mock_stfrtime, mock_os_remove):
if os.environ.get('FLARE_BROKEN'):
raise unittest.case.SkipTest('Flare broken, acknowledged')
f = Flare()
f._ask_for_email = lambda: None
f._open_tarfile()
f._tar.close()
with self.assertRaises(Exception) as cm:
f.upload()
self.assertEqual(str(cm.exception), "Your request is incorrect: Invalid inputs: 'API key unknown'")
def test_upload_no_case(self, mock_config, mock_tempdir, mock_stfrtime,
mock_version, mock_requests, mock_os_remove):
f = Flare()
f._ask_for_email = lambda: '*****@*****.**'
f._open_tarfile()
f._tar.close()
assert not mock_requests.called
f.upload()
assert mock_requests.called
args, kwargs = mock_requests.call_args_list[0]
self.assertEqual(
args, ('https://app.datadoghq.com/support/flare?api_key=APIKEY', ))
self.assertEqual(
kwargs['files']['flare_file'].name,
os.path.join(get_mocked_temp(), "stackstate-agent-1.tar.bz2"))
self.assertEqual(kwargs['data']['case_id'], None)
self.assertEqual(kwargs['data']['email'], '*****@*****.**')
assert kwargs['data']['hostname']
def test_upload_with_case(self, mock_config, mock_tempdir, mock_stfrtime, mock_version, mock_requests):
f = Flare(case_id=1337)
f._ask_for_email = lambda: '*****@*****.**'
f._open_tarfile()
f._tar.close()
assert not mock_requests.called
f.upload()
assert mock_requests.called
args, kwargs = mock_requests.call_args_list[0]
self.assertEqual(
args,
('https://6-6-6-flare.agent.datadoghq.com/support/flare/1337?api_key=APIKEY',)
)
self.assertEqual(
kwargs['files']['flare_file'].name,
os.path.join(get_mocked_temp(), "datadog-agent-1.tar.bz2")
)
self.assertEqual(kwargs['data']['case_id'], 1337)
self.assertEqual(kwargs['data']['email'], '*****@*****.**')
assert kwargs['data']['hostname']
def test_upload_with_case_proxy(self, mock_config, mock_tempdir, mock_stfrtime, mock_version, mock_requests):
f = Flare(case_id=1337)
f._ask_for_email = lambda: '*****@*****.**'
f._open_tarfile()
f._tar.close()
assert not mock_requests.called
f.upload()
assert mock_requests.called
args, kwargs = mock_requests.call_args_list[0]
self.assertEqual(
args,
('https://6-6-6-flare.agent.datadoghq.com/support/flare/1337?api_key=APIKEY',)
)
self.assertEqual(
kwargs['files']['flare_file'].name,
os.path.join(get_mocked_temp(), "datadog-agent-1.tar.bz2")
)
self.assertEqual(kwargs['data']['case_id'], 1337)
self.assertEqual(kwargs['data']['email'], '*****@*****.**')
assert kwargs['data']['hostname']
assert kwargs['proxies']['https'] == 'http://*****:*****@proxy.host.com:3128'
assert not kwargs['verify']
