Ejemplo n.º 1
0
 def decode(self, response, request):
     log.msg("Building credentials")
     auth = self._parseAuth(response)
     try:
         creds = StoopidCredential(auth["name"], auth["secret"])
     except KeyError, ke:
         raise LoginFailed("{0!r} not in authorization.".format(*ke.args))
Ejemplo n.º 2
0
 def cb_assertFileNotEmpty(response, path):
     fileSize = os.stat(path).st_size
     log.msg("%s is %dB: %r" % (path, fileSize,
         open(path).read()))
     self.assertTrue(fileSize > 500,
             "Expected more than %d bytes" % fileSize)
     return response
Ejemplo n.º 3
0
    def render_POST(self, request):
        contentType = request.getHeader("Content-type")
        if contentType == "application/json":
            content = json.load(request.content)
            resourceName = content["resource"]

            if resourceName not in self.parent.children:
                url = self.buildUrl(request, resourceName)

                log.msg("%r: Creating resource: %s" % (request, url))
                resource = _GettableResource()
                self.parent.putChild(resourceName, resource)
                log.msg("%r: Created resource: %s" % (request, url))

                request.setResponseCode(http.CREATED)
                request.setHeader("Location", url)

            else:
                #log.msg("Resource already exists: %s" % resourceName)
                request.setResponseCode(http.CONFLICT)
        else:
            request.setResponseCode(http.BAD_REQUEST)

        #log.msg("%r: finished" % request)
        return ""
Ejemplo n.º 4
0
    def _parseAuth(self, response):
        def unQuote(s):
            if s and (s[0] in "\"\'") and (s[0] == s[-1]):
                s = s[1:-1]
                return s

        auth = dict()
        log.msg("Parsing response: {0!r}".format(response))
        for segment in response.replace("\n", " ").split(","):
            log.msg("Parsing segment: {0!r}".format(segment))
            key, val = [s.strip() for s in segment.split("=", 1)]
            auth[key] = unQuote(val)

        return auth
Ejemplo n.º 5
0
 def _openPage(self, url):
     log.msg("Opening url: %r" % url)
     return urlopen(url).read()
Ejemplo n.º 6
0
 def _login(self, creds):
     log.msg("Logging in: {0!r}".format(creds))
     return HTTPAuthSessionWrapper._login(self, creds)
Ejemplo n.º 7
0
 def requestAvatarId(self, cred):
     log.msg("Requesting avatar id: {0}".format(cred))
     if cred.secret != self.secret:
         raise UnauthorizedLogin("Unauthorized login.")
     return cred.name
Ejemplo n.º 8
0
    def test_post(self):
        resource = "naboo"
        createdUrl = self.generatorUrl + resource
        postEntity = json.dumps({"resource": resource})
        headers = {"Content-type":"application/json"}

        # Trying to GET a resource before we generate it should fail
        try:
            yield self.getPage(createdUrl, method="GET")
        except WebError, we:
            self.assertEquals(http.NOT_FOUND, we.status)
        else:
            self.fail("Fetched %s before it was created" % createdUrl)

        # Generating the new resource should work
        log.msg("POSTing to generate content.")
        result = yield self.getPage(self.generatorUrl, method="POST",
                headers=headers, data=postEntity)
        log.msg("POSTed to generate content.")
        self.assertEquals(http.CREATED, result.status)
        self.assertHeader(result, "Location", createdUrl)

        # Now we should be able to GET the generated resource
        result = yield self.getPage(createdUrl, method="GET")
        self.assertTrue(result.status in OKAY_CODES)

        # Trying to generate the same resource again should fail
        try:
            yield self.getPage(self.generatorUrl, method="POST",
                    headers=headers, data=postEntity)