def test_get_serials(self):
'''
Test getting the all the serial numbers from a store
'''
expected = ['XYZABC', '123456']
mock = MagicMock(return_value='CertInfo\r\nSerial Number: XYZABC\r\nSerial Number: 123456\r\n')
with patch.dict(certutil.__salt__, {'cmd.run': mock}):
out = certutil.get_stored_cert_serials('TrustedPublisher')
mock.assert_called_once_with('certutil.exe -store TrustedPublisher')
self.assertEqual(expected, out)
def test_get_serials(self):
'''
Test getting all the serial numbers from a store
'''
expected = ['180720d39cd2db3244ba037417241e90',
'1768ac4e5b72bf1d0df0df118b34b959']
mock = MagicMock(return_value=(
'CertInfo\r\n'
'================ Certificate 0 ================\r\n'
'Serial Number: 180720d39cd2db3244ba037417241e90\r\n'
'OtherStuff\r\n'
'\r\n'
'================ Certificate 1 ================\r\n'
'Serial Number: 1768ac4e5b72bf1d0df0df118b34b959\r\n'
'OtherStuff'))
with patch.dict(certutil.__salt__, {'cmd.run': mock}):
out = certutil.get_stored_cert_serials('TrustedPublisher')
mock.assert_called_once_with(
'certutil.exe -store TrustedPublisher')
self.assertEqual(expected, out)
def test_get_serials():
"""
Test getting all the serial numbers from a store
"""
expected = [
"180720d39cd2db3244ba037417241e90",
"1768ac4e5b72bf1d0df0df118b34b959",
]
mock = MagicMock(
return_value=("CertInfo\r\n"
"================ Certificate 0 ================\r\n"
"Serial Number: 180720d39cd2db3244ba037417241e90\r\n"
"OtherStuff\r\n"
"\r\n"
"================ Certificate 1 ================\r\n"
"Serial Number: 1768ac4e5b72bf1d0df0df118b34b959\r\n"
"OtherStuff"))
with patch.dict(certutil.__salt__, {"cmd.run": mock}):
out = certutil.get_stored_cert_serials("TrustedPublisher")
mock.assert_called_once_with("certutil.exe -store TrustedPublisher")
assert expected == out
