Files
bugsink/issues/factories.py
2024-04-08 11:41:15 +02:00

58 lines
1.6 KiB
Python

import uuid
from django.utils import timezone
from projects.models import Project
from .models import Issue, Grouping
from .utils import get_issue_grouper_for_data
def get_or_create_issue(project=None, event_data=None):
"""create issue for testing purposes (code basically stolen from ingest/views.py)"""
if event_data is None:
event_data = create_event_data()
if project is None:
project = Project.objects.create(name="Test project")
grouping_key = get_issue_grouper_for_data(event_data)
if not Grouping.objects.filter(project=project, grouping_key=grouping_key).exists():
issue = Issue.objects.create(
project=project,
**denormalized_issue_fields(),
)
issue_created = True
grouping = Grouping.objects.create(
project=project,
grouping_key=grouping_key,
issue=issue,
)
else:
grouping = Grouping.objects.get(project=project, grouping_key=grouping_key)
issue = grouping.issue
issue_created = False
return issue, issue_created
def create_event_data():
"""create minimal event data that is valid as per from_json()"""
return {
"event_id": uuid.uuid4().hex,
"timestamp": timezone.now().isoformat(),
"platform": "python",
}
def denormalized_issue_fields():
"""return a dict of fields that are expected to be denormalized on Issue; for testing purposes"""
return {
"first_seen": timezone.now(),
"last_seen": timezone.now(),
"event_count": 1,
}