def register_user(osm_id, username, changeset_count, picture_url, email): """ Creates user in DB :param osm_id: Unique OSM user id :param username: OSM Username :param changeset_count: OSM changeset count """ new_user = User() new_user.id = osm_id new_user.username = username if picture_url is not None: new_user.picture_url = picture_url intermediate_level = current_app.config["MAPPER_LEVEL_INTERMEDIATE"] advanced_level = current_app.config["MAPPER_LEVEL_ADVANCED"] if changeset_count > advanced_level: new_user.mapping_level = MappingLevel.ADVANCED.value elif intermediate_level < changeset_count < advanced_level: new_user.mapping_level = MappingLevel.INTERMEDIATE.value else: new_user.mapping_level = MappingLevel.BEGINNER.value if email is not None: new_user.email_address = email new_user.create() return new_user
def create_canned_user() -> User: """ Generate a canned user in the DB """ test_user = User() test_user.id = TEST_USER_ID test_user.username = "******" test_user.mapping_level = 1 test_user.create() return test_user
def setUp(self): # Arrange test_user = User() test_user.role = UserRole.MAPPER.value test_user.id = 12 test_user.mapping_level = MappingLevel.BEGINNER.value test_user.username = "******" test_user.email_address = "*****@*****.**" self.test_user = test_user """ Setup test context so we can connect to database """ self.app = create_app() self.ctx = self.app.app_context() self.ctx.push() if self.skip_tests: return