def it_can_iterate_over_the_serialized_parts(self):
# mockery ----------------------
partname, content_type, blob = ('part/name.xml', 'app/vnd.type',
'<Part_1/>')
spart = Mock(name='spart',
partname=partname,
content_type=content_type,
blob=blob)
pkg_reader = PackageReader(None, None, [spart])
iter_count = 0
# exercise ---------------------
for retval in pkg_reader.iter_sparts():
iter_count += 1
# verify -----------------------
assert retval == (partname, content_type, blob)
assert iter_count == 1
def it_can_iterate_over_all_the_srels(self):
# mockery ----------------------
pkg_srels = ['srel1', 'srel2']
sparts = [
Mock(name='spart1', partname='pn1', srels=['srel3', 'srel4']),
Mock(name='spart2', partname='pn2', srels=['srel5', 'srel6']),
]
pkg_reader = PackageReader(None, pkg_srels, sparts)
# exercise ---------------------
generated_tuples = [t for t in pkg_reader.iter_srels()]
# verify -----------------------
expected_tuples = [
('/', 'srel1'),
('/', 'srel2'),
('pn1', 'srel3'),
('pn1', 'srel4'),
('pn2', 'srel5'),
('pn2', 'srel6'),
]
assert generated_tuples == expected_tuples
def it_can_iterate_over_all_the_srels(self):
# mockery ----------------------
pkg_srels = ["srel1", "srel2"]
sparts = [
Mock(name="spart1", partname="pn1", srels=["srel3", "srel4"]),
Mock(name="spart2", partname="pn2", srels=["srel5", "srel6"]),
]
pkg_reader = PackageReader(None, pkg_srels, sparts)
# exercise ---------------------
generated_tuples = [t for t in pkg_reader.iter_srels()]
# verify -----------------------
expected_tuples = [
("/", "srel1"),
("/", "srel2"),
("pn1", "srel3"),
("pn1", "srel4"),
("pn2", "srel5"),
("pn2", "srel6"),
]
assert generated_tuples == expected_tuples
