def save_user(self): """ Save user to database. params: user_info: dict returns: bloolean """ try: email = self.user_info['email'] name = self.user_info['name'] picture = self.user_info['picture'] user = User.query.filter_by(email=email).first() role = Role.query.filter_by(role='Default User').first() if not role: role = Role(role='Default User') role.save() if not user: try: user_data = User(email=email, name=name, picture=picture) user_data.roles.append(role) user_data.save() notification_settings = NotificationModel( user_id=user_data.id) notification_settings.save() except Exception as e: # noqa db_session.rollback() except SQLAlchemyError: pass return True
def setUp(self): app = self.create_app() self.app_test = app.test_client() with app.app_context(): Base.metadata.create_all(bind=engine) command.stamp(self.alembic_configuration, 'head') command.downgrade(self.alembic_configuration, '-1') command.upgrade(self.alembic_configuration, 'head') user = User(fullname="doe jon", username="******", email="*****@*****.**") user.save() challenge = Challenge(title="zanku challenge", start_date="2019-08-15 06:00:49.303402", end_date="2019-08-15 06:00:49.303402", description="*****@*****.**", image="https://picsum.photos/200", prize="10000") challenge.save() contestant = Contestant(challenge=1, contestant=2, state="subscribed") contestant.save() challenge_entry = ChallengeEntry(contestant_id=2, challenge_id=1, file_url="http://foobar.hj") challenge_entry.save()
def create_user(): user = User(email="*****@*****.**", location="Lagos", name="test test", picture="www.andela.com/test") user.save() db_session().commit()
def test_paginate__users_query(self): user = User(email='*****@*****.**', name="test test", picture="www.andela.com/test") user.save() self.client.execute(user_query, context_value={'session': db_session}) CommonTestCases.admin_token_assert_equal(self, paginated_users_query, paginated_users_response)
def func_wrapper(self): headers = {"Authorization": "Bearer" + " " + USER_TOKEN} user = User(email='*****@*****.**', name='this user', location="Nairobi", picture='www.andela.com/user') user.save() user.roles.append(role) db_session().commit() return headers
def create_user(): user = User(email='*****@*****.**', location="Kampala", name="test test", picture="www.andela.com/testuser") user.save() role = base.role user.roles.append(role) db_session().commit() return user
def func_wrapper(self): user_role = Role(role='Default User') user_role.save() user = User(email='*****@*****.**', name='Test user', location="Lagos", picture='www.andela.com/testuser') user.save() user.roles.append(user_role) db_session().commit()
def test_user_delete_admin(self): admin_user = User(email="*****@*****.**", location="Kampala", name="test test", picture="www.andela.com/test") admin_user.save() role = base.role admin_user.roles.append(role) CommonTestCases.user_token_assert_in( self, delete_user, "You are not authorized to perform this action")
def func_wrapper(self): headers = {"Authorization": "Bearer" + " " + user_api_token} user = User(email='*****@*****.**', name='this user', location="Nairobi", picture='www.andela.com/user') user.save() user_role = UsersRole(user_id=user.id, role_id=1) user_role.save() db_session().commit() return headers
def test_user_delete_admin(self): admin_user = User(email="*****@*****.**", location="Kampala", name="test test", picture="www.andela.com/test") admin_user.save() user_role = UsersRole(user_id=admin_user.id, role_id=1) user_role.save() CommonTestCases.user_token_assert_in( self, delete_user, "You are not authorized to perform this action")
def test_query_users_by_email(self): user = User(email='*****@*****.**', name="test test", picture="www.andela.com/test") user.save() db_session().commit() execute_query = self.client.execute( query_user_by_email, context_value={'session': db_session}) expected_responese = query_user_email_response self.assertEqual(execute_query, expected_responese)
def test_deleteuser_when_admin(self): user = User(email="*****@*****.**", location="Kampala", name="test test", picture="www.andela.com/test") user.save() role = Role(role="Default User") role.save() user.roles.append(role) CommonTestCases.admin_token_assert_equal(self, delete_user_2, expected_query_after_delete)
def test_change_user_role(self): user = User(email='*****@*****.**', location="Kampala", name="test test", picture="www.andela.com/testuser") user.save() user_role = UsersRole(user_id=user.id, role_id=1) user_role.save() db_session().commit() CommonTestCases.admin_token_assert_equal( self, change_user_role_mutation_query, change_user_role_mutation_response)
def test_query_users(self): """ Testing for User creation """ user = User(email='*****@*****.**', name="test test", picture="www.andela.com/test") user.save() execute_query = self.client.execute( user_query, context_value={'session': db_session}) expected_response = user_query_response self.assertEqual(execute_query, expected_response)
def test_query_users_role_by_role(self): user = User(email='*****@*****.**', location="Lagos", name="test test", picture="www.andela.com/testuser") user.save() role = base.role user.roles.append(role) db_session().commit() execute_query = self.client.execute( query_user_by_user_email, context_value={'session': db_session}) expected_response = query_user_by_user_email_response self.assertEqual(execute_query, expected_response)
def test_query_users_role_by_role(self): user = User(email='*****@*****.**', location="Lagos", name="test test", picture="www.andela.com/testuser") user.save() user_role = UsersRole(user_id=user.id, role_id=1) user_role.save() db_session().commit() execute_query = self.client.execute( query_user_by_user_id, context_value={'session': db_session}) expected_response = query_user_by_user_id_response self.assertEqual(execute_query, expected_response)
def mutate(self, info, **kwargs): validate_entity_types(['admin', 'citizen', 'politician'], 'user_type', **kwargs) # noqa user = UserModel(**kwargs) if not verify_email(user.email): raise GraphQLError("This email is not allowed") payload = { 'model': UserModel, 'field': 'email', 'value': kwargs['email'] } with SaveContextManager(user, 'User email', payload): token = user.generate_token() return CreateUser(user=user, token=token)
def test_user_role_creation(self): """ Testing for User Role creation """ user = User(email="*****@*****.**", location="Lagos", name="test test", picture="www.andela.com/test") user.save() db_session().commit() execute_query = self.client.execute( user_role_mutation_query, context_value={'session': db_session}) expected_responese = user_role_mutation_response self.assertEqual(execute_query, expected_responese)
def setUp(self): app = self.create_app() self.app_test = app.test_client() with app.app_context(): Base.metadata.create_all(bind=engine) admin_user = User(email="*****@*****.**", location="Kampala", name="Peter Walugembe", picture="https://www.andela.com/walugembe") admin_user.save() role = Role(role="Admin") role.save() user_role = UsersRole(user_id=admin_user.id, role_id=role.id) user_role.save() location = Location(name='Kampala', abbreviation='KLA') location.save() location_two = Location(name='Nairobi', abbreviation='NBO') location_two.save() office = Office(name="St. Catherines", location_id=location.id) office.save() office_two = Office(name="dojo", location_id=location_two.id) office_two.save() block = Block(name='EC', office_id=office.id) block.save() floor = Floor(name='3rd', block_id=block.id) floor.save() room = Room( name='Entebbe', room_type='meeting', capacity=6, floor_id=floor.id, calendar_id= '*****@*****.**', # noqa: E501 image_url= "https://www.officelovin.com/wp-content/uploads/2016/10/andela-office-main-1.jpg" ) # noqa: E501 room.save() resource = Resource(name='Markers', quantity=3, room_id=room.id) resource.save() device = Devices(resource_id=resource.id, last_seen="2018-06-08T11:17:58.785136", date_added="2018-06-08T11:17:58.785136", name="Samsung", location="Nairobi", device_type="External Display") device.save() db_session.commit()
def test_query_users_role(self): """ Testing for query User role """ user = User(email="*****@*****.**", location="Lagos", name="test test", picture="www.andela.com/testuser") user.save() user.roles.append(user_role) db_session().commit() execute_query = self.client.execute( user_role_query, context_value={'session': db_session}) expected_responese = user_role_query_response self.assertEqual(execute_query, expected_responese)
def create_user(): user = User(email=request.form.get('email'), name=request.form.get('name'), password=request.form.get('password')) db.session.add(user) db.session.commit() return 'Successfully created user'
def mutate(self, info, **kwargs): query_user = User.get_query(info) existing_user = query_user.filter_by(email=kwargs['email']).first() if existing_user: token = existing_user.generate_token() return CreateUser(user=existing_user, token=token) user = UserModel(**kwargs) if user.email and not verify_email(user.email): raise GraphQLError("This email is not allowed") payload = { 'model': UserModel, 'field': 'email', 'value': kwargs['email'], # noqa } with SaveContextManager(user, 'email', payload): token = user.generate_token() return CreateUser(user=user, token=token)
def setUp(self): app = self.create_app() self.app_test = app.test_client() with app.app_context(): Base.metadata.create_all(bind=engine) command.stamp(self.alembic_configuration, 'head') # command.downgrade(self.alembic_configuration, '-1') command.upgrade(self.alembic_configuration, 'head') user = User(name="Ken", password="******", user_type='default') user.save() product = Product(name="Eggs", manufacturer="none", price=200, tax=0, UOM="none", user_id="1") product.save() db_session.commit()
def wrapper(*args, **kwargs): auth_header = request.headers.get('Authorization') auth_token = auth_header.split(" ")[1] user_type = User.decode_token(auth_token).get('user_type') if user_type in expected_args: res = fn(*args, **kwargs) return res else: raise GraphQLError( "You are not authorized to perform this action" )
def wrapper(*args, **kwargs): if 'Authorization' not in request.headers: raise GraphQLError('Token not provided or is invalid.') user_id = None auth_header = request.headers.get('Authorization') if auth_header and len(auth_header.split(" ")) == 2: auth_token = auth_header.split(" ")[1] else: auth_token = '' try: user_id = User.decode_token(auth_token).get('id') user_type = User.decode_token(auth_token).get('user_type') g.user_id = user_id g.user_type = user_type except (jwt.ExpiredSignatureError, jwt.InvalidTokenError): raise GraphQLError('Token not provided or is invalid.') return fn(*args, **kwargs)
def seed_data(apps, schema_editor): user = User(name='boris', email='*****@*****.**', is_staff=True, is_superuser=True, gender='Male') user.set_password('1234qwe') user.save()
def seed_data(apps, schema_editor): user = User( name='sagar', email='*****@*****.**', is_staff=True, is_superuser=True, ) user.set_password('12345') user.save()
def seed_data(apps, schema_editor): user = User(name='Sumit', email='*****@*****.**', is_staff=True, is_superuser=True, phone='9876543210', gender="Male", address="17, Mayur Park Society") user.set_password('1234') user.save()
def save_user(self, email, *expected_args): # noqa: C901 """ Save user to database. params: user_info: dict returns: bloolean """ try: email = self.user_info['email'] name = self.user_info['name'] picture = self.user_info['picture'] user = User.query.filter_by(email=email).first() role = Role.query.filter_by(role='Default User').first() if not role: role = Role(role='Default User') role.save() if not user: try: response = self.get_user_details_from_api( email, *expected_args) user_data = User(email=email, name=name, picture=picture) user_data.roles.append(role) for value in response['values']: if user_data.email == value["email"]: if value['location']: # pragma: no cover check_and_add_location( value['location']['name']) notification_settings = NotificationModel( user_id=user_data.id) notification_settings.save() except Exception as e: # noqa db_session.rollback() except SQLAlchemyError: # pragma: no cover pass return True
def return_logged_in_users_id(): auth_header = request.headers.get('Authorization') auth_token = auth_header.split(" ")[1] user_id = User.decode_token(auth_token).get('id') return user_id