def test_verify_simple(self):
document = etree.XML('<Response>foo doc</Response>')
fake_etree = fudge.Fake('etree')
fake_etree.remember_order()
to_string = fake_etree.expects('tostring')
to_string.with_args(document)
to_string.returns('<Response>foo doc</Response>')
fake_tempfile = fudge.Fake('tempfile')
fake_tempfile.remember_order()
named_xmlfile = fake_tempfile.expects(
'NamedTemporaryFile'
)
named_xmlfile.with_args(delete=False)
xmlfile = named_xmlfile.returns_fake()
xmlfile.remember_order()
enter = xmlfile.expects('__enter__')
enter.with_arg_count(0)
enter.returns(xmlfile)
write = xmlfile.expects('write')
write.with_args('<Response>foo doc</Response>')
seek = xmlfile.expects('seek')
seek.with_args(0)
exit = xmlfile.expects('__exit__')
exit.with_args(None, None,None)
xmlfile.has_attr(name='xmlfile')
named_certfile = fake_tempfile.next_call(
'NamedTemporaryFile'
)
named_certfile.with_args(delete=False)
certfile = named_certfile.returns_fake()
certfile.remember_order()
enter = certfile.expects('__enter__')
enter.with_arg_count(0)
enter.returns(certfile)
write = certfile.expects('write')
write.with_args(
('-----BEGIN CERTIFICATE-----\nfoo signature\n'
+ '-----END CERTIFICATE-----'
)
)
seek = certfile.expects('seek')
seek.with_args(0)
exit = certfile.expects('__exit__')
exit.with_args(None, None,None)
certfile.has_attr(name='certfile')
fake_subprocess = fudge.Fake('subprocess')
fake_subprocess.remember_order()
popen = fake_subprocess.expects('Popen')
fake_subprocess.has_attr(PIPE=1)
popen.with_args(
[
'xmlsec1',
'--verify',
'--pubkey-cert-pem',
'certfile',
'--id-attr:ID',
'urn:oasis:names:tc:SAML:2.0:assertion:Assertion',
'xmlfile',
],
stderr=1,
stdout=1,
)
proc = popen.returns_fake()
proc.remember_order()
wait = proc.expects('wait')
wait.with_arg_count(0)
stderr = StringIO('OK')
proc.has_attr(stderr=stderr)
fake_os = fudge.Fake('os')
fake_os.remember_order()
remove = fake_os.expects('remove')
remove.with_args('certfile')
remove = fake_os.next_call('remove')
remove.with_args('xmlfile')
SignatureVerifier.verify(
document,
'foo signature',
_etree=fake_etree,
_tempfile=fake_tempfile,
_subprocess=fake_subprocess,
_os=fake_os,
)
