1 |
|
|
2 |
from gzz.mediaserver import util |
from gzz.mediaserver import util |
3 |
import md5 |
import md5 |
4 |
|
import tempfile |
5 |
|
import xreadlines |
6 |
|
|
7 |
testblockname='../Z/b_0000000008000000E817FEECDE00048B198A3B222DD5A647D9D98A0F9B38EFA74496C5C69E1FF4' |
blockname='../Z/b_0000000008000000E817FEECDE00048B198A3B222DD5A647D9D98A0F9B38EFA74496C5C69E1FF4' |
8 |
|
|
9 |
def testReadHeaders(): |
def testReadHeaders(): |
10 |
# """Tests reading headers from a mediaserver block file""" |
"""Tests reading headers from a mediaserver block file""" |
11 |
inFile = open(testblockname) |
inFile = open(blockname) |
12 |
headers = util.readHeaders(inFile) |
headers = util.readHeaders(inFile) |
13 |
failUnlessEqual(len(headers), 3) |
failUnlessEqual(len(headers), 3) |
14 |
failUnlessEqual(headers['Content-Type'], 'image/png') |
failUnlessEqual(headers['Content-Type'], 'image/png') |
16 |
failUnlessEqual(headers['Content-Transfer-Encoding'], 'binary') |
failUnlessEqual(headers['Content-Transfer-Encoding'], 'binary') |
17 |
|
|
18 |
def testStripBlock(): |
def testStripBlock(): |
19 |
# """Tests writing mediaserver block data into a normal file""" |
"""Tests writing mediaserver block data into a normal file""" |
20 |
tmpname = tempfile.mktemp() |
tmpname = tempfile.mktemp() |
21 |
util.stripBlock(testblockname, tmpname) |
util.stripBlock(blockname, tmpname) |
22 |
|
print 'Stripped' |
23 |
|
tmpfile = open(tmpname, 'rb') |
24 |
hash = md5.new() |
hash = md5.new() |
25 |
for line in tmpname.readlines(): |
for line in xreadlines.xreadlines(tmpfile): |
26 |
hash.update(line) |
hash.update(line) |
27 |
failUnlessEqual(hash.hexdigest(), "711be4dc69c602546b132c4fae3ab7f2") |
failUnlessEqual(hash.hexdigest(), "711be4dc69c602546b132c4fae3ab7f2") |