'Test that you can specify which courses you want to dump.'

@mock.patch(u'openedx.core.djangoapps.coursegraph.tasks.NodeSelector')
@mock.patch(u'openedx.core.djangoapps.coursegraph.tasks.Graph')
@ddt.data(1, 2)
def test_dump_specific_courses(self, number_of_courses, mock_graph_class, mock_selector_class):

    mock_graph = self.setup_mock_graph(mock_selector_class, mock_graph_class)
    call_command(u'dump_to_neo4j', courses=self.course_strings[:number_of_courses], host=u'mock_host', http_port=7474, user=u'mock_user', password=u'mock_password')
    self.assertCourseDump(mock_graph, number_of_courses=number_of_courses, number_commits=number_of_courses, number_rollbacks=0)
