コード例 #1
0
ファイル: __init__.py プロジェクト: clayg/swift
        def getheaders(self):
            etag = self.etag
            if not etag:
                if isinstance(self.body, str):
                    etag = '"' + md5(self.body).hexdigest() + '"'
                else:
                    etag = '"68b329da9893e34099c7d8ad5cb9c940"'

            headers = HeaderKeyDict({
                'content-length': len(self.body),
                'content-type': 'x-application/test',
                'x-timestamp': self.timestamp,
                'x-backend-timestamp': self.timestamp,
                'last-modified': self.timestamp,
                'x-object-meta-test': 'testing',
                'x-delete-at': '9876543210',
                'etag': etag,
                'x-works': 'yes',
            })
            if self.status // 100 == 2:
                headers['x-account-container-count'] = \
                    kwargs.get('count', 12345)
            if not self.timestamp:
                # when timestamp is None, HeaderKeyDict raises KeyError
                headers.pop('x-timestamp', None)
            try:
                if next(container_ts_iter) is False:
                    headers['x-container-timestamp'] = '1'
            except StopIteration:
                pass
            am_slow, value = self.get_slow()
            if am_slow:
                headers['content-length'] = '4'
            headers.update(self.headers)
            return headers.items()
コード例 #2
0
        def getheaders(self):
            etag = self.etag
            if not etag:
                if isinstance(self.body, bytes):
                    etag = '"' + md5(self.body).hexdigest() + '"'
                else:
                    etag = '"68b329da9893e34099c7d8ad5cb9c940"'

            am_slow, _junk = self.get_slow()
            headers = HeaderKeyDict({
                'content-length': len(self.body),
                'content-type': 'x-application/test',
                'x-timestamp': self.timestamp,
                'x-backend-timestamp': self.timestamp,
                'last-modified': self.timestamp,
                'x-object-meta-test': 'testing',
                'x-delete-at': '9876543210',
                'etag': etag,
                'x-works': 'yes',
            })
            if self.status // 100 == 2:
                headers['x-account-container-count'] = \
                    kwargs.get('count', 12345)
            if not self.timestamp:
                # when timestamp is None, HeaderKeyDict raises KeyError
                headers.pop('x-timestamp', None)
            try:
                if next(container_ts_iter) is False:
                    headers['x-container-timestamp'] = '1'
            except StopIteration:
                pass
            headers.update(self.headers)
            return headers.items()
コード例 #3
0
ファイル: test_direct_client.py プロジェクト: goulon/swift
class FakeConn(object):

    def __init__(self, status, headers=None, body='', **kwargs):
        self.status = status
        try:
            self.reason = RESPONSE_REASONS[self.status][0]
        except Exception:
            self.reason = 'Fake'
        self.body = body
        self.resp_headers = HeaderKeyDict()
        if headers:
            self.resp_headers.update(headers)
        self.etag = None

    def _update_raw_call_args(self, *args, **kwargs):
        capture_attrs = ('host', 'port', 'method', 'path', 'req_headers',
                         'query_string')
        for attr, value in zip(capture_attrs, args[:len(capture_attrs)]):
            setattr(self, attr, value)
        return self

    def getresponse(self):
        if self.etag:
            self.resp_headers['etag'] = str(self.etag.hexdigest())
        if isinstance(self.status, Exception):
            raise self.status
        return self

    def getheader(self, header, default=None):
        return self.resp_headers.get(header, default)

    def getheaders(self):
        return self.resp_headers.items()

    def read(self, amt=None):
        if isinstance(self.body, six.StringIO):
            return self.body.read(amt)
        elif amt is None:
            return self.body
        else:
            return Exception('Not a StringIO entry')

    def send(self, data):
        if not self.etag:
            self.etag = md5()
        self.etag.update(data)
コード例 #4
0
        def getheaders(self):
            etag = self.etag
            if not etag:
                if isinstance(self.body, str):
                    etag = '"' + md5(self.body).hexdigest() + '"'
                else:
                    etag = '"68b329da9893e34099c7d8ad5cb9c940"'

            headers = HeaderKeyDict(
                {
                    "content-length": len(self.body),
                    "content-type": "x-application/test",
                    "x-timestamp": self.timestamp,
                    "x-backend-timestamp": self.timestamp,
                    "last-modified": self.timestamp,
                    "x-object-meta-test": "testing",
                    "x-delete-at": "9876543210",
                    "etag": etag,
                    "x-works": "yes",
                }
            )
            if self.status // 100 == 2:
                headers["x-account-container-count"] = kwargs.get("count", 12345)
            if not self.timestamp:
                # when timestamp is None, HeaderKeyDict raises KeyError
                headers.pop("x-timestamp", None)
            try:
                if next(container_ts_iter) is False:
                    headers["x-container-timestamp"] = "1"
            except StopIteration:
                pass
            am_slow, value = self.get_slow()
            if am_slow:
                headers["content-length"] = "4"
            headers.update(self.headers)
            return headers.items()
コード例 #5
0
 def test_update(self):
     headers = HeaderKeyDict()
     headers.update({'Content-Length': '0'})
     headers.update([('Content-Type', 'text/plain')])
     self.assertEqual(headers['Content-Length'], '0')
     self.assertEqual(headers['Content-Type'], 'text/plain')
コード例 #6
0
 def test_update(self):
     headers = HeaderKeyDict()
     headers.update({'Content-Length': '0'})
     headers.update([('Content-Type', 'text/plain')])
     self.assertEqual(headers['Content-Length'], '0')
     self.assertEqual(headers['Content-Type'], 'text/plain')
コード例 #7
0
 def test_update(self):
     headers = HeaderKeyDict()
     headers.update({"Content-Length": "0"})
     headers.update([("Content-Type", "text/plain")])
     self.assertEqual(headers["Content-Length"], "0")
     self.assertEqual(headers["Content-Type"], "text/plain")