test/branch_test.py

   1 #!/usr/bin/env python
   2 from __future__ import absolute_import, division, unicode_literals
   3 import unittest
   4 try:
   5     from unittest.mock import MagicMock
   6 except ImportError:
   7     from mock import MagicMock
   8
   9 from cola.compat import odict
  10 from cola.widgets.branch import BranchesTreeHelper, BranchTreeWidgetItem
  11
  12
  13 class BranchesTreeHelperTestCase(unittest.TestCase):
  14     """Tests the BranchesTreeHelper class."""
  15
  16     def test_should_return_a_valid_dict_on_group_branches(self):
  17         """Test the group_branches function."""
  18         branches = ['top_1/child_1/child_1_1',
  19                     'top_1/child_1/child_1_2',
  20                     'top_1/child_2/child_2_1/child_2_1_1',
  21                     'top_1/child_2/child_2_1/child_2_1_2']
  22         tree_helper = BranchesTreeHelper()
  23
  24         result = tree_helper.group_branches(branches, '/')
  25         inner_child = {'child_2_1_2': {}, 'child_2_1_1': {}}
  26         self.assertEqual(
  27             {'top_1': {
  28                 'child_1': {'child_1_2': {}, 'child_1_1': {}},
  29                 'child_2': {'child_2_1': inner_child}}}, result)
  30
  31     def test_should_create_a_valid_top_item_on_create_top_level_item(self):
  32         """Test the create_top_level_item function."""
  33         dict_children = odict()
  34         dict_children['child_1'] = {}
  35         dict_children['child_2'] = odict()
  36         dict_children['child_2']['child_2_1'] = {}
  37         dict_children['child_2']['child_2_2'] = {}
  38         tree_helper = BranchesTreeHelper()
  39
  40         result = tree_helper.create_top_level_item('top', dict_children)
  41         self.assertEqual('top', result.name)
  42         self.assertEqual(2, result.childCount())
  43         self.assertEqual('child_1', result.child(0).name)
  44         self.assertEqual('child_2', result.child(1).name)
  45         self.assertEqual(2, result.child(1).childCount())
  46         self.assertEqual('child_2_1', result.child(1).child(0).name)
  47         self.assertEqual('child_2_2', result.child(1).child(1).name)
  48
  49     def test_should_return_a_valid_top_item_on_get_root(self):
  50         """Test the get_root function."""
  51         items = _create_top_item()
  52         tree_helper = BranchesTreeHelper()
  53
  54         result = tree_helper.get_root(items['child_1'])
  55         self.assertTrue(items['top'] is result)
  56
  57         result = tree_helper.get_root(items['sub_child_2_1'])
  58         self.assertTrue(items['top'] is result)
  59
  60     def test_should_return_a_valid_branch_name_on_get_full_name(self):
  61         """Test the get_full_name function."""
  62         items = _create_top_item()
  63         tree_helper = BranchesTreeHelper()
  64
  65         result = tree_helper.get_full_name(items['child_1'], '/')
  66         self.assertEqual('child_1', result)
  67
  68         result = tree_helper.get_full_name(items['sub_child_2_2'], '/')
  69         self.assertEqual('child_2/sub_child_2_2', result)
  70
  71     def test_should_return_a_valid_child_on_find_child(self):
  72         """Test the find_child function."""
  73         items = _create_top_item()
  74         tree_helper = BranchesTreeHelper()
  75
  76         child = tree_helper.find_child(items['top'], 'child_1')
  77         self.assertEqual('child_1', child.name)
  78
  79         child = tree_helper.find_child(items['top'], 'child_2/sub_child_2_2')
  80         self.assertEqual('sub_child_2_2', child.name)
  81
  82     def test_should_return_empty_state_on_save_state(self):
  83         """Test the save_state function."""
  84         top = _create_item('top', False)
  85         tree_helper = BranchesTreeHelper()
  86
  87         result = tree_helper.save_state(top)
  88         self.assertEqual({'top': {}}, result)
  89
  90     def test_should_return_a_valid_state_on_save_state(self):
  91         """Test the save_state function."""
  92         items = _create_top_item()
  93         tree_helper = BranchesTreeHelper()
  94
  95         result = tree_helper.save_state(items['top'])
  96         self.assertEqual({'top': {'child_1': {}, 'child_2': {
  97             'sub_child_2_1': {}, 'sub_child_2_2': {}}}}, result)
  98
  99
 100 def _create_top_item():
 101     top = _create_item('top', True)
 102     child_1 = _create_item('child_1', False)
 103     child_2 = _create_item('child_2', True)
 104     sub_child_2_1 = _create_item('sub_child_2_1', False)
 105     sub_child_2_2 = _create_item('sub_child_2_2', False)
 106
 107     child_2.addChildren([sub_child_2_1, sub_child_2_2])
 108     top.addChildren([child_1, child_2])
 109
 110     return {'top': top, 'child_1': child_1, 'sub_child_2_1': sub_child_2_1,
 111             'sub_child_2_2': sub_child_2_2}
 112
 113
 114 def _create_item(name, expanded):
 115     item = BranchTreeWidgetItem(name)
 116     item.isExpanded = MagicMock(return_value=expanded)
 117     return item
 118
 119
 120 if __name__ == '__main__':
 121     unittest.main()