diff options
-rw-r--r-- | askbot/tests/category_tree_tests.py | 6 | ||||
-rw-r--r-- | askbot/utils/category_tree.py | 2 |
2 files changed, 8 insertions, 0 deletions
diff --git a/askbot/tests/category_tree_tests.py b/askbot/tests/category_tree_tests.py index 54af4e79..39362bd2 100644 --- a/askbot/tests/category_tree_tests.py +++ b/askbot/tests/category_tree_tests.py @@ -126,3 +126,9 @@ class CategoryTreeTests(unittest.TestCase): 'buy', 'download' ]) ) + + def test_get_leaf_names_empty(self): + self.assertEqual( + set([]), + ct.get_leaf_names(None) + ) diff --git a/askbot/utils/category_tree.py b/askbot/utils/category_tree.py index 049315dc..4406bdfd 100644 --- a/askbot/utils/category_tree.py +++ b/askbot/utils/category_tree.py @@ -76,6 +76,8 @@ def _get_leaf_names(subtree): def get_leaf_names(tree = None): """returns set of leaf names""" data = tree or get_data() + if data is None: + return set() return _get_leaf_names(data[0][1]) def path_is_valid(tree, path): |