def test_empty_remove(self):
managed_mapped_list = ManagedMappedList(
'test_empty_remove', 'test_key', 'test_name', 'test_description',
TEST_ZK_HOSTS, TEST_AWS_KEYFILE, TEST_S3_BUCKET)
self.assertTrue(managed_mapped_list.remove_list('key'))
self.assertTrue(managed_mapped_list.remove_item('key', 'val'))
self.assertTrue(managed_mapped_list.remove_items('key', ['val']))
def test_set(self):
managed_mapped_list = ManagedMappedList('test_set',
'test_key',
'test_name',
'test_description',
TEST_ZK_HOSTS,
TEST_AWS_KEYFILE,
TEST_S3_BUCKET,
dedup=False)
self.assertTrue(managed_mapped_list.set_list('key', [1, 2, 2]))
self.assertEqual([1, 2, 2], managed_mapped_list.get_list('key'))
def test_contains(self):
managed_mapped_list = ManagedMappedList(
'test_contains', 'test_key', 'test_name', 'test_description',
TEST_ZK_HOSTS, TEST_AWS_KEYFILE, TEST_S3_BUCKET)
managed_mapped_list.set_list('key', [1, 2, 2, 3, 2])
self.assertTrue(managed_mapped_list.contain_item('key', 2))
self.assertFalse(managed_mapped_list.contain_item('key2', 2))
self.assertFalse(managed_mapped_list.contain_item('key', 4))
self.assertEqual([2], managed_mapped_list.contain_items('key', [2, 4]))
self.assertEqual([], managed_mapped_list.contain_items('key2', [2, 4]))
self.assertEqual([], managed_mapped_list.contain_items('key', [4]))
def test_add_item_dedup(self):
managed_mapped_list = ManagedMappedList(
'test_add_item_dedup', 'test_key', 'test_name', 'test_description',
TEST_ZK_HOSTS, TEST_AWS_KEYFILE, TEST_S3_BUCKET)
self.assertTrue(managed_mapped_list.add_item('key', 2))
self.assertEqual([2], managed_mapped_list.get_list('key'))
self.assertTrue(managed_mapped_list.add_item('key', 2))
self.assertEqual([2], managed_mapped_list.get_list('key'))
self.assertTrue(managed_mapped_list.add_item('key', 3))
self.assertEqual([2, 3], managed_mapped_list.get_list('key'))
def test_callback(self):
update_func = mock.Mock()
_patch = mock.patch.object(self, '_update', update_func)
_patch.start()
self.addCleanup(_patch.stop)
managed_mapped_list = ManagedMappedList('test_domain5',
'test_key',
'test_name',
'test_description',
TEST_ZK_HOSTS,
TEST_AWS_KEYFILE,
TEST_S3_BUCKET,
update_callback=self._update)
self.assertTrue(
managed_mapped_list.zk_config_manager.update_zk('{}', '{1:[2]}'))
self.assertTrue(update_func.is_called)
def test_add_items(self):
managed_mapped_list = ManagedMappedList('test_add_items',
'test_key',
'test_name',
'test_description',
TEST_ZK_HOSTS,
TEST_AWS_KEYFILE,
TEST_S3_BUCKET,
dedup=False)
self.assertTrue(managed_mapped_list.add_items('key', [2, 2, 3]))
self.assertEqual([2, 2, 3], managed_mapped_list.get_list('key'))
self.assertTrue(managed_mapped_list.add_items('key', [2, 3]))
self.assertEqual([2, 2, 3, 2, 3], managed_mapped_list.get_list('key'))
def test_get(self):
managed_mapped_list = ManagedMappedList(
'test_get', 'test_key', 'test_name', 'test_description',
TEST_ZK_HOSTS, TEST_AWS_KEYFILE, TEST_S3_BUCKET)
self.assertEqual({}, managed_mapped_list.get_all())
self.assertEqual([], managed_mapped_list.get_list('key'))
def test_set_and_remove(self):
managed_mapped_list = ManagedMappedList('test_set_and_remove',
'test_key',
'test_name',
'test_description',
TEST_ZK_HOSTS,
TEST_AWS_KEYFILE,
TEST_S3_BUCKET,
dedup=False)
managed_mapped_list.set_list('key', [1, 2, 2, 3, 2])
self.assertTrue(managed_mapped_list.remove_list('key'))
self.assertEqual([], managed_mapped_list.get_list('key'))
managed_mapped_list.set_list('key', [1, 2, 2, 3, 2])
self.assertTrue(managed_mapped_list.remove_item('key', 2))
self.assertEqual([1, 2, 3, 2], managed_mapped_list.get_list('key'))
managed_mapped_list.set_list('key', [1, 2, 2, 3, 2])
self.assertTrue(managed_mapped_list.remove_items('key', [2, 3]))
self.assertEqual([1, 2, 2], managed_mapped_list.get_list('key'))
