small change and test to get_user_from_username

This commit is contained in:
Mouse Reeve
2020-12-30 13:59:51 -08:00
parent 747167c579
commit 48f002727a
2 changed files with 16 additions and 4 deletions

View File

@ -23,6 +23,18 @@ class Views(TestCase):
connector_file='self_connector',
local=True
)
self.local_user = models.User.objects.create_user(
'mouse', 'mouse@mouse.mouse', 'password', local=True)
def test_get_user_from_username(self):
''' works for either localname or username '''
self.assertEqual(
views.get_user_from_username('mouse'), self.local_user)
self.assertEqual(
views.get_user_from_username('mouse@%s' % DOMAIN), self.local_user)
with self.assertRaises(models.User.DoesNotExist):
views.get_user_from_username('mojfse@example.com')
def test_search_json_response(self):