49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
# Copyright (c) Twisted Matrix Laboratories.
|
|
# See LICENSE for details.
|
|
|
|
"""
|
|
Tests for L{twisted.python.context}.
|
|
"""
|
|
|
|
|
|
from twisted.python import context
|
|
from twisted.trial.unittest import SynchronousTestCase
|
|
|
|
|
|
class ContextTests(SynchronousTestCase):
|
|
"""
|
|
Tests for the module-scope APIs for L{twisted.python.context}.
|
|
"""
|
|
|
|
def test_notPresentIfNotSet(self):
|
|
"""
|
|
Arbitrary keys which have not been set in the context have an associated
|
|
value of L{None}.
|
|
"""
|
|
self.assertIsNone(context.get("x"))
|
|
|
|
def test_setByCall(self):
|
|
"""
|
|
Values may be associated with keys by passing them in a dictionary as
|
|
the first argument to L{twisted.python.context.call}.
|
|
"""
|
|
self.assertEqual(context.call({"x": "y"}, context.get, "x"), "y")
|
|
|
|
def test_unsetAfterCall(self):
|
|
"""
|
|
After a L{twisted.python.context.call} completes, keys specified in the
|
|
call are no longer associated with the values from that call.
|
|
"""
|
|
context.call({"x": "y"}, lambda: None)
|
|
self.assertIsNone(context.get("x"))
|
|
|
|
def test_setDefault(self):
|
|
"""
|
|
A default value may be set for a key in the context using
|
|
L{twisted.python.context.setDefault}.
|
|
"""
|
|
key = object()
|
|
self.addCleanup(context.defaultContextDict.pop, key, None)
|
|
context.setDefault(key, "y")
|
|
self.assertEqual("y", context.get(key))
|