def test_add_existing_scope_and_code(self):
id_generator = IdGenerator()
_, gen_scope, gen_code = id_generator.generate_scope_and_code(
"portfolio")
id_generator.add_scope_and_code("portfolio", gen_scope, gen_code)
gen_vals = set(id_generator.pop_scope_and_codes())
self.assertEqual(1, len(gen_vals))
def test_add_scope_and_code_with_specified_scope(self):
test_scope = "test-scope"
id_generator = IdGenerator()
code = str(uuid.uuid4())
id_generator.add_scope_and_code("portfolio", test_scope, code)
val = next(id_generator.pop_scope_and_codes(), None)
self.assertEqual(test_scope, val[1])
self.assertEqual(code, val[2])
def test_add_scope_and_code(self):
id_generator = IdGenerator()
code = str(uuid.uuid4())
id_generator.add_scope_and_code("portfolio", IdGenerator.default_scope,
code)
val = next(id_generator.pop_scope_and_codes(), None)
self.assertEqual(IdGenerator.default_scope, val[1])
self.assertEqual(code, val[2])
def test_add_scope_and_code_with_annotations(self):
id_generator = IdGenerator()
test_scope = "test-scope"
test_code = "test-code"
annotations = ["val1", "val2"]
item = id_generator.add_scope_and_code("portfolio",
scope=test_scope,
code=test_code,
annotations=annotations)
self.assertEqual("portfolio", item[0])
self.assertEqual(test_scope, item[1])
self.assertEqual(test_code, item[2])
self.assertEqual("val1", item[3])
self.assertEqual("val2", item[4])
