from django.test import TestCase
from flashcards.models import User, Section


class UserTests(TestCase):
    def setUp(self):
        User.objects.create(email="none@none.com", password="1234")
        Section.objects.create(department='dept',
                               course_num='101a',
                               course_title='how 2 test',
                               instructor='George Lucas',
                               quarter='SP15')

    def test_section_list(self):
        section = Section.objects.get(course_num='101a')
        user = User.objects.get(email="none@none.com")
        self.assertNotIn(section, user.sections.all())
        user.sections.add(section)
        self.assertIn(section, user.sections.all())
        user.sections.add(section)
        self.assertEqual(user.sections.count(), 1)