Revision 03314970

b/kamaki/clients/test/image.py
183 183

  
184 184
    def tearDown(self):
185 185
        self.FR.json = example_images
186
        self.FR.status_code = 200
186 187

  
187 188
    def assert_dicts_are_deeply_equal(self, d1, d2):
188 189
        for k, v in d1.items():
......
287 288
            for k, v in props.items():
288 289
                self.assertEquals(a_headers['X-Image-Meta-Property-%s' % k], v)
289 290

  
290
    """
291 291
    def test_set_members(self):
292
        ""Test set_members""
293
        self._prepare_img()
294
        self._test_set_members()
295

  
296
    def _test_set_members(self):
297
        members = ['%s@fake.net' % self.now]
298
        for img in self._imglist.values():
299
            self.client.set_members(img['id'], members)
300
            r = self.client.list_members(img['id'])
301
            self.assertEqual(r[0]['member_id'], members[0])
292
        img0 = example_images_detailed[0]
293
        members = ['use3r-1d-0', 'us2r-1d-1', 'us3r-1d-2']
294
        with patch.object(
295
                self.C,
296
                'perform_request',
297
                return_value=self.FR()) as perform_req:
298
            self.assertRaises(
299
                ClientError,
300
                self.client.set_members,
301
                img0['id'], members)
302
            self.FR.status_code = 204
303
            self.client.set_members(img0['id'], members)
304
            (method, data, a_headers, a_params) = perform_req.call_args[0]
305
            from json import loads
306
            memberships = loads(data)['memberships']
307
            for membership in memberships:
308
                self.assertTrue(membership['member_id'] in members)
302 309

  
310
    """
303 311
    def test_list_members(self):
304 312
        ""Test list_members""
305 313
        self._test_list_members()

Also available in: Unified diff