11from tests .conftest import client
22from tests import db , User
3+ from tests .api import create_user_for_test_cases
34
45class TestUserView (object ):
56
7+ """
8+
9+ """
10+
11+ # valid data for user creation
612 valid_data = {
713 'name' : 'L Jone' ,
814 'bio' : 'coding...' ,
@@ -12,12 +18,6 @@ class TestUserView(object):
1218 'occupation' : 'cashier'
1319 }
1420
15- def create_user_for_test_cases (self ):
16- new_user = User (** self .valid_data )
17- db .session .add (new_user )
18- db .session .commit ()
19- return new_user .id
20-
2121 def test_create_user (self , client ):
2222 response = client .post ('/users' , json = self .valid_data )
2323 assert response .status_code == 201
@@ -26,7 +26,7 @@ def test_create_user(self, client):
2626 assert response .status_code == 400
2727
2828 def test_update_user (self , client ):
29- user_id = self . create_user_for_test_cases ()
29+ user_id = create_user_for_test_cases (self . valid_data )
3030
3131 response = client .post ('/users/1' , json = {})
3232 assert response .status_code == 400
@@ -46,7 +46,7 @@ def test_delete_user(self, client):
4646 response = client .delete ('/users/{}' .format (user_id ))
4747 assert response .status_code == 404
4848
49- user_id = self . create_user_for_test_cases ()
49+ user_id = create_user_for_test_cases (self . valid_data )
5050 response = client .delete ('/users/{}' .format (user_id ))
5151 assert response .status_code == 200
5252
@@ -55,15 +55,15 @@ def test_get_user(self, client):
5555 response = client .get ('/users/{}' .format (user_id ))
5656 assert response .status_code == 404
5757
58- user_id = self . create_user_for_test_cases ()
58+ user_id = create_user_for_test_cases (self . valid_data )
5959 response = client .get ('/users/{}' .format (user_id ))
6060 assert response .status_code == 200
6161
6262 created_user = User .query .filter_by (id = user_id ).first ()
6363 assert response .get_json () == created_user .as_dict ()
6464
6565 def test_get_all_users (self , client ):
66- self . create_user_for_test_cases ()
66+ create_user_for_test_cases (self . valid_data )
6767
6868 self .valid_data = {
6969 'name' : 'Valid' ,
@@ -74,7 +74,7 @@ def test_get_all_users(self, client):
7474 'occupation' : 'cashier2.1'
7575 }
7676
77- self . create_user_for_test_cases ()
77+ create_user_for_test_cases (self . valid_data )
7878
7979 response = client .get ('/users' )
8080 assert response .status_code == 200
@@ -89,3 +89,9 @@ def test_get_all_users(self, client):
8989 response = client .get ('/users' )
9090 assert response .status_code == 200
9191 assert response .get_json () == [users_dict [1 ]]
92+
93+ l = User (** self .valid_data )
94+ db .session .add (l )
95+ db .session .commit ()
96+
97+ assert l .name == 1
0 commit comments