def test_iterable_media(self): media_ids = ['abc', 'cdef', 'ghj'] media = [ mock.MagicMock(spec=object, id=media_id) for media_id in media_ids ] result = utils.get_ids(media) self.assertListEqual(result, media_ids)
def test_iterable_ids(self): media = ['123', '221', '311'] result = utils.get_ids(media) self.assertListEqual(result, media)
def test_media(self): media_id = 'abc' media = mock.MagicMock(spec=object, id=media_id) result = utils.get_ids(media) self.assertListEqual(result, [media_id])
def test_id_str(self): media = '123' result = utils.get_ids(media) self.assertListEqual(result, [media])
def remove(self, media): """ Remove media from a group. Media can be a single ID, media instance, or list of IDs or media instances. """ ids = utils.get_ids(media) return self._client.request('DELETE', self.model.list_url, data={'ids': ids})
def add(self, media): """ Add media to a group. Media can be a single ID, media instance, or list of IDs or media instances. """ ids = utils.get_ids(media) return self._client.request('PUT', self.model.list_url, data={'ids': ids})