diff options
author | Duda Dornelles <ddornell@thoughtworks.com> | 2015-02-13 11:56:58 -0200 |
---|---|---|
committer | Duda Dornelles <ddornell@thoughtworks.com> | 2015-02-13 12:47:12 -0200 |
commit | 9af1553353e8fb837e5c4323531dda8e69dc8915 (patch) | |
tree | 6e096392337506f47c84e92e20ee80a2606c38e8 /service/test/unit/resources/__init__.py | |
parent | c52ce25ac49cdfdc1791bcb65c5262aec63dadbd (diff) |
Making sure that no private key can be retrieved by the KeysResource
Diffstat (limited to 'service/test/unit/resources/__init__.py')
-rw-r--r-- | service/test/unit/resources/__init__.py | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/service/test/unit/resources/__init__.py b/service/test/unit/resources/__init__.py index e69de29b..b8214a8c 100644 --- a/service/test/unit/resources/__init__.py +++ b/service/test/unit/resources/__init__.py @@ -0,0 +1,23 @@ +from twisted.internet.defer import succeed +from twisted.web import server +from twisted.web.server import Site + + +def resolve_result(request, result): + if isinstance(result, str): + request.write(result) + request.finish() + return succeed(request) + elif result is server.NOT_DONE_YET: + if request.finished: + return succeed(request) + else: + return request.notifyFinish().addCallback(lambda _: request) + else: + raise ValueError("Unexpected return value: %r" % (result,)) + + +class DummySite(Site): + def get(self, request): + return resolve_result(request, self.getResourceFor(request).render(request)) + |