def test_04_convert_to_json_returns_string_of_hierarchy(self):
mock_dp = Mock(spec=DependencyGraph)
mock_dp.hierarchy = Tree()
mock_dp.hierarchy.create_node('level_one', 'level_one')
mock_dp.hierarchy.create_node('level_two_one', 'level_two_one', parent='level_one')
mock_dp.hierarchy.create_node('level_two_two', 'level_two_two', parent='level_one')
mock_dp.hierarchy.create_node('level_three', 'level_three', parent='level_two_one')
jsonf = JsonFormatter(mock_dp)
self.assertEqual(jsonf.convert_hierarchy_to_json(), mock_dp.hierarchy.to_json(), 'JSON not formatted correclty')
def test_03_raises_error_if_non_tree_graph_passed_to_convert_to_json(self):
mock_dp = Mock(spec=DependencyGraph)
mock_dp.hierarchy = {'test': 'test'}
jsonf = JsonFormatter(mock_dp)
with self.assertRaises(TypeError):
jsonf.convert_hierarchy_to_json()
