def test_save_stats_to_file(mock_ContentFile, mock_storage):
mock_ContentFile.return_value = mock.sentinel.content
theme_update_count = ThemeUpdateCount(
addon_id=321, date='2016-01-18', count=123)
save_stats_to_file(theme_update_count)
mock_storage.assert_called_once_with(
'321/2016/01/2016_01_18_themeupdatecount.json', mock.sentinel.content)
def test_get_unlisted_addon(self):
self.user = UserProfile.objects.create(
read_dev_agreement=datetime.datetime.now())
self.api_key = self.create_api_key(self.user, 'bar')
AddonUser.objects.create(user=self.user, addon=self.addon)
save_stats_to_file(self.theme_update_count)
self.make_addon_unlisted(self.addon)
response = self.get(
reverse('stats.archive', kwargs={
'slug': 'a3615', 'year': '2016', 'month': '01',
'day': '18', 'model_name': 'themeupdatecount'}))
assert response.status_code == 404
def test_list_unlisted_addon(self):
self.user = UserProfile.objects.create(
read_dev_agreement=datetime.datetime.now())
self.api_key = self.create_api_key(self.user, 'bar')
AddonUser.objects.create(user=self.user, addon=self.addon)
save_stats_to_file(self.theme_update_count)
self.make_addon_unlisted(self.addon)
response = self.get(
reverse('stats.archive_list', kwargs={
'slug': 'a3615', 'year': '2016', 'month': '01'}))
# There are stats for this add-on for the requested timeframe, but it
# has no listed versions to it should return a 404.
assert response.status_code == 404
def test_get(self):
self.user = UserProfile.objects.create(
read_dev_agreement=datetime.datetime.now())
self.api_key = self.create_api_key(self.user, 'bar')
AddonUser.objects.create(user=self.user, addon=self.addon)
save_stats_to_file(self.theme_update_count)
response = self.get(
reverse('stats.archive', kwargs={
'slug': 'a3615', 'year': '2016', 'month': '01',
'day': '18', 'model_name': 'themeupdatecount'}))
assert response.status_code == 200
assert json.loads(response.content) == {
'count': 123,
'date': '2016-01-18',
'addon': 3615
}
def test_get(self):
self.user = UserProfile.objects.create(
read_dev_agreement=datetime.datetime.now())
self.api_key = self.create_api_key(self.user, 'bar')
AddonUser.objects.create(user=self.user, addon=self.addon)
save_stats_to_file(self.theme_update_count)
response = self.get(
reverse('stats.archive', kwargs={
'slug': 'a3615', 'year': '2016', 'month': '01',
'day': '18', 'model_name': 'themeupdatecount'}))
assert response.status_code == 200
assert json.loads(response.content) == {
'count': 123,
'date': '2016-01-18',
'addon': 3615
}
