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_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)
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)
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)
