mirror of
https://github.com/PostHog/posthog.git
synced 2024-11-24 18:07:17 +01:00
5d5ede19e4
* Fix REST hooks being only fired if project has webhook
* Fix firing more
* Actually still filter but also for Zapier
* Don't .only() Team in EE process_event to optimize REST hook firing
* Fix REST hook duplication in EE
* Update test_process_event.py
* Debug
* Revert "Debug"
This reverts commit 6d4c9f754d.
* Fix organization-less teams partly
* Fix more organization-less teams
* Fix moar organization-less teams
* Explicitly bootstrap org without user in tests
* Fix org bootstrap
* Fix typing
* Fix bootstrap usage
* Update test_process_event.py
* Fix num queries
* Add test_action_on_perform_hook_fired_once
* Fix Action.objects.create call
* Don't test hook firing as it's offloaded to worker
82 lines
4.1 KiB
Python
82 lines
4.1 KiB
Python
from typing import cast
|
|
from unittest.mock import patch
|
|
|
|
from django.test.utils import tag
|
|
from rest_framework import status
|
|
|
|
from ee.api.test.base import APILicensedTest
|
|
from posthog.models.organization import Organization, OrganizationMembership
|
|
from posthog.models.team import Team
|
|
|
|
|
|
class TestOrganizationEnterpriseAPI(APILicensedTest):
|
|
def test_create_organization(self):
|
|
response = self.client.post("/api/organizations/", {"name": "Test"})
|
|
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
|
|
self.assertEqual(Organization.objects.count(), 2)
|
|
response_data = response.json()
|
|
self.assertEqual(response_data.get("name"), "Test")
|
|
self.assertEqual(OrganizationMembership.objects.filter(organization_id=response_data.get("id")).count(), 1)
|
|
self.assertEqual(
|
|
OrganizationMembership.objects.get(organization_id=response_data.get("id"), user=self.user).level,
|
|
OrganizationMembership.Level.OWNER,
|
|
)
|
|
|
|
def test_delete_second_managed_organization(self):
|
|
organization, _, team = Organization.objects.bootstrap(self.user, name="X")
|
|
self.assertTrue(Organization.objects.filter(id=organization.id).exists())
|
|
self.assertTrue(Team.objects.filter(id=team.id).exists())
|
|
response = self.client.delete(f"/api/organizations/{organization.id}")
|
|
self.assertEqual(response.status_code, 204)
|
|
self.assertFalse(Organization.objects.filter(id=organization.id).exists())
|
|
self.assertFalse(Team.objects.filter(id=team.id).exists())
|
|
|
|
def test_no_delete_last_organization(self):
|
|
org_id = self.organization.id
|
|
self.assertTrue(Organization.objects.filter(id=org_id).exists())
|
|
response = self.client.delete(f"/api/organizations/{org_id}")
|
|
self.assertEqual(
|
|
response.data,
|
|
{
|
|
"attr": None,
|
|
"detail": f"Cannot remove organization since that would leave member {self.CONFIG_USER_EMAIL} organization-less, which is not supported yet.",
|
|
"code": "invalid_input",
|
|
"type": "validation_error",
|
|
},
|
|
)
|
|
self.assertEqual(response.status_code, 400)
|
|
self.assertTrue(Organization.objects.filter(id=org_id).exists())
|
|
|
|
def test_no_delete_organization_not_administrating(self):
|
|
organization, organization_membership, team = Organization.objects.bootstrap(self.user)
|
|
organization_membership = cast(OrganizationMembership, organization_membership)
|
|
organization_membership.level = OrganizationMembership.Level.MEMBER
|
|
organization_membership.save()
|
|
self.assertTrue(Organization.objects.filter(id=organization.id).exists())
|
|
self.assertTrue(Team.objects.filter(id=team.id).exists())
|
|
response = self.client.delete(f"/api/organizations/{organization.id}")
|
|
self.assertEqual(response.status_code, 403)
|
|
self.assertTrue(Organization.objects.filter(id=organization.id).exists())
|
|
self.assertTrue(Team.objects.filter(id=team.id).exists())
|
|
|
|
def test_no_delete_organization_not_belonging_to(self):
|
|
# as member only
|
|
self.organization_membership.level = OrganizationMembership.Level.ADMIN
|
|
self.organization_membership.save()
|
|
organization = Organization.objects.create(name="Some Other Org")
|
|
response_1 = self.client.delete(f"/api/organizations/{organization.id}")
|
|
self.assertEqual(
|
|
response_1.data, {"attr": None, "detail": "Not found.", "code": "not_found", "type": "invalid_request"}
|
|
)
|
|
self.assertEqual(response_1.status_code, 404)
|
|
self.assertTrue(Organization.objects.filter(id=organization.id).exists())
|
|
# as admin
|
|
self.organization_membership.level = OrganizationMembership.Level.MEMBER
|
|
self.organization_membership.save()
|
|
response_2 = self.client.delete(f"/api/organizations/{organization.id}")
|
|
self.assertEqual(
|
|
response_2.data, {"attr": None, "detail": "Not found.", "code": "not_found", "type": "invalid_request"}
|
|
)
|
|
self.assertEqual(response_2.status_code, 404)
|
|
self.assertTrue(Organization.objects.filter(id=organization.id).exists())
|