Synnefo » ./kamaki

    def update_image(

            self, image_id,

            name=None, disk_format=None, container_format=None,

            status=None, public=None, owner_id=None, **properties):

        path = path4url('images', image_id)

        if name is not None:

            self.set_header('X-Image-Meta-Name', name)

        if disk_format is not None:

            self.set_header('X-Image-Meta-Disk-Format', disk_format)

        if container_format is not None:

            self.set_header('X-Image-Meta-Container-Format', container_format)

        if status is not None:

            self.set_header('X-Image-Meta-Status', status)

        if public is not None:

            self.set_header('X-Image-Meta-Is-Public', bool(public))

        if owner_id is not None:

            self.set_header('X-Image-Meta-Owner', owner_id)

        for k, v in properties.items():

            self.set_header('X-Image-Meta-Property-%s' % k, v)

        r = self.put(path, success=200)

        return r.headers

    @patch('%s.put' % image_pkg, return_value=FR())

    @patch('%s.set_header' % image_pkg)

    def test_update_image(self, set_header, put):

        FR.headers = 'some headers'

        hcnt = 0

        for args in product(

                ('some id', 'other id'),

                ('image name', None), ('disk fmt', None), ('cnt format', None),

                ('status', None), (True, False, None), ('owner id', None),

                (dict(k1='v1', k2='v2'), {})):

            r = self.client.update_image(*args[:-1], **args[-1])

            (image_id, name, disk_format, container_format,

            status, public, owner_id, properties) = args

            self.assertEqual(r, FR.headers)

            header_calls = []

            prf = 'X-Image-Meta-'

            if name:

                header_calls.append(call('%sName' % prf, name))

            if disk_format:

                header_calls.append(call('%sDisk-Format' % prf, disk_format))

            if container_format:

                header_calls.append(

                    call('%sContainer-Format' % prf, container_format))

            if status:

                header_calls.append(call('%sStatus' % prf, status))

            if public is not None:

                header_calls.append(call('%sIs-Public' % prf, public))

            if owner_id:

                header_calls.append(call('%sOwner' % prf, owner_id))

            for k, v in properties.items():

                header_calls.append(call('%sProperty-%s' % (prf, k), v))

            self.assertEqual(

                sorted(set_header.mock_calls[hcnt:]), sorted(header_calls))

            hcnt = len(set_header.mock_calls)

            self.assertEqual(

                put.mock_calls[-1], call('/images/%s' % image_id, success=200))