@@ -18,27 +18,52 @@ def mapper(self):
18
18
19
19
def test_get (self , mapper ):
20
20
url = self .get_mapper_url (mapper )
21
- expected_attr = {"id" : 1 , "name" : "test" }
21
+ expected_attr = {
22
+ "count" : 1 , "next" : None , "previous" : None ,
23
+ "results" : [{"id" : 1 , "name" : "test" }]
24
+ }
22
25
with requests_mock .Mocker () as m :
23
26
m .register_uri ("get" , url , json = expected_attr )
24
27
child_mapper = next (mapper .get ())
25
28
26
- for key , val in expected_attr .items ():
29
+ for key , val in expected_attr [ "results" ][ 0 ] .items ():
27
30
assert getattr (child_mapper , key ) == val
28
31
29
32
def test_get_submodel (self , mapper ):
30
33
url = self .get_mapper_url (mapper )
31
- expected_attr = {"id" : 1 , "name" : "first_model" }
34
+ expected_attr = {
35
+ "count" : 1 , "next" : None , "previous" : None ,
36
+ "results" : [{"id" : 1 , "name" : "first_model" }]
37
+ }
32
38
33
39
with requests_mock .Mocker () as m :
34
- m .register_uri ("get" , url , json = {"id" : 1 , })
40
+ m .register_uri (
41
+ "get" , url ,
42
+ json = {
43
+ "count" : 1 , "next" : None , "previous" : None ,
44
+ "results" : [{"id" : 1 }]
45
+ }
46
+ )
35
47
parent_mapper = next (mapper .get ())
36
- m .register_uri ("get" , url + "submodel/" , json = expected_attr )
48
+ m .register_uri (
49
+ "get" , url + "{}/{}/" .format (parent_mapper .id , "submodel" ),
50
+ json = expected_attr
51
+ )
37
52
submodel_mapper = next (parent_mapper .get ("submodel" ))
38
53
39
- for key , val in expected_attr .items ():
54
+ for key , val in expected_attr [ "results" ][ 0 ] .items ():
40
55
assert getattr (submodel_mapper , key ) == val
41
56
57
+ def test_get_query (self , mapper ):
58
+ url = self .get_mapper_url (mapper ) + "?name=test"
59
+ expected_attr = {
60
+ "count" : 1 , "next" : None , "previous" : None ,
61
+ "results" : [{"id" : 1 , "name" : "test" }]
62
+ }
63
+ with requests_mock .Mocker () as m :
64
+ m .register_uri ("get" , url , json = expected_attr )
65
+ next (mapper .get (name = "test" ))
66
+
42
67
def test_post (self , mapper ):
43
68
url = self .get_mapper_url (mapper )
44
69
@@ -53,7 +78,7 @@ def test_post(self, mapper):
53
78
54
79
def test_put (self , mapper ):
55
80
child_mapper = self .get_child_mapper (mapper )
56
- url = self .get_mapper_url (child_mapper )
81
+ url = self .get_mapper_url (child_mapper ) + "{}/" . format ( child_mapper . id )
57
82
with requests_mock .Mocker () as m :
58
83
received_req = m .register_uri (
59
84
"put" , url , json = self .update_or_create_resource_json_callback
@@ -70,7 +95,10 @@ def update_or_create_resource_json_callback(self, request, context):
70
95
71
96
def get_child_mapper (self , mapper ):
72
97
url = self .get_mapper_url (mapper )
73
- expected_attr = {"id" : 1 , "name" : "test" }
98
+ expected_attr = {
99
+ "count" : 1 , "next" : None , "previous" : None ,
100
+ "results" : [{"id" : 1 , "name" : "test" }]
101
+ }
74
102
with requests_mock .Mocker () as m :
75
103
m .register_uri ("get" , url , json = expected_attr )
76
104
child_mapper = next (mapper .get ())
@@ -93,11 +121,13 @@ def test_delete_from_child(self, mapper):
93
121
url = self .get_mapper_url (mapper ) + "1/"
94
122
95
123
with requests_mock .Mocker () as m :
96
- m .register_uri ("get" , url , json = {"id" : 1 , })
124
+ m .register_uri (
125
+ "get" , url , json = {"id" : 1 }
126
+ )
97
127
obj_mapper = next (mapper .get (1 ))
98
128
99
129
with requests_mock .Mocker () as m :
100
- m .register_uri ("delete" , url )
130
+ m .register_uri ("delete" , url + "1/" )
101
131
response = obj_mapper .delete ()
102
132
103
133
assert response .status_code == 200
@@ -106,7 +136,9 @@ def test_delete_other_id_from_child(self, mapper):
106
136
url = self .get_mapper_url (mapper ) + "1/"
107
137
108
138
with requests_mock .Mocker () as m :
109
- m .register_uri ("get" , url , json = {"id" : 1 , })
139
+ m .register_uri (
140
+ "get" , url , json = {"id" : 1 }
141
+ )
110
142
obj_mapper = next (mapper .get (1 ))
111
143
112
144
with pytest .raises (ForbiddenAsChildError ):
0 commit comments