def get_attribute(self, **kwargs): client = AttributeRequest(self.username, self.apikey) keys = ['query', 'type'] params = self._cleanup_params(keys, **kwargs) if params.get('type') == 'tracker': return client.get_host_attribute_trackers(**params) else: return client.get_host_attribute_components(**params)
def get_attribute(self, **kwargs): client = AttributeRequest(self.username, self.apikey) keys = ['query', 'type'] params = self._cleanup_params(keys, **kwargs) if params.get('type') == 'tracker': return client.get_host_attribute_trackers(**params) else: return client.get_host_attribute_components(**params)
class AttributeTestCase(unittest.TestCase): """Test case for attribute methods.""" formats = ['json', 'xml', 'csv', 'table'] def setup_class(self): self.patcher = patch('passivetotal.api.Client._get', fake_request) self.patcher.start() self.client = AttributeRequest('--No-User--', '--No-Key--') def teardown_class(self): self.patcher.stop() def test_trackers(self): """Test getting tracker codes.""" payload = {'query': 'passivetotal.org'} response = self.client.get_host_attribute_trackers(**payload) assert ('results' in response) def test_process_trackers(self): """Test processing tracker data.""" payload = {'query': 'passivetotal.org'} response = self.client.get_host_attribute_trackers(**payload) wrapped = AttributeResponse(response) record = wrapped.get_records().pop(0) assert (record.hostname) == 'passivetotal.org' assert (record.lastSeen) == '2016-01-26 13:47:45' assert (record.attributeType) == 'GoogleAnalyticsAccountNumber' assert (record.firstSeen) == '2015-10-09 17:05:38' assert (record.attributeValue) == 'UA-61048133' def test_components(self): """Test getting component data.""" payload = {'query': 'passivetotal.org'} response = self.client.get_host_attribute_components(**payload) assert ('results' in response) def test_process_components(self): """Test processing component data.""" payload = {'query': 'passivetotal.org'} response = self.client.get_host_attribute_components(**payload) wrapped = AttributeResponse(response) record = wrapped.get_records().pop(0) assert (record.hostname) == 'passivetotal.org' assert (record.lastSeen) == '2016-01-07 21:52:30' assert (record.category) == 'JavaScript Library' assert (record.firstSeen) == '2015-12-26 11:17:43' assert (record.label) == 'jQuery' def test_trackers_search(self): """Test searching trakcer data.""" payload = {'query': 'UA-49901229', 'type': 'GoogleAnalyticsAccountNumber'} response = self.client.search_trackers(**payload) assert ('results' in response) def test_process_trackers_search(self): """Test processing component data.""" payload = {'query': 'UA-49901229', 'type': 'GoogleAnalyticsAccountNumber'} response = self.client.search_trackers(**payload) wrapped = AttributeResponse(response) record = wrapped.get_records().pop(0) assert not (record.everBlacklisted) assert (record.alexaRank) == 38 assert (record.hostname) == 'demo.paypal.com'
class AttributeTestCase(unittest.TestCase): """Test case for attribute methods.""" formats = ['json', 'xml', 'csv', 'table'] def setup_class(self): self.patcher = patch('passivetotal.api.Client._get', fake_request) self.patcher.start() self.client = AttributeRequest('--No-User--', '--No-Key--') def teardown_class(self): self.patcher.stop() def test_trackers(self): """Test getting tracker codes.""" payload = {'query': 'passivetotal.org'} response = self.client.get_host_attribute_trackers(**payload) assert ('results' in response) def test_process_trackers(self): """Test processing tracker data.""" payload = {'query': 'passivetotal.org'} response = self.client.get_host_attribute_trackers(**payload) wrapped = Response(response) record = wrapped.results.pop(0) record = Response(record) assert (record.hostname) == 'passivetotal.org' assert (record.lastSeen) == '2016-01-26 13:47:45' assert (record.attributeType) == 'GoogleAnalyticsAccountNumber' assert (record.firstSeen) == '2015-10-09 17:05:38' assert (record.attributeValue) == 'UA-61048133' def test_components(self): """Test getting component data.""" payload = {'query': 'passivetotal.org'} response = self.client.get_host_attribute_components(**payload) assert ('results' in response) def test_process_components(self): """Test processing component data.""" payload = {'query': 'passivetotal.org'} response = self.client.get_host_attribute_components(**payload) wrapped = Response(response) record = wrapped.results.pop(0) record = Response(record) assert (record.hostname) == 'passivetotal.org' assert (record.lastSeen) == '2016-01-07 21:52:30' assert (record.category) == 'JavaScript Library' assert (record.firstSeen) == '2015-12-26 11:17:43' assert (record.label) == 'jQuery' def test_trackers_search(self): """Test searching trakcer data.""" payload = {'query': 'UA-49901229', 'type': 'GoogleAnalyticsAccountNumber'} response = self.client.search_trackers(**payload) assert ('results' in response) def test_process_trackers_search(self): """Test processing component data.""" payload = {'query': 'UA-49901229', 'type': 'GoogleAnalyticsAccountNumber'} response = self.client.search_trackers(**payload) wrapped = Response(response) record = wrapped.results.pop(0) record = Response(record) assert not (record.everBlacklisted) assert (record.alexaRank) == 38 assert (record.hostname) == 'demo.paypal.com'