def test_If_pic_is_not_in_cache_but_on_filesystem_provides_that(self):
import niprov.pictures
myImg = Mock()
myImg.provenance = {'id': '007'}
pictures = niprov.pictures.PictureCache(sentinel.dependencies)
pictures.keep(io.BytesIO('/x10/x05/x5f'), for_=myImg)
pictures.saveToDisk(for_=myImg)
niprov.pictures._CACHE = {} # reset cache
picfpath = os.path.expanduser('~/.niprov-snapshots/007.png')
self.assertEqual(picfpath, pictures.getFilepath(for_=myImg))
def test_If_pic_is_not_in_cache_but_on_filesystem_provides_that(self):
import niprov.pictures
myImg = Mock()
myImg.provenance = {'id':'007'}
pictures = niprov.pictures.PictureCache(sentinel.dependencies)
pictures.keep(io.BytesIO('/x10/x05/x5f'), for_=myImg)
pictures.saveToDisk(for_=myImg)
niprov.pictures._CACHE = {} # reset cache
picfpath = os.path.expanduser('~/.niprov-snapshots/007.png')
self.assertEqual(picfpath, pictures.getFilepath(for_=myImg))
def test_Can_be_told_to_persist_picture_to_disk_now(self):
from niprov.pictures import PictureCache
myImg = Mock()
myImg.provenance = {'id': '007'}
pictures = PictureCache(sentinel.dependencies)
pictures.saveToDisk(for_=myImg) #shouldn't do anything
pictures.keep(io.BytesIO('/x10/x05/x5f'), for_=myImg)
pictures.saveToDisk(for_=myImg)
picfpath = os.path.expanduser('~/.niprov-snapshots/007.png')
self.assertTrue(os.path.isfile(picfpath))
with open(picfpath) as picFile:
self.assertEqual(picFile.read(), '/x10/x05/x5f')
def test_Can_be_told_to_persist_picture_to_disk_now(self):
from niprov.pictures import PictureCache
myImg = Mock()
myImg.provenance = {'id':'007'}
pictures = PictureCache(sentinel.dependencies)
pictures.saveToDisk(for_=myImg) #shouldn't do anything
pictures.keep(io.BytesIO('/x10/x05/x5f'), for_=myImg)
pictures.saveToDisk(for_=myImg)
picfpath = os.path.expanduser('~/.niprov-snapshots/007.png')
self.assertTrue(os.path.isfile(picfpath))
with open(picfpath) as picFile:
self.assertEqual(picFile.read(), '/x10/x05/x5f')
