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])