mirror of
https://github.com/hanzoai/sentinel.git
synced 2026-09-29 22:34:33 +00:00
36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
from sentry.analytics import Analytics
|
|
from sentry.testutils.cases import TestCase
|
|
|
|
|
|
class DummyAnalytics(Analytics):
|
|
def __init__(self):
|
|
self.events = []
|
|
super().__init__()
|
|
|
|
def record_event(self, event):
|
|
self.events.append(event)
|
|
|
|
|
|
class AnalyticsTest(TestCase):
|
|
def test_record(self):
|
|
organization = self.create_organization()
|
|
provider = DummyAnalytics()
|
|
provider.record("organization.created", organization)
|
|
assert len(provider.events) == 1
|
|
event = provider.events.pop(0)
|
|
assert event.type == "organization.created"
|
|
assert event.datetime
|
|
assert event.data["slug"] == organization.slug
|
|
assert not event.data["actor_id"]
|
|
|
|
def test_record_with_attrs(self):
|
|
organization = self.create_organization()
|
|
provider = DummyAnalytics()
|
|
provider.record("organization.created", organization, actor_id=1)
|
|
assert len(provider.events) == 1
|
|
event = provider.events.pop(0)
|
|
assert event.type == "organization.created"
|
|
assert event.datetime
|
|
assert event.data["slug"] == organization.slug
|
|
assert event.data["actor_id"] == "1"
|