def test_add_items_dedup(self): managed_mapped_list = ManagedMappedList('test_add_items_dedup', 'test_key', 'test_name', 'test_description', TEST_ZK_HOSTS, TEST_AWS_KEYFILE, TEST_S3_BUCKET) self.assertTrue(managed_mapped_list.add_items('key', [2, 2, 3])) self.assertEqual([2, 3], managed_mapped_list.get_list('key')) self.assertTrue(managed_mapped_list.add_items('key', [3])) self.assertEqual([2, 3], managed_mapped_list.get_list('key'))
def test_add_item(self): managed_mapped_list = ManagedMappedList('test_add_item', 'test_key', 'test_name', 'test_description', TEST_ZK_HOSTS, TEST_AWS_KEYFILE, TEST_S3_BUCKET, dedup=False) 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, 2], managed_mapped_list.get_list('key'))
def test_set_and_remove_dedup(self): managed_mapped_list = ManagedMappedList('test_set_and_remove_dedup', '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.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, 3], 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], managed_mapped_list.get_list('key'))
def test_set_dedup(self): managed_mapped_list = ManagedMappedList( 'test_set_unique', 'test_key', 'test_name', 'test_description', TEST_ZK_HOSTS, TEST_AWS_KEYFILE, TEST_S3_BUCKET) self.assertTrue(managed_mapped_list.set_list('key', [1, 2])) self.assertEqual([1, 2], 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'))