def seed(number_of_fake_users=100): seed_data = config("seed_api_data", namespace="cis", default="false") if seed_data.lower() == "true": table = get_table_resource() user_profile = user.Profile(table, None, False) if len(user_profile.all) > 0: logger.info( "Identity vault is already seeded. Passing on the additiona of users." ) pass else: logger.info("Beginning the creation of seed users.") identities = batch_create_fake_profiles(1337, number_of_fake_users) for identity in identities: identity["pronouns"]["metadata"]["display"] = None identity_vault_data_structure = { "id": identity.get("user_id").get("value"), "primary_email": identity.get("primary_email").get("value"), "user_uuid": identity.get("uuid").get("value"), "primary_username": identity.get("primary_username").get("value"), "sequence_number": "1234567890", "profile": dumps(identity), } user_profile.create(identity_vault_data_structure) identities = batch_create_fake_profiles(1337, 1) for identity in identities: identity["pronouns"]["metadata"]["display"] = None identity["active"]["value"] = False identity_vault_data_structure = { "id": identity.get("user_id").get("value"), "primary_email": identity.get("primary_email").get("value"), "user_uuid": identity.get("uuid").get("value"), "primary_username": identity.get("primary_username").get("value"), "sequence_number": "1234567890", "profile": dumps(identity), } user_profile.create(identity_vault_data_structure) logger.info( "Count: {} seed users created.".format(number_of_fake_users))
def seed(number_of_fake_users=100): seed_data = config("seed_api_data", namespace="cis", default="false") if seed_data.lower() == "true": table = get_table_resource() user_profile = user.Profile(table, None, False) if len(user_profile.all) > 0: pass else: identities = batch_create_fake_profiles(1337, number_of_fake_users) for identity in identities: identity["pronouns"]["metadata"]["display"] = None identity_vault_data_structure = { "id": identity.get("user_id").get("value"), "primary_email": identity.get("primary_email").get("value"), "user_uuid": identity.get("uuid").get("value"), "primary_username": identity.get("primary_username").get("value"), "sequence_number": "1234567890", "profile": dumps(identity), } user_profile.create(identity_vault_data_structure) identities = batch_create_fake_profiles(1337, 1) for identity in identities: identity["pronouns"]["metadata"]["display"] = None identity["active"]["value"] = False identity_vault_data_structure = { "id": identity.get("user_id").get("value"), "primary_email": identity.get("primary_email").get("value"), "user_uuid": identity.get("uuid").get("value"), "primary_username": identity.get("primary_username").get("value"), "sequence_number": "1234567890", "profile": dumps(identity), } user_profile.create(identity_vault_data_structure)
def test_batch_create(self): profiles = fake_profile.batch_create_fake_profiles(seed=1337, number=3) assert len(profiles) == 3 for i, p in enumerate(profiles, 1): assert p is not None assert p["access_information"]["hris"]["values"][ "employee_id"] == i
def batch(seed, number): """Create many IAM profile v2 objects.""" profiles = batch_create_fake_profiles(seed, number) click.echo(json.dumps(profiles, indent=2))