class CategorySearchTests(TestCase): """Tests for the client's search_category() and get_category() methods""" def setUp(self): self.client = OctopartClient(api_key='TEST_TOKEN') def test_get_category(self): response_body = { "__class__": "Category", "ancestor_names": ["Electronic Parts", "Passive Components"], "ancestor_uids": ["8a1e4714bb3951d9", "7542b8484461ae85"], "children_uids": [ "db8a1680eaa4004d", "eb282622da0730ef", "e2992f93d3b3395e", "e7ca2ac0de173c0d", "f6473b4dcf9d2d80", "1db9d0b32462c67e", "e0b52dbfd96d3b72", "d820152ae1f903e7", "f0433d49c9e52b84" ], "name": "Capacitors", "num_parts": 1395137, "parent_uid": "7542b8484461ae85", "uid": "f8883582c9a8234f" } with octopart_mock_response(response_body) as response: dict_ = self.client.get_category('f8883582c9a8234f') assert dict_['name'] == 'Capacitors' called_url = request_url_from_request_mock(response) assert '/categories/f8883582c9a8234f' in called_url def test_search_category(self): response_body = { "__class__": "SearchResponse", "results": [{ "__class__": "SearchResult", "item": { "__class__": "Category", "ancestor_names": ["Electronic Parts", "Passive Components"], "ancestor_uids": ["8a1e4714bb3951d9", "7542b8484461ae85"], "children_uids": [ "db8a1680eaa4004d", "eb282622da0730ef", "e2992f93d3b3395e", "e7ca2ac0de173c0d", "f6473b4dcf9d2d80", "1db9d0b32462c67e", "e0b52dbfd96d3b72", "d820152ae1f903e7", "f0433d49c9e52b84" ], "name": "Capacitors", "num_parts": 1395137, "parent_uid": "7542b8484461ae85", "uid": "f8883582c9a8234f" } }] } with octopart_mock_response(response_body) as response: dict_ = self.client.search_category('Capacitors') [result] = dict_['results'] assert result['item']['name'] == 'Capacitors' called_url = request_url_from_request_mock(response) assert '/categories/search' in called_url assert 'q=Capacitors' in called_url
def get_category(uid: str) -> models.Category: client = OctopartClient() cat_dict = client.get_category(uid) return models.Category(cat_dict, strict=False)