libsecret/tests/mock-service-prompt.py

   1 #!/usr/bin/env python
   2
   3 #
   4 # Copyright 2012 Red Hat Inc.
   5 #
   6 # This program is free software: you can redistribute it and/or modify
   7 # it under the terms of the GNU Lesser General Public License as published
   8 # by the Free Software Foundation; either version 2.1 of the licence or (at
   9 # your option) any later version.
  10 #
  11 # See the included COPYING file for more information.
  12 #
  13
  14 import dbus
  15 import mock
  16 import sys
  17
  18 class ErrorPrompt(mock.SecretPrompt):
  19         def __init__(self, service, sender, prompt_name):
  20                 mock.SecretPrompt.__init__(self, service, sender, prompt_name)
  21
  22         @dbus.service.method('org.freedesktop.Secret.Prompt')
  23         def Prompt(self, window_id):
  24                 raise mock.NotSupported("This should cause prompting to fail")
  25
  26 class VanishPrompt(mock.SecretPrompt):
  27         def __init__(self, service, sender, prompt_name):
  28                 mock.SecretPrompt.__init__(self, service, sender, prompt_name)
  29
  30         @dbus.service.method('org.freedesktop.Secret.Prompt')
  31         def Prompt(self, window_id):
  32                 sys.exit(0)
  33
  34 class WindowPrompt(mock.SecretPrompt):
  35         def __init__(self, service, sender, prompt_name):
  36                 mock.SecretPrompt.__init__(self, service, sender, prompt_name)
  37
  38         @dbus.service.method('org.freedesktop.Secret.Prompt')
  39         def Prompt(self, window_id):
  40                 self.result = dbus.String(window_id, variant_level=1)
  41                 mock.SecretPrompt.Prompt(self, window_id)
  42
  43 service = mock.SecretService()
  44 service.add_standard_objects()
  45
  46 mock.SecretPrompt(service, None, "simple")
  47 mock.SecretPrompt(service, None, "delay", delay=0.1)
  48 def prompt_callback():
  49         return dbus.String("Special Result", variant_level=1)
  50 mock.SecretPrompt(service, None, "result", action=prompt_callback)
  51 ErrorPrompt(service, None, "error")
  52 VanishPrompt(service, None, "vanish")
  53 WindowPrompt(service, None, "window")
  54
  55 service.listen()