from gzz.mediaserver import util import md5 testblockname='../Z/b_0000000008000000E817FEECDE00048B198A3B222DD5A647D9D98A0F9B38EFA74496C5C69E1FF4' def testReadHeaders(): # """Tests reading headers from a mediaserver block file""" inFile = open(testblockname) headers = util.readHeaders(inFile) failUnlessEqual(len(headers), 3) failUnlessEqual(headers['Content-Type'], 'image/png') failUnlessEqual(headers['X-Injected-By'], 'antkaij') failUnlessEqual(headers['Content-Transfer-Encoding'], 'binary') def testStripBlock(): # """Tests writing mediaserver block data into a normal file""" tmpname = tempfile.mktemp() util.stripBlock(testblockname, tmpname) hash = md5.new() for line in tmpname.readlines(): hash.update(line) failUnlessEqual(hash.hexdigest(), "711be4dc69c602546b132c4fae3ab7f2")