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