Add a couple tests for Connection.get_context
diff --git a/test/test_ssl.py b/test/test_ssl.py
index c460741..d7e04da 100644
--- a/test/test_ssl.py
+++ b/test/test_ssl.py
@@ -316,6 +316,25 @@
self.assertConsistentType(Connection, 'Connection', ctx, None)
+ def test_get_context(self):
+ """
+ L{Connection.get_context} returns the L{Context} instance used to
+ construct the L{Connection} instance.
+ """
+ context = Context(TLSv1_METHOD)
+ connection = Connection(context, None)
+ self.assertIdentical(connection.get_context(), context)
+
+
+ def test_get_context_wrong_args(self):
+ """
+ L{Connection.get_context} raises L{TypeError} if called with any
+ arguments.
+ """
+ connection = Connection(Context(TLSv1_METHOD), None)
+ self.assertRaises(TypeError, connection.get_context, None)
+
+
class ErrorTests(TestCase):
"""