def test_setup_header_with_blank_header(self):
test_header = {}
return_expected = {
'User-Agent': FuzzWord('FuzzingTool Requester Agent'),
'Accept-Encoding': FuzzWord('gzip, deflate')
}
returned_data = Requester._Requester__setup_header(
Requester, test_header)
self.assertIsInstance(returned_data, dict)
self.assertIsInstance(returned_data['User-Agent'], FuzzWord)
self.assertIsInstance(returned_data['Accept-Encoding'], FuzzWord)
self.assertDictEqual(returned_data, return_expected)
def test_setup_header_with_filled_header(self):
test_header = {'User-Agent': 'Test User Agent', 'Content-Length': '21'}
return_expected = {
'User-Agent': FuzzWord('Test User Agent'),
'Accept-Encoding': FuzzWord('gzip, deflate')
}
returned_data = Requester._Requester__setup_header(
Requester, test_header)
self.assertIsInstance(returned_data, dict)
self.assertIsInstance(returned_data['User-Agent'], FuzzWord)
self.assertIsInstance(returned_data['Accept-Encoding'], FuzzWord)
self.assertEqual(('Content-Length' in returned_data.keys()), False)
self.assertDictEqual(returned_data, return_expected)