From 02bef78f7f5a7eb102341c2152d7898829ed6f52 Mon Sep 17 00:00:00 2001 From: zhengxie Date: Fri, 4 Dec 2015 11:25:26 +0800 Subject: [PATCH] [tests] Add user options tests --- tests/seahub/options/test_models.py | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) create mode 100644 tests/seahub/options/test_models.py diff --git a/tests/seahub/options/test_models.py b/tests/seahub/options/test_models.py new file mode 100644 index 0000000000..b3c25e80e5 --- /dev/null +++ b/tests/seahub/options/test_models.py @@ -0,0 +1,27 @@ +from seahub.test_utils import BaseTestCase +from seahub.options.models import (UserOptions, KEY_USER_GUIDE, + VAL_USER_GUIDE_ON, VAL_USER_GUIDE_OFF) + +class UserOptionsManagerTest(BaseTestCase): + def test_is_user_guide_enabled(self): + assert UserOptions.objects.is_user_guide_enabled(self.user.email) is True + + UserOptions.objects.create(email=self.user.email, + option_key=KEY_USER_GUIDE, + option_val=VAL_USER_GUIDE_OFF) + + assert UserOptions.objects.is_user_guide_enabled(self.user.email) is False + + def test_is_user_guide_enabled_with_multiple_records(self): + UserOptions.objects.create(email=self.user.email, + option_key=KEY_USER_GUIDE, + option_val=VAL_USER_GUIDE_OFF) + UserOptions.objects.create(email=self.user.email, + option_key=KEY_USER_GUIDE, + option_val=VAL_USER_GUIDE_ON) + + assert len(UserOptions.objects.filter(email=self.user.email, + option_key=KEY_USER_GUIDE)) == 2 + assert UserOptions.objects.is_user_guide_enabled(self.user.email) is True + assert len(UserOptions.objects.filter(email=self.user.email, + option_key=KEY_USER_GUIDE)) == 1