def test_init(self, monkeypatch):
media = Media(Media.IMAGE,
'foo',
'binary',
path='/path/to/ble/',
info='lorem ipsum')
assert media.path == '/path/to/ble/'
assert media.content == 'binary'
media = Media(Media.IMAGE, 'foo', 'binary')
assert media.path != None
assert media.info == None
assert media.content == 'binary'
def test_save_content(self, monkeypatch):
media = Media(Media.IMAGE,
'foo',
'binary',
path='/path/to/ble/',
info='lorem ipsum')
# TODO: How test that this works?
assert media.save_content() == None
def raise_ex():
raise KeyError
monkeypatch.setattr(__builtin__, 'open', lambda x, y: raise_ex())
media = Media(Media.IMAGE,
'foo',
'binary',
path='/path/to/ble/',
info='lorem ipsum')
assert media.save_content() == None
def test_type(self):
media = Media(Media.VIDEO,
'foo',
'binary',
path='/path/to/ble/',
info='lorem ipsum')
assert media.is_video()
media = Media(Media.IMAGE,
'foo',
'binary',
path='/path/to/ble/',
info='lorem ipsum')
assert media.is_image()
media = Media(Media.MAP,
'foo',
'binary',
path='/path/to/ble/',
info='lorem ipsum')
assert media.is_map()
