def test_deleting_user(self):
utils.create_users(10)
code, msg = utils.delete(
self.app,
f'{self.link}/user/3',
self.headers
)
self.assertAlmostEqual(code, 200)
self.assertAlmostEqual(User.select().count(), 10)
code, msg = utils.delete(
self.app,
f'{self.link}/user/23',
self.headers
)
self.assertAlmostEqual(code, 404)
self.assertAlmostEqual(msg['error'], 'User does not exist.')
utils.create_posts(50)
seventh_user = User.get_by_id(7)
self.assertTrue(seventh_user.posts.count() > 0)
data = {'delete_posts': True}
code, msg = utils.delete(
self.app,
f'{self.link}/user/7',
self.headers,
'application/json',
data
)
self.assertAlmostEqual(code, 200)
num_of_posts = Post.select().where(Post.author == seventh_user).count()
self.assertAlmostEqual(num_of_posts, 0)
def get_others_posts(current_user):
posts = [
post.to_dict()
for post in Post.select().where(Post.author != current_user)
]
if not posts:
return error('No posts from other users.', 404)
return jsonify(posts), 200
def test_deleting_users_post(self):
utils.create_users(3)
utils.create_posts(50)
post_id = User.get_by_id(2).posts.first().get_id()
code, _ = utils.delete(
self.app,
f'{self.link}/user/2/post/70',
self.headers
)
self.assertAlmostEqual(code, 404)
code, post = utils.delete(
self.app,
f'{self.link}/user/2/post/{post_id}',
self.headers
)
self.assertAlmostEqual(code, 200)
self.assertIs(None, Post.select().where(Post.id == post_id).first())
def test_delete_post(self):
code, _ = utils.delete(self.app, f'{self.link}/:15', self.headers)
self.assertAlmostEqual(code, 404)
code, _ = utils.delete(self.app, f'{self.link}/:1', self.headers)
self.assertAlmostEqual(code, 200)
self.assertAlmostEqual(Post.select().count(), 0)