def test_group_empty(self): request = GroupRequest() transaction = GroupTransaction() # Make sure that prepare and commit do not throw an exception. transaction.prepare(request) transaction.commit(self._trie, self._get_blessed_version(), self._store)
def test_group(self): request = GroupRequest() request.namePrefix = '/some_dir/' request.groupSuffix = '/' transaction = GroupTransaction() transaction.prepare(request) response = transaction.commit(self._trie, self._get_blessed_version(), self._store) expected_groups = set() for i in range(0, 10): expected_groups.add('/some_dir/some_token_%d' % i) groups = set() for group, count in response.counts.iteritems(): groups.add(group) self.assertEqual(11, count) self.assertEqual(expected_groups, groups)