libsecret/libsecret/test-py-lookup.py

76 lines
2.0 KiB
Python

#!/usr/bin/env python
#
# Copyright 2012 Red Hat Inc.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published
# by the Free Software Foundation; either version 2.1 of the licence or (at
# your option) any later version.
#
# See the included COPYING file for more information.
#
import unittest
import gi
gi.require_version('MockService', '0')
gi.require_version('Secret', '1')
from gi.repository import MockService as Mock
from gi.repository import Secret, GLib
STORE_SCHEMA = Secret.Schema.new("org.mock.Schema",
Secret.SchemaFlags.NONE,
{
"number": Secret.SchemaAttributeType.INTEGER,
"string": Secret.SchemaAttributeType.STRING,
"even": Secret.SchemaAttributeType.BOOLEAN,
}
)
class TestLookup(unittest.TestCase):
def setUp(self):
Mock.start("mock-service-normal.py")
def tearDown(self):
Secret.Service.disconnect()
Mock.stop()
def testSynchronous(self):
password = Secret.password_lookup_sync (STORE_SCHEMA, { "number": "1", "even": "false" }, None)
self.assertEqual("111", password)
def testSyncNotFound(self):
password = Secret.password_lookup_sync (STORE_SCHEMA, { "number": "5", "even": "true" }, None)
self.assertEqual(None, password)
def testAsynchronous(self):
loop = GLib.MainLoop(None)
def on_result_ready(source, result, unused):
loop.quit()
password = Secret.password_lookup_finish(result)
self.assertEquals("222", password)
Secret.password_lookup (STORE_SCHEMA, { "number": "2", "string": "two" },
None, on_result_ready, None)
loop.run()
def testAsyncNotFound(self):
loop = GLib.MainLoop(None)
def on_result_ready(source, result, unused):
loop.quit()
password = Secret.password_lookup_finish(result)
self.assertEquals(None, password)
Secret.password_lookup (STORE_SCHEMA, { "number": "7", "string": "five" },
None, on_result_ready, None)
loop.run()
if __name__ == '__main__':
unittest.main()