"""
Enterprise support admin tests.
"""

import csv
import os
import tempfile

from django.contrib.messages import get_messages
from django.test import Client
from django.urls import reverse

from openedx.core.djangoapps.catalog.tests.factories import CourseRunFactory
from openedx.features.enterprise_support.admin.forms import CSVImportForm
from common.djangoapps.student.models import CourseEnrollment, CourseEnrollmentAttribute
from common.djangoapps.student.tests.factories import TEST_PASSWORD, AdminFactory, CourseEnrollmentFactory, UserFactory
from xmodule.modulestore.tests.django_utils import ModuleStoreTestCase  # lint-amnesty, pylint: disable=wrong-import-order


class EnrollmentAttributeOverrideViewTest(ModuleStoreTestCase):
    """
    Tests for course creator admin.
    """

    def setUp(self):
        """ Test case setup """
        super().setUp()

        self.client = Client()
        user = AdminFactory()
        self.view_url = reverse('admin:enterprise_override_attributes')
        self.client.login(username=user.username, password=TEST_PASSWORD)

        self.users = []
        for _ in range(3):
            self.users.append(UserFactory())

        self.course = CourseRunFactory()
        self.course_id = self.course.get('key')  # lint-amnesty, pylint: disable=no-member
        self.csv_data = [
            [self.users[0].id, self.course_id, 'OP_4321'],
            [self.users[1].id, self.course_id, 'OP_8765'],
            [self.users[2].id, self.course_id, 'OP_2109'],
        ]
        self.csv_data_for_existing_attributes = [
            [self.users[0].id, self.course_id, 'OP_1234'],
            [self.users[1].id, self.course_id, 'OP_5678'],
            [self.users[2].id, self.course_id, 'OP_9012'],
        ]

        for user in self.users:
            CourseEnrollmentFactory(
                course_id=self.course_id,
                user=user
            )

    def create_csv(self, header=None, data=None):
        """Create csv"""
        header = header or ['lms_user_id', 'course_id', 'opportunity_id']
        data = data or self.csv_data
        tmp_csv_path = os.path.join(tempfile.gettempdir(), 'data.csv')
        with open(tmp_csv_path, 'w') as csv_file:
            csv_writer = csv.writer(csv_file)
            csv_writer.writerow(header)
            csv_writer.writerows(data)

        return tmp_csv_path

    def verify_enrollment_attributes(self, data=None):
        """
        Verify that data from csv is imported correctly and tables have correct data.
        """
        data = data or self.csv_data
        for user_id, course_id, opportunity_id in data:
            enrollment = CourseEnrollment.objects.get(user_id=user_id, course_id=course_id)
            enrollment_attribute = CourseEnrollmentAttribute.objects.get(
                enrollment=enrollment,
                namespace='salesforce',
                name='opportunity_id'
            )
            assert enrollment_attribute.value == opportunity_id

    def test_get(self):
        """
        Tests that HTTP GET is working as expected.
        """
        response = self.client.get(self.view_url)
        assert response.status_code == 200
        assert isinstance(response.context['form'], CSVImportForm)

    def test_post(self):
        """
        Tests that HTTP POST is working as expected when creating new attributes and updating.
        """
        csv_path = self.create_csv()
        with open(csv_path) as csv_file:
            post_data = {'csv_file': csv_file}
            response = self.client.post(self.view_url, data=post_data)
            assert response.status_code == 302
        self.verify_enrollment_attributes()

        # override existing
        csv_path = self.create_csv(data=self.csv_data_for_existing_attributes)
        with open(csv_path) as csv_file:
            post_data = {'csv_file': csv_file}
            response = self.client.post(self.view_url, data=post_data)
            assert response.status_code == 302
        self.verify_enrollment_attributes(data=self.csv_data_for_existing_attributes)

    def test_post_with_no_csv(self):
        """
        Tests that HTTP POST without out csv file is working as expected.
        """
        response = self.client.post(self.view_url)
        assert response.context['form'].errors == {'csv_file': ['This field is required.']}

    def test_post_with_incorrect_csv_header(self):
        """
        Tests that HTTP POST with incorrect csv header is working as expected.
        """
        csv_path = self.create_csv(header=['a', 'b'])
        post_data = {'csv_file': open(csv_path)}
        response = self.client.post(self.view_url, data=post_data)
        assert response.context['form'].errors == {
            'csv_file': [
                'Expected a CSV file with [lms_user_id, course_id, opportunity_id] '
                'columns, but found [a, b] columns instead.'
            ]
        }

    def test_post_with_no_enrollment_error(self):
        """
        Tests that HTTP POST is working as expected when for some records there is no enrollment.
        """
        csv_data = self.csv_data + [[999, self.course_id, 'NOPE'], [1000, self.course_id, 'NONE']]
        csv_path = self.create_csv(data=csv_data)
        post_data = {'csv_file': open(csv_path)}
        response = self.client.post(self.view_url, data=post_data)
        assert response.status_code == 302
        messages = []
        for msg in get_messages(response.wsgi_request):
            messages.append(str(msg))
        assert messages == [
            'Successfully updated learner enrollment opportunity ids.',
            'Enrollment attributes were not updated for records at following line numbers '
            'in csv because no enrollment found for these records: 4, 5'
        ]