def test_vt_name_check(self): vt_api = os.environ["VT_API"] self.assertIsNone(ub.vt_name_check('asdf', vt_api)) vt_name_data = ub.vt_name_check("github.com", vt_api) self.assertIsInstance(vt_name_data, dict) self.assertIn('resolutions', vt_name_data) is_gh = False for resolution in vt_name_data['resolutions']: if resolution['ip_address'] == '192.30.252.130': is_gh = True self.assertTrue(is_gh) time.sleep(15) # VT rate limiting
def test_vt_rate_limiting(self): vt_api = os.environ["VT_API"] # Exceed 4x in 60 seconds data = ub.vt_hash_check("d41d8cd98f00b204e9800998ecf8427e", vt_api) self.assertIsInstance(data, dict) data = ub.vt_hash_check("d41d8cd98f00b204e9800998ecf8427e", vt_api) data = ub.vt_hash_check("d41d8cd98f00b204e9800998ecf8427e", vt_api) data = ub.vt_hash_check("d41d8cd98f00b204e9800998ecf8427e", vt_api) data = ub.vt_name_check("example.org", vt_api) self.assertIsNone(data) data = ub.vt_ip_check("192.30.252.130", vt_api) self.assertIsNone(data) data = ub.vt_hash_check("d41d8cd98f00b204e9800998ecf8427e", vt_api) self.assertIsNone(data) time.sleep(15)