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_asnum(self): responses.add(responses.GET, 'https://dshield.org/api/asnum/10/4837?json', body='{"asnum":"test"}', match_querystring=True, content_type='text/json') responses.add(responses.GET, 'https://dshield.org/api/asnum/10?json', body='{"asnum":"test"}', match_querystring=True, content_type='text/json') data = {'asnum': 'test'} self.assertEquals(dshield.asnum(10), data) self.assertEquals(dshield.asnum('10'), data) self.assertEquals(dshield.asnum(10, 4837), data) self.assertEquals(dshield.asnum(10, '4837'), data) self.assertEquals(dshield.asnum(10, return_format=dshield.JSON), '{"asnum":"test"}')