151bd13666
* Generate common/djangoapps import shims for LMS * Generate common/djangoapps import shims for Studio * Stop appending project root to sys.path * Stop appending common/djangoapps to sys.path * Import from common.djangoapps.course_action_state instead of course_action_state * Import from common.djangoapps.course_modes instead of course_modes * Import from common.djangoapps.database_fixups instead of database_fixups * Import from common.djangoapps.edxmako instead of edxmako * Import from common.djangoapps.entitlements instead of entitlements * Import from common.djangoapps.pipline_mako instead of pipeline_mako * Import from common.djangoapps.static_replace instead of static_replace * Import from common.djangoapps.student instead of student * Import from common.djangoapps.terrain instead of terrain * Import from common.djangoapps.third_party_auth instead of third_party_auth * Import from common.djangoapps.track instead of track * Import from common.djangoapps.util instead of util * Import from common.djangoapps.xblock_django instead of xblock_django * Add empty common/djangoapps/__init__.py to fix pytest collection * Fix pylint formatting violations * Exclude import_shims/ directory tree from linting
80 lines
2.3 KiB
Python
80 lines
2.3 KiB
Python
"""Helpers for tests related to emitting events to the tracking logs."""
|
|
|
|
|
|
from datetime import datetime
|
|
|
|
from django.test import TestCase
|
|
from django.test.utils import override_settings
|
|
from eventtracking import tracker
|
|
from eventtracking.django import DjangoTracker
|
|
from freezegun import freeze_time
|
|
from pytz import UTC
|
|
|
|
FROZEN_TIME = datetime(2013, 10, 3, 8, 24, 55, tzinfo=UTC)
|
|
IN_MEMORY_BACKEND_CONFIG = {
|
|
'mem': {
|
|
'ENGINE': 'common.djangoapps.track.tests.InMemoryBackend'
|
|
}
|
|
}
|
|
|
|
|
|
class InMemoryBackend(object):
|
|
"""A backend that simply stores all events in memory"""
|
|
|
|
def __init__(self):
|
|
super(InMemoryBackend, self).__init__()
|
|
self.events = []
|
|
|
|
def send(self, event):
|
|
"""Store the event in a list"""
|
|
self.events.append(event)
|
|
|
|
|
|
@override_settings(
|
|
EVENT_TRACKING_BACKENDS=IN_MEMORY_BACKEND_CONFIG
|
|
)
|
|
class EventTrackingTestCase(TestCase):
|
|
"""
|
|
Supports capturing of emitted events in memory and inspecting them.
|
|
|
|
Each test gets a "clean slate" and can retrieve any events emitted during their execution.
|
|
|
|
"""
|
|
|
|
# Make this more robust to the addition of new events that the test doesn't care about.
|
|
|
|
def setUp(self):
|
|
freezer = freeze_time(FROZEN_TIME)
|
|
freezer.start()
|
|
self.addCleanup(freezer.stop)
|
|
|
|
super(EventTrackingTestCase, self).setUp()
|
|
|
|
self.recreate_tracker()
|
|
|
|
def recreate_tracker(self):
|
|
"""
|
|
Re-initialize the tracking system using updated django settings.
|
|
|
|
Use this if you make use of the @override_settings decorator to customize the tracker configuration.
|
|
"""
|
|
self.tracker = DjangoTracker()
|
|
tracker.register_tracker(self.tracker)
|
|
|
|
@property
|
|
def backend(self):
|
|
"""A reference to the in-memory backend that stores the events."""
|
|
return self.tracker.backends['mem']
|
|
|
|
def get_event(self, idx=0):
|
|
"""Retrieve an event emitted up to this point in the test."""
|
|
return self.backend.events[idx]
|
|
|
|
def assert_no_events_emitted(self):
|
|
"""Ensure no events were emitted at this point in the test."""
|
|
self.assertEqual(len(self.backend.events), 0)
|
|
|
|
def assert_events_emitted(self):
|
|
"""Ensure at least one event has been emitted at this point in the test."""
|
|
self.assertGreaterEqual(len(self.backend.events), 1)
|