git.maemo.org Git - wifi-assistant/blob - package/test/unit/gui/edit_network_dialog_test.py

   1 from gnome import gconf
   2 from wifi_assistant.gui.edit_network_dialog import EditNetworkDialog
   3
   4 import unittest
   5 from unit.pie import *
   6
   7 ssid = 'Network Name'
   8 iter = 'Really a GTK TreeView iter'
   9
  10 class EditNetworkDialogTest(unittest.TestCase):
  11
  12     def setUp(self):
  13         self.parent_window = None
  14         self.save_callback = Mock()
  15
  16         #given(self.save_callback).save().willReturn(None)
  17         self.save_callback.replay()
  18
  19         self.testee = EditNetworkDialog(self.parent_window, self.save_callback.save)
  20
  21
  22     def test_clickingYesButtonCallsCallback(self):
  23         self._makeRunDialogReturn(EditNetworkDialog.YES_BUTTON)
  24         self.testee.show(iter, ssid)
  25         verify(self.save_callback).save(iter, ssid, True)
  26
  27
  28     def test_clickingNoButtonCallsCallback(self):
  29         self._makeRunDialogReturn(EditNetworkDialog.NO_BUTTON)
  30         self.testee.show(iter, ssid)
  31         verify(self.save_callback).save(iter, ssid, False)
  32
  33
  34     def test_cancellingDialogDoesntCallCallback(self):
  35         self._makeRunDialogReturn(-1)
  36         self.testee.show(iter, ssid)
  37         verify(self.save_callback, never()).save()
  38
  39
  40     def _makeRunDialogReturn(self, value):
  41         class method():
  42             def __init__(self, value_to_return):
  43                 self._value_to_return = value_to_return
  44             def __call__(self, x):
  45                 return self._value_to_return
  46         self.testee._runDialog = method(value)
  47
  48
  49 if __name__ == '__main__':
  50     unittest.main()