test/test_cola_settings.py

   1 import unittest
   2 import os
   3
   4 from cola import settings
   5 import helper
   6
   7 _tmp_path = helper.tmp_path('colarc')
   8 settings.SettingsModel.path = lambda *args: _tmp_path
   9
  10 class SettingsTestCase(unittest.TestCase):
  11     """Tests the cola.settings module"""
  12     def tearDown(self):
  13         if os.path.exists(_tmp_path):
  14             os.remove(_tmp_path)
  15
  16     def model(self):
  17         return settings.SettingsModel()
  18
  19     def test_path(self):
  20         """Test the test path() helper above"""
  21         model = self.model()
  22         self.assertEqual(model.path(), helper.tmp_path('colarc'))
  23
  24     def test_gui_save_restore(self):
  25         """Test saving and restoring gui state"""
  26         model = self.model()
  27         model.gui_state['test-gui'] = {'foo':'bar'}
  28         model.save()
  29         model = self.model()
  30         state = model.gui_state.get('test-gui', {})
  31         self.assertTrue('foo' in state)
  32         self.assertEqual(state['foo'], 'bar')
  33
  34     def test_bookmarks_save_restore(self):
  35         """Test the bookmark save/restore feature"""
  36         model = self.model()
  37         model.add_bookmark('test-bookmark')
  38         model.save()
  39         model = self.model()
  40         bookmarks = model.bookmarks
  41         self.assertEqual(len(bookmarks), 1)
  42         self.assertTrue('test-bookmark' in bookmarks)
  43
  44         model.remove_bookmark('test-bookmark')
  45         bookmarks = model.bookmarks
  46         self.assertEqual(len(bookmarks), 0)
  47         self.assertFalse('test-bookmark' in bookmarks)
  48
  49 if __name__ == '__main__':
  50     unittest.main()