def test_execute(self, mock_client):
mock_response = {
'count': '1',
'limit': '1',
'offset': '1',
'results': []
}
mock_client.post.return_value = mock_response
report = StatsReport({
'applicationIds': [{}],
'offset': 1,
'pageSize': 1
})
search = Search(report)
search_result = search.execute()
mock_client.post.assert_called_once_with(report, 'search/stats')
self.assertIsInstance(search_result, SearchResult)
mock_client.post.return_value = (r for r in [mock_response])
search.next_page()
self.assertEqual(len(mock_client.post.mock_calls), 2)
def test_find_all(self, mock_client):
StatsReport.find_all()
mock_client.get.assert_called_with('reports/all')
StatsReport.find_all('123')
mock_client.get.assert_called_with('reports/all?appId=123')
def test_init(self):
stats_report = StatsReport(MOCK_STATS_REPORT)
for key, value in MOCK_STATS_REPORT.items():
self.assertEqual(getattr(stats_report, key), value)
def test_new_for(self):
stats_report = StatsReport.new_for('123', '456', 'Test Report')
self.assertIsInstance(stats_report, StatsReport)
def test_new_for(self, mock_user):
mock_user.find.return_value = MOCK_USER
stats_report = StatsReport.new_for('123', '456', 'Test Report')
self.assertIsInstance(stats_report, StatsReport)
