@@ -100,6 +100,7 @@ def testCreateAndGetMultiple(self):
100100
101101 rv = loads (r .data )
102102 self .assertEqual (len (rv ['replies' ]), 20 )
103+ self .assertEqual (rv ['total_replies' ], 20 )
103104
104105 def testCreateInvalidParent (self ):
105106
@@ -185,15 +186,18 @@ def testGetInvalid(self):
185186 self .assertEqual (self .get ('/?uri=%2Fpath%2F&id=123' ).status_code , 200 )
186187 data = loads (self .get ('/?uri=%2Fpath%2F&id=123' ).data )
187188 self .assertEqual (len (data ['replies' ]), 0 )
189+ self .assertEqual (data ['total_replies' ], 0 )
188190
189191 self .assertEqual (
190192 self .get ('/?uri=%2Fpath%2Fspam%2F&id=123' ).status_code , 200 )
191193 data = loads (self .get ('/?uri=%2Fpath%2Fspam%2F&id=123' ).data )
192194 self .assertEqual (len (data ['replies' ]), 0 )
195+ self .assertEqual (data ['total_replies' ], 0 )
193196
194197 self .assertEqual (self .get ('/?uri=?uri=%foo%2F' ).status_code , 200 )
195198 data = loads (self .get ('/?uri=?uri=%foo%2F' ).data )
196199 self .assertEqual (len (data ['replies' ]), 0 )
200+ self .assertEqual (data ['total_replies' ], 0 )
197201
198202 def testFetchEmpty (self ):
199203
@@ -214,6 +218,7 @@ def testGetLimited(self):
214218
215219 rv = loads (r .data )
216220 self .assertEqual (len (rv ['replies' ]), 10 )
221+ self .assertEqual (rv ['total_replies' ], 20 )
217222
218223 def testGetNested (self ):
219224
@@ -226,6 +231,7 @@ def testGetNested(self):
226231
227232 rv = loads (r .data )
228233 self .assertEqual (len (rv ['replies' ]), 1 )
234+ self .assertEqual (rv ['total_replies' ], 1 )
229235
230236 def testGetLimitedNested (self ):
231237
@@ -239,6 +245,7 @@ def testGetLimitedNested(self):
239245
240246 rv = loads (r .data )
241247 self .assertEqual (len (rv ['replies' ]), 10 )
248+ self .assertEqual (rv ['total_replies' ], 20 )
242249
243250 def testUpdate (self ):
244251
@@ -289,7 +296,7 @@ def testDeleteWithReference(self):
289296 self .assertIn ('/path/' , self .app .db .threads )
290297
291298 data = loads (client .get ("/?uri=%2Fpath%2F" ).data )
292- self .assertEqual (data ["total_replies" ], 1 )
299+ self .assertEqual (data ["total_replies" ], 2 )
293300
294301 self .assertEqual (self .get ('/?uri=%2Fpath%2F&id=1' ).status_code , 200 )
295302 self .assertEqual (self .get ('/?uri=%2Fpath%2F&id=2' ).status_code , 200 )
0 commit comments