d81af828fb
updated the tests minor changes changes made verify capitalized country code changes made verify capitalized country code changes made stay consistent with country code implementation changes made to test with previous implementation updated the django-countries version to latest updated the tests input to match the output updated the constants to use updated country names according to new ISO standards
109 lines
3.5 KiB
Python
109 lines
3.5 KiB
Python
"""Unit tests for custom UserProfile properties."""
|
|
|
|
from __future__ import absolute_import
|
|
|
|
import datetime
|
|
|
|
import ddt
|
|
from django.core.cache import cache
|
|
|
|
from openedx.core.djangolib.testing.utils import CacheIsolationTestCase
|
|
from student.models import UserProfile
|
|
from student.tests.factories import UserFactory
|
|
|
|
|
|
@ddt.ddt
|
|
class UserProfilePropertiesTest(CacheIsolationTestCase):
|
|
"""Unit tests for age, gender_display, and level_of_education_display properties ."""
|
|
|
|
password = "test"
|
|
|
|
ENABLED_CACHES = ['default']
|
|
|
|
def setUp(self):
|
|
super(UserProfilePropertiesTest, self).setUp()
|
|
self.user = UserFactory.create(password=self.password)
|
|
self.profile = self.user.profile
|
|
|
|
def _set_year_of_birth(self, year_of_birth):
|
|
"""
|
|
Helper method that sets a birth year for the specified user.
|
|
"""
|
|
self.profile.year_of_birth = year_of_birth
|
|
self.profile.save()
|
|
|
|
def _set_level_of_education(self, level_of_education):
|
|
"""
|
|
Helper method that sets a level of education for the specified user.
|
|
"""
|
|
self.profile.level_of_education = level_of_education
|
|
self.profile.save()
|
|
|
|
def _set_gender(self, gender):
|
|
"""
|
|
Helper method that sets a gender for the specified user.
|
|
"""
|
|
self.profile.gender = gender
|
|
self.profile.save()
|
|
|
|
@ddt.data(0, 1, 13, 20, 100)
|
|
def test_age(self, years_ago):
|
|
"""Verify the age calculated correctly."""
|
|
current_year = datetime.datetime.now().year
|
|
self._set_year_of_birth(current_year - years_ago)
|
|
|
|
# In the year that your turn a certain age you will also have been a
|
|
# year younger than that in that same year. We calculate age based off of
|
|
# the youngest you could be that year.
|
|
age = years_ago - 1
|
|
self.assertEqual(self.profile.age, age)
|
|
|
|
def test_age_no_birth_year(self):
|
|
"""Verify nothing is returned."""
|
|
self.assertIsNone(self.profile.age)
|
|
|
|
@ddt.data(*UserProfile.LEVEL_OF_EDUCATION_CHOICES)
|
|
@ddt.unpack
|
|
def test_display_level_of_education(self, level_enum, display_level):
|
|
"""Verify the level of education is displayed correctly."""
|
|
self._set_level_of_education(level_enum)
|
|
|
|
self.assertEqual(self.profile.level_of_education_display, display_level)
|
|
|
|
def test_display_level_of_education_none_set(self):
|
|
"""Verify nothing is returned."""
|
|
self.assertIsNone(self.profile.level_of_education_display)
|
|
|
|
@ddt.data(*UserProfile.GENDER_CHOICES)
|
|
@ddt.unpack
|
|
def test_display_gender(self, gender_enum, display_gender):
|
|
"""Verify the gender displayed correctly."""
|
|
self._set_gender(gender_enum)
|
|
|
|
self.assertEqual(self.profile.gender_display, display_gender)
|
|
|
|
def test_display_gender_none_set(self):
|
|
"""Verify nothing is returned."""
|
|
self._set_gender(None)
|
|
|
|
self.assertIsNone(self.profile.gender_display)
|
|
|
|
def test_invalidate_cache_user_profile_country_updated(self):
|
|
|
|
country = 'US'
|
|
self.profile.country = country
|
|
self.profile.save()
|
|
|
|
cache_key = UserProfile.country_cache_key_name(self.user.id)
|
|
self.assertIsNone(cache.get(cache_key))
|
|
|
|
cache.set(cache_key, self.profile.country)
|
|
self.assertEqual(cache.get(cache_key), country)
|
|
|
|
country = 'bd'
|
|
self.profile.country = country
|
|
self.profile.save()
|
|
|
|
self.assertNotEqual(cache.get(cache_key), country)
|
|
self.assertIsNone(cache.get(cache_key))
|