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)