39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
# lint-amnesty, pylint: disable=missing-module-docstring
|
|
|
|
from django.test import TestCase
|
|
from mock import patch
|
|
|
|
from common.djangoapps.track.backends.mongodb import MongoBackend
|
|
|
|
|
|
class TestMongoBackend(TestCase): # lint-amnesty, pylint: disable=missing-class-docstring
|
|
def setUp(self):
|
|
super(TestMongoBackend, self).setUp() # lint-amnesty, pylint: disable=super-with-arguments
|
|
self.mongo_patcher = patch('common.djangoapps.track.backends.mongodb.MongoClient')
|
|
self.mongo_patcher.start()
|
|
self.addCleanup(self.mongo_patcher.stop)
|
|
|
|
self.backend = MongoBackend()
|
|
|
|
def test_mongo_backend(self):
|
|
events = [{'test': 1}, {'test': 2}]
|
|
|
|
self.backend.send(events[0])
|
|
self.backend.send(events[1])
|
|
|
|
# Check if we inserted events into the database
|
|
|
|
calls = self.backend.collection.insert.mock_calls
|
|
|
|
assert len(calls) == 2
|
|
|
|
# Unpack the arguments and check if the events were used
|
|
# as the first argument to collection.insert
|
|
|
|
def first_argument(call):
|
|
_, args, _ = call
|
|
return args[0]
|
|
|
|
assert events[0] == first_argument(calls[0])
|
|
assert events[1] == first_argument(calls[1])
|