Request organization account

tests/common/db/organizations.py

@@ -0,0 +1,97 @@
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import datetime
+
+import factory
+import faker
+
+from warehouse.organizations.models import (
+    Organization,
+    OrganizationInvitation,
+    OrganizationNameCatalog,
+    OrganizationProject,
+    OrganizationRole,
+)
+
+from .accounts import UserFactory
+from .base import WarehouseFactory
+from .packaging import ProjectFactory
+
+fake = faker.Faker()
+
+
+class OrganizationFactory(WarehouseFactory):
+    class Meta:
+        model = Organization
+
+    id = factory.Faker("uuid4", cast_to=None)
+    name = factory.Faker("word")
+    normalized_name = factory.Faker("word")
+    display_name = factory.Faker("word")
+    orgtype = "Community"
+    link_url = factory.Faker("uri")
+    description = factory.Faker("sentence")
+    is_active = True
+    is_approved = False
+    created = factory.Faker(
+        "date_time_between_dates",
+        datetime_start=datetime.datetime(2020, 1, 1),
+        datetime_end=datetime.datetime(2022, 1, 1),
+    )
+    date_approved = factory.Faker(
+        "date_time_between_dates", datetime_start=datetime.datetime(2020, 1, 1)
+    )
+
+
+class OrganizationEventFactory(WarehouseFactory):
+    class Meta:
+        model = Organization.Event
+
+    source = factory.SubFactory(OrganizationFactory)
+
+
+class OrganizationNameCatalogFactory(WarehouseFactory):
+    class Meta:
+        model = OrganizationNameCatalog
+
+    name = factory.Faker("orgname")
+    organization_id = factory.Faker("uuid4", cast_to=None)
+
+
+class OrganizationRoleFactory(WarehouseFactory):
+    class Meta:
+        model = OrganizationRole
+
+    role_name = "Owner"
+    user = factory.SubFactory(UserFactory)
+    organization = factory.SubFactory(OrganizationFactory)
+
+
+class OrganizationInvitationFactory(WarehouseFactory):
+    class Meta:
+        model = OrganizationInvitation
+
+    invite_status = "pending"
+    token = "test_token"
+    user = factory.SubFactory(UserFactory)
+    organization = factory.SubFactory(OrganizationFactory)
+
+
+class OrganizationProjectFactory(WarehouseFactory):
+    class Meta:
+        model = OrganizationProject
+
+    id = factory.Faker("uuid4", cast_to=None)
+    is_active = True
+    organization = factory.SubFactory(OrganizationFactory)
+    project = factory.SubFactory(ProjectFactory)

tests/conftest.py

@@ -45,6 +45,7 @@
 from warehouse.email.interfaces import IEmailSender
 from warehouse.macaroons import services as macaroon_services
 from warehouse.metrics import IMetricsService
+from warehouse.organizations import services as organization_services
 
 from .common.db import Session
 
@@ -285,6 +286,13 @@ def macaroon_service(db_session):
     return macaroon_services.DatabaseMacaroonService(db_session)
 
 
+@pytest.fixture
+def organization_service(db_session, remote_addr):
+    return organization_services.DatabaseOrganizationService(
+        db_session, remote_addr=remote_addr
+    )
+
+
 @pytest.fixture
 def token_service(app_config):
     return account_services.TokenService(secret="secret", salt="salt", max_age=21600)

tests/functional/manage/test_views.py

@@ -15,26 +15,91 @@
 from webob.multidict import MultiDict
 
 from warehouse.accounts.interfaces import IPasswordBreachedService, IUserService
+from warehouse.admin.flags import AdminFlagValue
 from warehouse.manage import views
+from warehouse.organizations.interfaces import IOrganizationService
+from warehouse.organizations.models import OrganizationType
 
 from ...common.db.accounts import EmailFactory, UserFactory
 
 
 class TestManageAccount:
     def test_save_account(self, pyramid_services, user_service, db_request):
         breach_service = pretend.stub()
+        organization_service = pretend.stub()
         pyramid_services.register_service(user_service, IUserService, None)
         pyramid_services.register_service(
             breach_service, IPasswordBreachedService, None
         )
+        pyramid_services.register_service(
+            organization_service, IOrganizationService, None
+        )
         user = UserFactory.create(name="old name")
         EmailFactory.create(primary=True, verified=True, public=True, user=user)
         db_request.user = user
         db_request.method = "POST"
         db_request.path = "/manage/accounts/"
         db_request.POST = MultiDict({"name": "new name", "public_email": ""})
-        views.ManageAccountViews(db_request).save_account()
 
+        views.ManageAccountViews(db_request).save_account()
         user = user_service.get_user(user.id)
+
         assert user.name == "new name"
         assert user.public_email is None
+
+
+class TestManageOrganizations:
+    def test_create_organization(
+        self,
+        pyramid_services,
+        user_service,
+        organization_service,
+        db_request,
+        monkeypatch,
+    ):
+        pyramid_services.register_service(user_service, IUserService, None)
+        pyramid_services.register_service(
+            organization_service, IOrganizationService, None
+        )
+        user = UserFactory.create(name="old name")
+        EmailFactory.create(primary=True, verified=True, public=True, user=user)
+        db_request.user = user
+        db_request.method = "POST"
+        db_request.path = "/manage/organizations/"
+        db_request.POST = MultiDict(
+            {
+                "name": "psf",
+                "display_name": "Python Software Foundation",
+                "orgtype": "Community",
+                "link_url": "https://www.python.org/psf/",
+                "description": (
+                    "To promote, protect, and advance the Python programming "
+                    "language, and to support and facilitate the growth of a "
+                    "diverse and international community of Python programmers"
+                ),
+            }
+        )
+        monkeypatch.setattr(
+            db_request,
+            "flags",
+            pretend.stub(enabled=pretend.call_recorder(lambda *a: False)),
+        )
+        send_email = pretend.call_recorder(lambda *a, **kw: None)
+        monkeypatch.setattr(
+            views, "send_admin_new_organization_requested_email", send_email
+        )
+        monkeypatch.setattr(views, "send_new_organization_requested_email", send_email)
+
+        views.ManageOrganizationsViews(db_request).create_organization()
+        organization = organization_service.get_organization_by_name(
+            db_request.POST["name"]
+        )
+
+        assert db_request.flags.enabled.calls == [
+            pretend.call(AdminFlagValue.DISABLE_ORGANIZATIONS),
+        ]
+        assert organization.name == db_request.POST["name"]
+        assert organization.display_name == db_request.POST["display_name"]
+        assert organization.orgtype == OrganizationType[db_request.POST["orgtype"]]
+        assert organization.link_url == db_request.POST["link_url"]
+        assert organization.description == db_request.POST["description"]

tests/unit/accounts/test_services.py

@@ -408,6 +408,14 @@ def test_get_user_by_email_failure(self, user_service):
 
         assert found_user is None
 
+    def test_get_admins(self, user_service):
+        admin = UserFactory.create(is_superuser=True)
+        user = UserFactory.create(is_superuser=False)
+        admins = user_service.get_admins()
+
+        assert admin in admins
+        assert user not in admins
+
     def test_disable_password(self, user_service):
         user = UserFactory.create()
 

tests/unit/admin/test_routes.py

@@ -26,6 +26,11 @@ def test_includeme():
 
     assert config.add_route.calls == [
         pretend.call("admin.dashboard", "/admin/", domain=warehouse),
+        pretend.call(
+            "admin.organization.approve",
+            "/admin/organizations/approve/",
+            domain=warehouse,
+        ),
         pretend.call("admin.user.list", "/admin/users/", domain=warehouse),
         pretend.call("admin.user.detail", "/admin/users/{user_id}/", domain=warehouse),
         pretend.call(

tests/unit/admin/views/test_organizations.py

@@ -0,0 +1,20 @@
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import pretend
+
+from warehouse.admin.views import organizations as views
+
+
+class TestOrganizations:
+    def test_approve(self):
+        assert views.approve(pretend.stub()) == {}