def do_test(self, size, status_code, exp):
status = FakeStatus(status_code)
if size:
self.request_data['size'] = [size]
yield self.request.test_render(pngstatus.PngStatusResource(status))
self.assertPngFile(self.request.written, exp)
def testReturnedUnknownPngOnUnkwnownBuilder(self):
status = FakeStatus()
self.request = FakeRequest({'builder': [':)'], 'size': 'normal'})
self.request.uri = '/png'
self.request.method = 'GET'
yield self.request.test_render(pngstatus.PngStatusResource(status))
self.assertPngFile(self.request.written, 'unknown_normal')
def testReturnedSuccesPngOnSuccesBuild(self):
status = FakeStatus()
yield self.request.test_render(pngstatus.PngStatusResource(status))
png_file = open(
'%s/../../status/web/files/success_normal.png' % (
os.path.dirname(__file__),), 'rb'
)
png = png_file.read()
png_file.close()
self.assertEqual(self.request.written, png)
def testReturnedSuccesPngOnWheteverStatusBuild(self):
i = random.randint(0, 5)
status = FakeStatus()
FakeBuilder.status_code = i
yield self.request.test_render(pngstatus.PngStatusResource(status))
png_file = open(
'%s/../../status/web/files/%s_normal.png' % (
os.path.dirname(__file__), results.Results[i]), 'rb'
)
png = png_file.read()
png_file.close()
self.assertEqual(self.request.written, png)
def testReturnedUnknownPngOnunkwnownBuilder(self):
status = FakeStatus()
self.request = FakeRequest({'builder': ':)', 'size': 'normal'})
self.request.uri = '/png'
self.request.method = 'GET'
yield self.request.test_render(pngstatus.PngStatusResource(status))
png_file = open(
'%s/../../status/web/files/unknown_normal.png' % (
os.path.dirname(__file__),), 'rb'
)
png = png_file.read()
png_file.close()
self.assertEqual(self.request.written, png)
def testReturnedWheteverStatusAndWhateverSize(self):
sizes = [['small'], ['normal'], ['large']]
i = random.randint(0, 5)
x = sizes[random.randint(0, 2)]
status = FakeStatus()
FakeBuilder.status_code = i
request_data = self.request_data.copy()
request_data['size'] = x
self.request = FakeRequest(request_data)
self.request.uri = '/png'
self.request.method = 'GET'
yield self.request.test_render(pngstatus.PngStatusResource(status))
png_file = open(
'%s/../../status/web/files/%s_%s.png' % (
os.path.dirname(__file__), results.Results[i], x[0]), 'rb'
)
png = png_file.read()
png_file.close()
self.assertEqual(self.request.written, png)
def testReturnedSuccesPngOnSuccesBuild(self):
status = FakeStatus()
yield self.request.test_render(pngstatus.PngStatusResource(status))
self.assertPngFile(self.request.written, 'success_normal')
