|
1 | 1 | from unittest import mock
|
2 | 2 |
|
3 | 3 | import pytest
|
| 4 | +from django.contrib.auth.models import User |
4 | 5 | from django.contrib.sessions.backends.base import SessionBase
|
5 | 6 | from django.http import HttpResponse
|
6 | 7 | from django.test import TestCase, override_settings
|
|
12 | 13 | from readthedocs.builds.constants import BRANCH, EXTERNAL, LATEST, TAG
|
13 | 14 | from readthedocs.builds.models import Version
|
14 | 15 | from readthedocs.core.middleware import ReadTheDocsSessionMiddleware
|
| 16 | +from readthedocs.organizations.models import Organization |
15 | 17 | from readthedocs.projects.constants import GITHUB_BRAND, GITLAB_BRAND, PRIVATE, PUBLIC
|
16 | 18 | from readthedocs.projects.models import Project
|
17 | 19 | from readthedocs.subscriptions.constants import TYPE_CNAME
|
@@ -335,7 +337,11 @@ def setUp(self):
|
335 | 337 | self.pip = Project.objects.get(slug="pip")
|
336 | 338 | self.pip.versions.update(built=True)
|
337 | 339 | self.pip.show_version_warning = True
|
| 340 | + self.pip.privacy_level = PUBLIC |
338 | 341 | self.pip.save()
|
| 342 | + self.pip.versions.update(privacy_level=PUBLIC) |
| 343 | + |
| 344 | + self.user = User.objects.get(username="eric") |
339 | 345 |
|
340 | 346 | def test_highest_version_from_stable(self):
|
341 | 347 | base_version = self.pip.get_stable_version()
|
@@ -446,14 +452,17 @@ def test_highest_version_without_tags(self):
|
446 | 452 | returned_data = get_version_compare_data(self.pip, base_version)
|
447 | 453 | self.assertDictEqual(valid_data, returned_data)
|
448 | 454 |
|
| 455 | + @override_settings( |
| 456 | + RTD_ALLOW_ORGANIZATIONS=True, |
| 457 | + ) |
449 | 458 | def test_private_highest_version(self):
|
| 459 | + get(Organization, projects=[self.pip], owners=[self.user]) |
450 | 460 | self.pip.versions.update(privacy_level=PRIVATE)
|
451 |
| - owner = self.pip.users.first() |
452 | 461 | base_version = self.pip.versions.get(slug="0.8")
|
453 | 462 | returned_data = get_version_compare_data(self.pip, base_version)
|
454 | 463 | self.assertTrue(returned_data["is_highest"])
|
455 | 464 |
|
456 |
| - returned_data = get_version_compare_data(self.pip, base_version, user=owner) |
| 465 | + returned_data = get_version_compare_data(self.pip, base_version, user=self.user) |
457 | 466 | valid_data = {
|
458 | 467 | "project": "Version 0.8.1 of Pip (19)",
|
459 | 468 | "url": "https://pip.readthedocs.io/en/0.8.1/",
|
|
0 commit comments