sentinel/tests/sentry/models/test_releaseenvironment.py

from datetime import timedelta

from django.utils import timezone

from sentry.models.environment import Environment
from sentry.models.release import Release
from sentry.models.releaseenvironment import ReleaseEnvironment
from sentry.testutils.cases import TestCase
from sentry.testutils.silo import region_silo_test


@region_silo_test
class GetOrCreateTest(TestCase):
    def test_simple(self):
        project = self.create_project(name="foo")
        datetime = timezone.now()

        release = Release.objects.create(organization_id=project.organization_id, version="abcdef")
        release.add_project(project)
        env = Environment.objects.create(organization_id=project.organization_id, name="prod")
        relenv = ReleaseEnvironment.get_or_create(
            project=project, release=release, environment=env, datetime=datetime
        )

        assert relenv.organization_id == project.organization_id
        assert relenv.release_id == release.id
        assert relenv.environment_id == env.id

        datetime_new = datetime + timedelta(days=1)

        relenv = ReleaseEnvironment.get_or_create(
            project=project, release=release, environment=env, datetime=datetime_new
        )

        assert relenv.first_seen == datetime
        assert relenv.last_seen == datetime_new

        datetime_new2 = datetime_new + timedelta(seconds=1)

        # this should not update immediately as the window is too close
        relenv = ReleaseEnvironment.get_or_create(
            project=project, release=release, environment=env, datetime=datetime_new2
        )

        assert relenv.first_seen == datetime
        assert relenv.last_seen == datetime_new

        # shouldn't create new release env if same env, release and org
        project2 = self.create_project(name="bar", organization=project.organization)
        release.add_project(project2)

        relenv2 = ReleaseEnvironment.get_or_create(
            project=project2, release=release, environment=env, datetime=datetime
        )
        assert relenv.id == relenv2.id
        assert ReleaseEnvironment.objects.get(id=relenv.id).last_seen == relenv2.last_seen