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_port(self): responses.add(responses.GET, 'https://dshield.org/api/port/80?json', body='{"port":80}', match_querystring=True, content_type='text/json') responses.add(responses.GET, 'https://dshield.org/api/port/badport?json', body='{"error":"bad port number"}', status=200, match_querystring=True, content_type='text/json') self.assertEquals(dshield.port('80'), {'port': 80}) self.assertEquals(dshield.port(80), {'port': 80}) self.assertEquals(dshield.port('80', dshield.JSON), '{"port":80}') self.assertRaises(dshield.Error, dshield.port, 'badport')