Synnefo » ./kamaki

    @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))