def test_06_get(self, mock_readall):
"""Test06 KeyStore().get()."""
url = u'https://xxx'
email = u'user@domain'
auth = u'xxx'
credentials = {url: {u'email': email, u'auth': auth}}
mock_readall.return_value = credentials
kstore = KeyStore("filename")
self.assertTrue(kstore.get(url))
self.assertFalse(kstore.get("NOT EXISTING ENTRY"))
def test_03__read_all(self, mock_jload):
"""Test03 KeyStore()._read_all()."""
url = u'https://xxx'
email = u'user@domain'
auth = u'xxx'
credentials = {url: {u'email': email, u'auth': auth}}
mock_jload.return_value = credentials
with patch(BUILTINS + '.open', mock_open()):
kstore = KeyStore("filename")
status = kstore._read_all()
self.assertEqual(status, credentials)
def test_05__write_all(self, mock_umask, mock_jdump, mock_verks):
"""Test05 KeyStore()._write_all()."""
url = u'https://xxx'
email = u'user@domain'
auth = u'xxx'
credentials = {url: {u'email': email, u'auth': auth}}
mock_umask.return_value = 0o77
mock_jdump.side_effect = IOError('json dump')
mock_verks.return_value = None
with patch(BUILTINS + '.open', mock_open()):
kstore = KeyStore("filename")
status = kstore._write_all(credentials)
self.assertEqual(status, 1)
def test_08_delete(self, mock_readall, mock_writeall, mock_shred,
mock_verks):
"""Test08 KeyStore().delete()."""
url = u'https://xxx'
email = u'user@domain'
auth = u'xxx'
credentials = {url: {u'email': email, u'auth': auth}}
mock_readall.return_value = credentials
mock_writeall.return_value = 0
mock_shred.return_value = 0
mock_verks.return_value = None
kstore = KeyStore("filename")
status = kstore.delete(url)
mock_writeall.assert_called_once_with({})
self.assertEqual(status, 0)
def test_09_erase(self, mock_shred, mock_verks, mock_unlink):
"""Test09 KeyStore().erase()."""
mock_verks.return_value = None
mock_shred.return_value = None
kstore = KeyStore("filename")
self.assertEqual(kstore.erase(), 0)
mock_unlink.assert_called_once_with("filename")
mock_verks.return_value = None
mock_shred.return_value = None
mock_unlink.side_effect = IOError
kstore = KeyStore("filename")
self.assertEqual(kstore.erase(), 1)
def test_04__shred(self, mock_size, mock_verks):
"""Test04 KeyStore()._shred()."""
mock_verks.return_value = None
with patch(BUILTINS + '.open', mock_open()):
kstore = KeyStore("filename")
status = kstore._shred()
self.assertEqual(status, 0)
mock_size.return_value = 123
mock_verks.return_value = None
with patch(BUILTINS + '.open', mock_open()):
kstore = KeyStore("filename")
status = kstore._shred()
self.assertEqual(status, 0)
def test_07_put(self, mock_readall, mock_writeall, mock_shred):
"""Test07 KeyStore().put()."""
url = u'https://xxx'
email = u'user@domain'
auth = u'xxx'
credentials = {url: {u'email': email, u'auth': auth}}
mock_shred.return_value = None
kstore = KeyStore("filename")
status = kstore.put("", "", "")
self.assertEqual(status, 1)
mock_shred.return_value = None
mock_readall.return_value = dict()
kstore = KeyStore("filename")
status = kstore.put(url, auth, email)
mock_writeall.assert_called_once_with(credentials)
def test_01_init(self):
"""Test01 KeyStore() constructor."""
kstore = KeyStore("filename")
status = kstore.keystore_file
self.assertEqual(status, "filename")