check buffer size for GSSAPI

thobbs · Jan 27, 2016 · 4f87cdd · 4f87cdd
1 parent 8045568
commit 4f87cdd
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/unit/test_mechanism.py b/tests/unit/test_mechanism.py
@@ -115,15 +115,17 @@ def test_process_qop(self, *args):
         self.assertRaises(SASLProtocolException, self.sasl.process, msg)
 
         max_len = 100
+        self.assertLess(max_len, self.sasl.max_buffer)
         for i, qop in ((1, 'auth'), (2, 'auth-int'), (4, 'auth-conf')):  # 1, 2, 4 --> qop flag
-            qop_size = struct.pack('!i', i << 24 | 100)
+            qop_size = struct.pack('!i', i << 24 | max_len)
             response = base64.b64encode(qop_size)
             with patch('puresasl.mechanisms.kerberos.authGSSClientResponse', return_value=response), \
                     patch('puresasl.mechanisms.kerberos.authGSSClientWrap') as authGSSClientWrap:
                 self.mechanism.complete = False
                 self.assertEqual(self.sasl.process(msg), qop_size)
                 self.assertTrue(self.mechanism.complete)
                 self.assertEqual(self.mechanism.qop, qop)
+                self.assertEqual(self.mechanism.max_buffer, max_len)
 
                 args = authGSSClientWrap.call_args[0]
                 out_data = args[1]