diff options
| -rwxr-xr-x | source4/scripting/python/samba/tests/dcerpc/rpc_talloc.py | 11 |
1 files changed, 4 insertions, 7 deletions
diff --git a/source4/scripting/python/samba/tests/dcerpc/rpc_talloc.py b/source4/scripting/python/samba/tests/dcerpc/rpc_talloc.py index 6dd3d07e84..5058e08960 100755 --- a/source4/scripting/python/samba/tests/dcerpc/rpc_talloc.py +++ b/source4/scripting/python/samba/tests/dcerpc/rpc_talloc.py @@ -23,7 +23,6 @@ samba.talloc_enable_null_tracking() class TallocTests(samba.tests.TestCase): '''test talloc behaviour of pidl generated python code''' - def check_blocks(self, object, num_expected): '''check that the number of allocated blocks is correct''' nblocks = samba.talloc_total_blocks(object) @@ -31,7 +30,6 @@ class TallocTests(samba.tests.TestCase): nblocks -= self.initial_blocks self.assertEquals(nblocks, num_expected) - def get_rodc_partial_attribute_set(self): '''get a list of attributes for RODC replication''' partial_attribute_set = drsuapi.DsPartialAttributeSet() @@ -46,22 +44,21 @@ class TallocTests(samba.tests.TestCase): # we expect one block object, a structure, an ARRAY, and a # reference to the array - self.check_blocks(partial_attribute_set, 4) + self.check_blocks(partial_attribute_set, 3) return partial_attribute_set def pas_test(self): pas = self.get_rodc_partial_attribute_set() - self.check_blocks(pas, 4) + self.check_blocks(pas, 3) req8 = drsuapi.DsGetNCChangesRequest8() self.check_blocks(req8, 2) - self.check_blocks(None, 6) + self.check_blocks(None, 5) req8.partial_attribute_set = pas if req8.partial_attribute_set.attids[1] != 2: raise Exception("Wrong value in attids[2]") # we now get an additional reference - samba.talloc_report_full(None) - self.check_blocks(None, 7) + self.check_blocks(None, 6) def test_run(self): self.initial_blocks = samba.talloc_total_blocks(None) |