def test_setting_default_notice_settings_for_a_user(self): """ Test class method for saving default settings for a user. """ NoticeSetting.set_user_default_notice_settings(self.user,0) obj_from_db_for_user = NoticeSetting.objects.filter(user = self.user) self.assertEqual(len(obj_from_db_for_user),2)
def test_setting_default_notice_settings_correct_notices(self): """ Test class method for saving default settings for a user. Make sure right notices have been saved. """ NoticeSetting.set_user_default_notice_settings(self.user,0) obj_from_db_for_user = NoticeSetting.objects.filter(user = self.user).order_by('id') for notice_setting in obj_from_db_for_user: self.assertEqual(notice_setting.user,self.user) self.assertEqual(notice_setting.send,True)