File tree Expand file tree Collapse file tree 2 files changed +5
-13
lines changed Expand file tree Collapse file tree 2 files changed +5
-13
lines changed Original file line number Diff line number Diff line change @@ -38,10 +38,7 @@ def delete(self, id):
38
38
.update_one (pull__following = user .to_dbref ())
39
39
User .objects (id = id ) \
40
40
.update_one (pull__follower = current_user .to_dbref ())
41
- status_fields = generate_status_fields (OK )
42
- else :
43
- status_fields = generate_status_fields (NOTFOUND )
44
- return status_fields , 200
41
+ return '' , 204
45
42
46
43
47
44
class BlockUserApi (Resource ):
@@ -69,7 +66,4 @@ def delete(self, id):
69
66
if id in current_user .blocked_user_id :
70
67
User .objects (id = current_user .id ) \
71
68
.update_one (pull__blocked_user_id = id )
72
- status_fields = generate_status_fields (OK )
73
- else :
74
- status_fields = generate_status_fields (NOTFOUND )
75
- return status_fields , 200
69
+ return '' , 204
Original file line number Diff line number Diff line change 1
- from __future__ import absolute_import
2
1
# coding=utf-8
2
+ from __future__ import absolute_import
3
3
from flask import url_for
4
4
from flask_login import current_user
5
5
import pytest
@@ -51,8 +51,7 @@ def test_follow_user_api(self):
51
51
# test unfollow
52
52
url = url_for ('api.followuserapi' , id = self .users [1 ].id )
53
53
rv = self .client .delete (url , buffered = True )
54
- assert rv .status_code == 200
55
- assert rv .json ['status' ] == OK
54
+ assert rv .status_code == 204
56
55
self .users [0 ].reload ()
57
56
self .users [1 ].reload ()
58
57
assert self .users [0 ].following == []
@@ -72,7 +71,6 @@ def test_block_user_api(self):
72
71
# test unblock
73
72
url = url_for ('api.blockuserapi' , id = self .users [1 ].id )
74
73
rv = self .client .delete (url , buffered = True )
75
- assert rv .status_code == 200
76
- assert rv .json ['status' ] == OK
74
+ assert rv .status_code == 204
77
75
self .users [0 ].reload ()
78
76
assert self .users [0 ].following == []
You can’t perform that action at this time.
0 commit comments