def test_queue_status(self):
"""queue_message() should set the queued status."""
router = DatabaseRouter()
dbm = router.queue_message("I", [self.create_connection()], "foo")
self.assertEqual("Q", dbm.status)
transmission = dbm.transmissions.all()[0]
self.assertEqual("Q", transmission.status)
def test_create_message_from_dbm(self):
"""Make sure the proper fields are passed through."""
router = DatabaseRouter()
dbm = router.queue_message(direction='I', text="foo",
connections=[self.create_connection()])
dbm2 = router.create_message_from_dbm(dbm, {'a': 'b'})
self.assertEqual({'a': 'b'}, dbm2.fields)
def test_queue_queryset_connections(self):
"""queue_message() can accept a queryset of connections."""
# 2 connections were created in setUp
connections = Connection.objects.all()
router = DatabaseRouter()
dbm = router.queue_message("I", connections, "foo")
self.assertEqual(2, dbm.transmissions.count())
def test_queue_queryset_connections(self):
"""queue_message() can accept a queryset of connections."""
# 2 connections were created in setUp
connections = Connection.objects.all()
router = DatabaseRouter()
dbm = router.queue_message("I", connections, "foo")
self.assertEqual(2, dbm.transmissions.count())
def test_queue_multi_connections(self):
"""Multiple transmissions should be created for > 1 connection."""
connections = [self.create_connection(), self.create_connection()]
router = DatabaseRouter()
dbm = router.queue_message("I", connections, "foo")
self.assertEqual(2, dbm.transmissions.count())
def test_queue_single_connection(self):
"""A single transmission should be created for 1 connection."""
connections = [self.create_connection()]
router = DatabaseRouter()
dbm = router.queue_message("I", connections, "foo")
self.assertEqual(1, dbm.transmissions.count())
