mirror of
https://github.com/PostHog/posthog.git
synced 2024-12-01 04:12:23 +01:00
90 lines
3.4 KiB
Python
90 lines
3.4 KiB
Python
from datetime import datetime
|
|
from uuid import uuid4
|
|
|
|
from freezegun.api import freeze_time
|
|
|
|
from ee.clickhouse.client import sync_execute
|
|
from ee.clickhouse.models.event import create_event
|
|
from ee.clickhouse.models.person import create_person_distinct_id
|
|
from posthog.models.person import Person
|
|
from posthog.models.team import Team
|
|
from posthog.models.utils import UUIDT
|
|
from posthog.tasks.status_report import status_report
|
|
from posthog.tasks.test.test_status_report import factory_status_report
|
|
|
|
|
|
def _create_event(distinct_id: str, event: str, lib: str, created_at: datetime, team: Team):
|
|
create_event(
|
|
event_uuid=uuid4(),
|
|
event=event,
|
|
distinct_id=distinct_id,
|
|
timestamp=created_at,
|
|
team=team,
|
|
properties={"$lib": lib},
|
|
)
|
|
|
|
|
|
def _create_person(distinct_id: str, team: Team) -> Person:
|
|
person = Person.objects.create(team=team, distinct_ids=[distinct_id])
|
|
return Person(id=person.uuid)
|
|
|
|
|
|
class TestStatusReport(factory_status_report(_create_event, _create_person)): # type: ignore
|
|
# CH only
|
|
def test_status_report_duplicate_distinct_ids(self) -> None:
|
|
create_person_distinct_id(self.team.id, "duplicate_id1", str(UUIDT()))
|
|
create_person_distinct_id(self.team.id, "duplicate_id1", str(UUIDT()))
|
|
create_person_distinct_id(self.team.id, "duplicate_id2", str(UUIDT()))
|
|
create_person_distinct_id(self.team.id, "duplicate_id2", str(UUIDT()))
|
|
create_person_distinct_id(self.team.id, "duplicate_id2", str(UUIDT()))
|
|
|
|
for index in range(0, 2):
|
|
sync_execute(
|
|
"INSERT INTO person_distinct_id SELECT %(distinct_id)s, %(person_id)s, %(team_id)s, 1, %(timestamp)s, 0 VALUES",
|
|
{
|
|
"distinct_id": "duplicate_id_old",
|
|
"person_id": str(UUIDT()),
|
|
"team_id": self.team.id,
|
|
"timestamp": "2020-01-01 12:01:0%s" % index,
|
|
},
|
|
)
|
|
|
|
report = status_report(dry_run=True).get("teams")[self.team.id] # type: ignore
|
|
|
|
duplicate_ids_report = report["duplicate_distinct_ids"]
|
|
|
|
expected_result = {
|
|
"prev_total_ids_with_duplicates": 1,
|
|
"prev_total_extra_distinct_id_rows": 1,
|
|
"new_total_ids_with_duplicates": 2,
|
|
"new_total_extra_distinct_id_rows": 4,
|
|
}
|
|
|
|
self.assertEqual(duplicate_ids_report, expected_result)
|
|
|
|
# CH only
|
|
def test_status_report_multiple_ids_per_person(self) -> None:
|
|
person_id1 = str(UUIDT())
|
|
person_id2 = str(UUIDT())
|
|
|
|
create_person_distinct_id(self.team.id, "id1", person_id1)
|
|
create_person_distinct_id(self.team.id, "id2", person_id1)
|
|
create_person_distinct_id(self.team.id, "id3", person_id1)
|
|
create_person_distinct_id(self.team.id, "id4", person_id1)
|
|
create_person_distinct_id(self.team.id, "id5", person_id1)
|
|
|
|
create_person_distinct_id(self.team.id, "id6", person_id2)
|
|
create_person_distinct_id(self.team.id, "id7", person_id2)
|
|
create_person_distinct_id(self.team.id, "id8", person_id2)
|
|
|
|
report = status_report(dry_run=True).get("teams")[self.team.id] # type: ignore
|
|
|
|
multiple_ids_report = report["multiple_ids_per_person"]
|
|
|
|
expected_result = {
|
|
"total_persons_with_more_than_2_ids": 2,
|
|
"max_distinct_ids_for_one_person": 5,
|
|
}
|
|
|
|
self.assertEqual(multiple_ids_report, expected_result)
|