libsecret/tests/mock-service-prompt.py

   1 #!/usr/bin/env python
   2
   3 import dbus
   4 import mock
   5 import sys
   6
   7 class ErrorPrompt(mock.SecretPrompt):
   8         def __init__(self, service, sender, prompt_name):
   9                 mock.SecretPrompt.__init__(self, service, sender, prompt_name)
  10
  11         @dbus.service.method('org.freedesktop.Secret.Prompt')
  12         def Prompt(self, window_id):
  13                 raise mock.NotSupported("This should cause prompting to fail")
  14
  15 class VanishPrompt(mock.SecretPrompt):
  16         def __init__(self, service, sender, prompt_name):
  17                 mock.SecretPrompt.__init__(self, service, sender, prompt_name)
  18
  19         @dbus.service.method('org.freedesktop.Secret.Prompt')
  20         def Prompt(self, window_id):
  21                 sys.exit(0)
  22
  23 class WindowPrompt(mock.SecretPrompt):
  24         def __init__(self, service, sender, prompt_name):
  25                 mock.SecretPrompt.__init__(self, service, sender, prompt_name)
  26
  27         @dbus.service.method('org.freedesktop.Secret.Prompt')
  28         def Prompt(self, window_id):
  29                 self.result = dbus.String(window_id, variant_level=1)
  30                 mock.SecretPrompt.Prompt(self, window_id)
  31
  32 service = mock.SecretService()
  33 service.add_standard_objects()
  34
  35 mock.SecretPrompt(service, None, "simple")
  36 mock.SecretPrompt(service, None, "delay", delay=0.1)
  37 def prompt_callback():
  38         return dbus.String("Special Result", variant_level=1)
  39 mock.SecretPrompt(service, None, "result", action=prompt_callback)
  40 ErrorPrompt(service, None, "error")
  41 VanishPrompt(service, None, "vanish")
  42 WindowPrompt(service, None, "window")
  43
  44 service.listen()