def test_no_functions_throw_exceptions(self): try: dshield.backscatter() dshield.handler() dshield.infocon() dshield.ip('8.8.8.8') dshield.port(80) dshield.portdate(80) dshield.topports() dshield.topips() dshield.sources() dshield.porthistory(80) dshield.asnum(1) dshield.dailysummary() dshield.daily404summary(datetime.date(2011, 12, 1)) dshield.daily404detail(datetime.date(2011, 12, 1)) dshield.glossary() dshield.webhoneypotsummary(datetime.date(2011, 12, 1)) dshield.webhoneypotbytype(datetime.date(2011, 12, 1)) except requests.RequestException: # don't care about network errors pass except Exception: # anything else is a fail self.assertTrue(False)
def test_portdate(self): responses.add(responses.GET, 'https://dshield.org/api/portdate/80/2011-07-03?json', body='{"portdate":"test"}', match_querystring=True, content_type='text/json') responses.add(responses.GET, 'https://dshield.org/api/portdate/80?json', body='{"portdate":"test"}', match_querystring=True, content_type='text/json') responses.add(responses.GET, 'https://dshield.org/api/portdate/badport?json', body='{"error":"bad port number"}', status=200, match_querystring=True, content_type='text/json') self.assertEquals(dshield.portdate('80'), {'portdate': 'test'}) self.assertEquals(dshield.portdate(80), {'portdate': 'test'}) self.assertEquals(dshield.portdate('80', datetime.date(2011, 7, 3)), {'portdate': 'test'}) self.assertEquals(dshield.portdate('80', '2011-07-03'), {'portdate': 'test'}) self.assertEquals(dshield.portdate('80', return_format=dshield.JSON), '{"portdate":"test"}') self.assertRaises(dshield.Error, dshield.portdate, 'badport')