def setUp(self): self.task = mock.MagicMock() self.retry_after = 25 self.task.info = {'done': 25, 'total': 100, 'retry-after': 25} self.username = '******' self.response = AsyncRestoreResponse(self.task, self.username)
class TestAsyncRestoreResponse(TestXmlMixin, SimpleTestCase): def setUp(self): self.task = mock.MagicMock() self.retry_after = 25 self.task.info = {'done': 25, 'total': 100, 'retry-after': 25} self.username = '******' self.response = AsyncRestoreResponse(self.task, self.username) def test_response(self): expected = """ <OpenRosaResponse xmlns="http://openrosa.org/http/response"> <message nature='ota_restore_pending'>Asynchronous restore under way for {username}</message> <Sync xmlns="http://commcarehq.org/sync"> <progress total="{total}" done="{done}" retry-after="{retry_after}"/> </Sync> </OpenRosaResponse> """.format( username=self.username, total=self.task.info['total'], done=self.task.info['done'], retry_after=self.retry_after, ) self.assertXmlEqual(self.response.compile_response(), expected) def test_html_response(self): http_response = self.response.get_http_response() self.assertEqual(http_response.status_code, 202) self.assertTrue(http_response.has_header('Retry-After')) self.assertEqual(http_response['retry-after'], str(self.retry_after)) self.assertXmlEqual( list(http_response.streaming_content)[0], self.response.compile_response())
class TestAsyncRestoreResponse(TestXmlMixin, SimpleTestCase): def setUp(self): self.task = mock.MagicMock() self.retry_after = 25 self.task.info = {'done': 25, 'total': 100, 'retry-after': 25} self.username = '******' self.response = AsyncRestoreResponse(self.task, self.username) def test_response(self): expected = """ <OpenRosaResponse xmlns="http://openrosa.org/http/response"> <message nature='ota_restore_pending'>Asynchronous restore under way for {username}</message> <Sync xmlns="http://commcarehq.org/sync"> <progress total="{total}" done="{done}" retry-after="{retry_after}"/> </Sync> </OpenRosaResponse> """.format( username=self.username, total=self.task.info['total'], done=self.task.info['done'], retry_after=self.retry_after, ) self.assertXmlEqual(self.response.compile_response(), expected) def test_html_response(self): http_response = self.response.get_http_response() self.assertEqual(http_response.status_code, 202) self.assertTrue(http_response.has_header('Retry-After')) self.assertEqual(http_response['retry-after'], str(self.retry_after)) self.assertXmlEqual(list(http_response.streaming_content)[0], self.response.compile_response())