def test_more_from_relationship_summary_returns_error(self, mock_get): """Load more relationship data from summary""" mock_json = make_mock_response_json({'data': {'error': 'error'}}) mock_get.return_value = mock_json rs = Relationship('past_team', json.loads(PAST_TEAM_RELATIONSHIP)) rs = self.cb.more(rs) self.assertIsNone(rs) mock_get.assert_called_with( 'https://api.crunchbase.com/v3.1/organizations/example/past_team' '?user_key=123')
def test_more_relationship_for_relationship_summary(self, mock_get): """Load more relationship data from summary""" mock_json = make_mock_response_json( {'data': json.loads(MOCK_RELATIONSHIP_PAGE)}) mock_get.return_value = mock_json rs = Relationship('past_team', json.loads(PAST_TEAM_RELATIONSHIP)) rs = self.cb.more(rs) self.assertIsNotNone(rs) self.assertEquals(3, len(rs)) mock_get.assert_called_with( 'https://api.crunchbase.com/v3.1/organizations/example/past_team' '?user_key=123')
def test_more_relationship_for_relationship_page2(self, mock_get): """At page 1, there a next_page, get it and return the Relationship""" mock_json = make_mock_response_json({ "data": { "paging": { "items_per_page": 8, "current_page": 2, "number_of_pages": 2, "next_page_url": None, "prev_page_url": None, "total_items": 10, "sort_order": "custom" }, "items": [{}, {}] } }) mock_get.return_value = mock_json data = { "paging": { "items_per_page": 8, "current_page": 1, "number_of_pages": 2, "next_page_url": "https://api.crunchbase.com/v/2/" "organization/example/past_team?page=2", "prev_page_url": None, "total_items": 10, "sort_order": "custom" }, "items": [{}, {}, {}, {}, {}, {}, {}, {}] } rs = Relationship('past_team', data) more_rs = self.cb.more(rs) self.assertEqual(2, len(more_rs)) mock_get.assert_called_with( 'https://api.crunchbase.com/v/2/organization/example/past_team' '?page=2&user_key=123')
def test_relationship(self): past_team = Relationship('past_team', PAST_TEAM_RELATIONSHIP) one = past_team.get(0) self.assertEqual(one.first_name, "First") self.assertEqual(one.permalink, "first-last") self.assertEqual('past_team', past_team.name)
def test_one_to_many_relationship(self): past_team = Relationship('past_team', PAST_TEAM_RELATIONSHIP) job = past_team.get(0) self.assertEqual(job.title, 'Co-Founder / President') self.assertEqual(job.person.first_name, 'Tom') self.assertEqual('past_team', past_team.name)
def test_one_to_many_relationship(self): past_team = Relationship("past_team", PAST_TEAM_RELATIONSHIP) one = past_team.get(0) self.assertEqual(one.first_name, "First") self.assertEqual("past_team", past_team.name)
def test_nested_relationships(self): r = Relationship('investments', T_R) self.assertEqual(r[0].investors[0].name, 'Investor')
def test_one_to_one_relationship(self): hq = Relationship('headquarters', HEADQUARTERS_RELATIONSHIP) self.assertEqual(hq.name, "Headquarters")
def test_one_to_many_relationship(self): past_team = Relationship('past_team', PAST_TEAM_RELATIONSHIP) one = past_team.get(0) self.assertEqual(one.first_name, "First") self.assertEqual('past_team', past_team.name)