#!/usr/bin/env python from sentry.runner import configure configure() from datetime import timedelta from uuid import uuid4 from django.conf import settings from django.db.models import F from django.utils import timezone from sentry.models.organization import Organization from sentry.models.project import Project from sentry.utils.samples import create_trace, generate_user, random_normal def main(slow=False): project_names = {"Ludic Science", "Earth", "Fire", "Wind", "Water", "Heart"} project_map = {} if settings.SENTRY_SINGLE_ORGANIZATION: org = Organization.get_default() print(f"Mocking org {org.name}") # NOQA else: print("Mocking org {}".format("Default")) # NOQA org, _ = Organization.objects.get_or_create(slug="default") for project_name in project_names: print(f" > Mocking project {project_name}") # NOQA project, _ = Project.objects.get_or_create( name=project_name, defaults={ "organization": org, "first_event": timezone.now(), "flags": Project.flags.has_releases, }, ) project_map[project_name] = project if not project.first_event: project.update(first_event=project.date_added) if not project.flags.has_releases: project.update(flags=F("flags").bitor(Project.flags.has_releases)) if not project.flags.has_transactions: project.update(flags=F("flags").bitor(Project.flags.has_transactions)) mobile_project = project_map["Ludic Science"] frontend_project = project_map["Fire"] backend_project = project_map["Earth"] service_projects = [ project_map["Wind"], project_map["Water"], project_map["Heart"], ] timestamp = timezone.now() print(f" > Loading normal trace") # NOQA # Normal trace create_trace( slow, timestamp - timedelta(milliseconds=random_normal(4000, 250, 1000)), timestamp, generate_user(), uuid4().hex, None, { "project": frontend_project, "transaction": "/plants/:plantId/", "frontend": True, "errors": 1, "children": [ { "project": backend_project, "transaction": "/api/plants/", "children": [ { "project": service_projects[0], "transaction": "/products/all/", "children": [], }, { "project": service_projects[1], "transaction": "/analytics/", "children": [], }, { "project": service_projects[2], "transaction": "tasks.create_invoice", "children": [], }, ], }, ], }, ) print(f" > Loading normal trace, but with performance issue") # NOQA # Normal trace create_trace( slow, timestamp - timedelta(milliseconds=random_normal(4000, 250, 1000)), timestamp, generate_user(), uuid4().hex, None, { "project": frontend_project, "transaction": "/plants/:plantId/", "frontend": True, "errors": 1, "children": [ { "project": backend_project, "transaction": "/api/plants/", "performance_issues": ["n+1"], "children": [ { "project": service_projects[0], "transaction": "/products/all/", "children": [], }, { "project": service_projects[1], "transaction": "/analytics/", "children": [], }, { "project": service_projects[2], "transaction": "tasks.create_invoice", "children": [], }, ], }, ], }, ) print(f" > Loading mobile trace") # NOQA # Normal trace create_trace( slow, timestamp - timedelta(milliseconds=random_normal(4000, 250, 1000)), timestamp, generate_user(), uuid4().hex, None, { "project": mobile_project, "transaction": "MainActivity", "mobile": True, "errors": 1, "performance_issues": ["db-main-thread"], "children": [ { "project": backend_project, "transaction": "/api/plants/", "children": [ { "project": service_projects[0], "transaction": "/products/all/", "children": [], }, { "project": service_projects[1], "transaction": "/analytics/", "children": [], }, { "project": service_projects[2], "transaction": "tasks.create_invoice", "children": [], }, ], }, ], }, ) print(f" > Loading orphan data") # NOQA # Trace only with orphans create_trace( slow, timestamp - timedelta(milliseconds=random_normal(4000, 250, 1000)), timestamp, generate_user(), uuid4().hex, uuid4().hex[:16], { "project": frontend_project, "transaction": "/orphans/:orphanId/", "frontend": True, "children": [ { "project": backend_project, "transaction": "/api/orphans/", "errors": 1, "children": [ { "project": service_projects[0], "transaction": "/orphans/all/", "errors": 1, "children": [], }, { "project": service_projects[1], "transaction": "/orphan/analytics/", "children": [], }, { "project": service_projects[2], "transaction": "tasks.invoice_orphans", "errors": 1, "children": [], }, ], }, ], }, ) print(f" > Loading trace with many siblings") # NOQA create_trace( slow, timestamp - timedelta(milliseconds=random_normal(4000, 250, 1000)), timestamp, generate_user(), uuid4().hex, None, { "project": frontend_project, "transaction": "/siblings/:count/", "frontend": True, "children": [ { "project": backend_project, "transaction": f"/api/sibling_{i}/", "children": [], } for i in range(15) ], }, ) print(f" > Loading trace with many roots") # NOQA trace_id = uuid4().hex for _ in range(15): create_trace( slow, timestamp - timedelta(milliseconds=random_normal(4000, 250, 1000)), timestamp, generate_user(), trace_id, None, { "project": frontend_project, "transaction": "/multiple-root/:root/", "frontend": True, "children": [ { "project": backend_project, "transaction": "/multiple-root/child/", "children": [], } ], }, ) print(f" > Loading chained trace with orphans") # NOQA trace_id = uuid4().hex create_trace( slow, timestamp - timedelta(milliseconds=random_normal(4000, 250, 1000)), timestamp, generate_user(), trace_id, None, { "project": frontend_project, "transaction": "/chained/:login/", "frontend": True, "children": [ { "project": backend_project, "transaction": "/api/auth/", "children": [ { "project": service_projects[0], "transaction": "/auth/check-login/", "errors": 1, "children": [ { "project": service_projects[1], "transaction": "/analytics/", "errors": 1, "children": [ { "project": service_projects[2], "transaction": "tasks.check_login", "errors": 1, "children": [], } ], } ], }, ], }, ], }, ) create_trace( slow, timestamp - timedelta(milliseconds=random_normal(4000, 250, 1000)), timestamp, generate_user(), trace_id, uuid4().hex[:16], { "project": frontend_project, "transaction": "/orphans/:orphanId/", "frontend": True, "children": [ { "project": backend_project, "transaction": "/api/orphans/", "errors": 1, "children": [], } ], }, ) print(f" > Loading traces missing instrumentation") # NOQA create_trace( slow, timestamp - timedelta(milliseconds=random_normal(4000, 250, 1000)), timestamp, generate_user(), uuid4().hex, None, { "project": frontend_project, "transaction": "/missing/:frontend/", "frontend": True, "children": [], }, ) create_trace( slow, timestamp - timedelta(milliseconds=random_normal(4000, 250, 1000)), timestamp, generate_user(), uuid4().hex, None, { "project": backend_project, "transaction": "/missing/backend", "children": [], }, ) if __name__ == "__main__": settings.CELERY_ALWAYS_EAGER = True from optparse import OptionParser parser = OptionParser() parser.add_option( "--slow", default=False, action="store_true", help="sleep between each transaction to let clickhouse rest", ) (options, args) = parser.parse_args() try: main( slow=options.slow, ) except Exception: # Avoid reporting any issues recursively back into Sentry import sys import traceback traceback.print_exc() sys.exit(1)