def test_delete_membership_before_user(self): """ if user's membership is deleted then user should still exist """ u, m = self._create_user_with_membership() self.assertIsNotNone(m) self.assertEqual(User.get_by_id(u.id).members, [m]) m.delete() self.assertIsNone(Member.get_by_id(m.id)) self.assertIsNotNone(User.get_by_id(u.id)) self.assertEqual(User.get_by_id(u.id).members, [])
def test_delete_purchase(self): """ Purchase deleted -> Member NOT deleted -> User NOT deleted """ p, m, u = self._create_purchase_with_member_with_user() p.delete() self.assertEqual(Purchase.query.count(), 0) assert not Purchase.get_by_id(p.id) and Member.get_by_id(m.id) and User.get_by_id(u.id) self.assertEqual(len(m.purchases), 0)
def test_delete_membership(self): """ Member deleted -> Purchases by that member deleted -> User NOT deleted """ p, m, u = self._create_purchase_with_member_with_user() m.delete() self.assertIsNone(Purchase.get_by_id(p.id)) self.assertIsNone(Member.get_by_id(m.id)) self.assertIsNotNone(User.get_by_id(u.id)) self.assertEqual(u.members, [])
def test_delete_user_before_membership(self): """ if user is deleted then user's membership should be deleted """ u, m = self._create_user_with_membership() self.assertEqual(u.members, [m]) self.assertIsNotNone(m) u.delete() self.assertIsNone(User.get_by_id(u.id)) self.assertIsNone(Member.get_by_id(m.id))
def get_blogs(user_id=None): user = None if user_id: user = User.get_by_id(user_id) elif not user_id: user = User.get_by_email(session.get('email')) if not user: return render_template('error_404.html') user_blogs = user.get_blogs() return render_template('myblogs.html', blogs=user_blogs, email=session.get('email'), c=0)
def user_interact(user_id): """ Get user from DB by id param: user_id returns user """ # update user if request.method == 'POST': form = AuthForm(request.form) User.update(user_id, { 'email': form.email.data, 'password': form.password.data }) return f"Successfully updated user {user_id}\n" # get user user = User.get_by_id(user_id) return make_response(json.dumps(user, cls=AlchemyEncoder))
def load_user(user_id): """ User by the login manager Load the user object from user id stored in session """ return User.get_by_id(user_id)
def test_delete_user(self): u = self._create_user() u.delete() self.assertEqual(User.query.count(), 0) self.assertIsNone(User.get_by_id(u.id))
def test_update_user(self): u = self._create_user() new_account_id = 'abc456' u.update(account_id=new_account_id) self.assertEqual(User.get_by_id(u.id).account_id, new_account_id)