twisted/web/test/_util.py

   1
   2 # Copyright (c) Twisted Matrix Laboratories.
   3 # See LICENSE for details.
   4
   5 """
   6 General helpers for L{twisted.web} unit tests.
   7 """
   8
   9 from twisted.internet.defer import succeed
  10 from twisted.web import server
  11 from twisted.trial.unittest import TestCase
  12 from twisted.python.failure import Failure
  13 from twisted.web._flatten import flattenString
  14 from twisted.web.error import FlattenerError
  15
  16
  17 def _render(resource, request):
  18     result = resource.render(request)
  19     if isinstance(result, str):
  20         request.write(result)
  21         request.finish()
  22         return succeed(None)
  23     elif result is server.NOT_DONE_YET:
  24         if request.finished:
  25             return succeed(None)
  26         else:
  27             return request.notifyFinish()
  28     else:
  29         raise ValueError("Unexpected return value: %r" % (result,))
  30
  31
  32
  33 class FlattenTestCase(TestCase):
  34     """
  35     A test case that assists with testing L{twisted.web._flatten}.
  36     """
  37     def assertFlattensTo(self, root, target):
  38         """
  39         Assert that a root element, when flattened, is equal to a string.
  40         """
  41         d = flattenString(None, root)
  42         d.addCallback(lambda s: self.assertEqual(s, target))
  43         return d
  44
  45
  46     def assertFlattensImmediately(self, root, target):
  47         """
  48         Assert that a root element, when flattened, is equal to a string, and
  49         performs no asynchronus Deferred anything.
  50
  51         This version is more convenient in tests which wish to make multiple
  52         assertions about flattening, since it can be called multiple times
  53         without having to add multiple callbacks.
  54         """
  55         results = []
  56         it = self.assertFlattensTo(root, target)
  57         it.addBoth(results.append)
  58         # Do our best to clean it up if something goes wrong.
  59         self.addCleanup(it.cancel)
  60         if not results:
  61             self.fail("Rendering did not complete immediately.")
  62         result = results[0]
  63         if isinstance(result, Failure):
  64             result.raiseException()
  65
  66
  67     def assertFlatteningRaises(self, root, exn):
  68         """
  69         Assert flattening a root element raises a particular exception.
  70         """
  71         d = self.assertFailure(self.assertFlattensTo(root, ''), FlattenerError)
  72         d.addCallback(lambda exc: self.assertIsInstance(exc._exception, exn))
  73         return d
  74
  75
  76
  77