From: Stefan Metzmacher Date: Wed, 11 Nov 2020 15:49:25 +0000 (+0100) Subject: RawDCERPCTest: add some more auth_length related asserts X-Git-Tag: tdb-1.4.13~961 X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=bb8ad1f22924b581bfb66555713e98efa91372b2;p=thirdparty%2Fsamba.git RawDCERPCTest: add some more auth_length related asserts BUG: https://bugzilla.samba.org/show_bug.cgi?id=14356 Signed-off-by: Stefan Metzmacher Reviewed-by: Andreas Schneider --- diff --git a/python/samba/tests/dcerpc/raw_testcase.py b/python/samba/tests/dcerpc/raw_testcase.py index aaf063c2179..e45f20ef929 100644 --- a/python/samba/tests/dcerpc/raw_testcase.py +++ b/python/samba/tests/dcerpc/raw_testcase.py @@ -889,6 +889,7 @@ class RawDCERPCTest(TestCase): if len(ai) > samba.dcerpc.dcerpc.DCERPC_AUTH_TRAILER_LENGTH: p.auth_length = len(ai) - samba.dcerpc.dcerpc.DCERPC_AUTH_TRAILER_LENGTH else: + self.assertEqual(len(ai), 0) p.auth_length = 0 p.call_id = call_id p.u = payload @@ -1001,10 +1002,10 @@ class RawDCERPCTest(TestCase): if len(ai) > samba.dcerpc.dcerpc.DCERPC_AUTH_TRAILER_LENGTH: self.assertEqual(p.auth_length, len(ai) - samba.dcerpc.dcerpc.DCERPC_AUTH_TRAILER_LENGTH) - elif auth_length is not None: - self.assertEqual(p.auth_length, auth_length) - else: + elif auth_length is None: self.assertEqual(p.auth_length, 0) + if auth_length is not None: + self.assertEqual(p.auth_length, auth_length) self.assertEqual(p.call_id, call_id) return