tests/test_interface.py

   1 # -*- Mode: Python -*-
   2
   3 from __future__ import absolute_import
   4
   5 import unittest
   6
   7 from gi.repository import GObject
   8 import testhelper
   9
  10
  11 GUnknown = GObject.type_from_name("TestUnknown")
  12 Unknown = GUnknown.pytype
  13
  14
  15 class MyUnknown(Unknown, testhelper.Interface):
  16     some_property = GObject.Property(type=str)
  17
  18     def __init__(self):
  19         Unknown.__init__(self)
  20         self.called = False
  21
  22     def do_iface_method(self):
  23         self.called = True
  24         Unknown.do_iface_method(self)
  25
  26
  27 GObject.type_register(MyUnknown)
  28
  29
  30 class MyObject(GObject.GObject, testhelper.Interface):
  31     some_property = GObject.Property(type=str)
  32
  33     def __init__(self):
  34         GObject.GObject.__init__(self)
  35         self.called = False
  36
  37     def do_iface_method(self):
  38         self.called = True
  39
  40
  41 GObject.type_register(MyObject)
  42
  43
  44 class TestIfaceImpl(unittest.TestCase):
  45
  46     def test_reimplement_interface(self):
  47         m = MyUnknown()
  48         m.iface_method()
  49         self.assertEqual(m.called, True)
  50
  51     def test_implement_interface(self):
  52         m = MyObject()
  53         m.iface_method()
  54         self.assertEqual(m.called, True)