import csv
from StringIO import StringIO

from django.core.urlresolvers import reverse

from courseware.tests.factories import InstructorFactory
from student.tests.factories import CourseEnrollmentFactory
from xmodule.modulestore.tests.django_utils import ModuleStoreTestCase
from xmodule.modulestore.tests.factories import CourseFactory, ItemFactory

from fun.tests.utils import skipUnlessLms

@skipUnlessLms
class BaseCourseDashboardTestCase(ModuleStoreTestCase):

    def setUp(self):
        super(BaseCourseDashboardTestCase, self).setUp(create_user=False)
        self.course = CourseFactory.create()
        self.problem_module = None
        self._generate_modules_tree(self.course, 'chapter', 'sequential',
                                    'vertical', 'problem')
        self.instructor = InstructorFactory.create(course_key=self.course.id)
        self.client.login(username=self.instructor.username, password="test")

    def enroll_student(self, course, **kwargs):
        return CourseEnrollmentFactory.create(
            course_id=self.get_course_id(course),
            **kwargs
        )

    def get_course_id(self, course):
        return course.id.to_deprecated_string()

    def get_response(self, url_name, course, response_format=None):
        return self.get_course_id_response(url_name, self.get_course_id(course), response_format=response_format)

    def get_course_id_response(self, url_name, course_id, response_format=None):
        url = reverse(url_name, kwargs={"course_id": course_id})
        if response_format is not None:
            url += "?format=" + response_format
        return self.client.get(url)

    def get_csv_response_rows(self, response):
        response_content = StringIO(response.content)
        response_content.seek(0)
        return [row for row in csv.reader(response_content)]

    def _generate_modules_tree(self, module, *args):
        if not args:
            self.problem_module = module
            return
        category = args[0]
        self._generate_modules_tree(ItemFactory(parent=module,
                                                category=category,
                                                display_name=None),
                                    *args[1:])

    # TODO merge this function with the _generate_modules_tree
    def _generate_modules_tree_with_display_names(self, module, *args):
        if not args:
            self.problem_module = module
            return
        category = args[0]
        self._generate_modules_tree_with_display_names(ItemFactory(parent=module,
                                                                   category=category,
                                                                   display_name=category),
                                                       *args[1:])