def test_should_return_user_for_connection(self):
user, created = User.objects.get_or_create(username='******', email='*****@*****.**')
user.set_password('secret')
user.save()
contact, created = Contact.objects.get_or_create(name='shaggy', user=user)
contact.save()
backend, created = Backend.objects.get_or_create(name='scoobydoo')
backend.save()
connection1, created = Connection.objects.get_or_create(identity='0794339344', backend=backend)
connection1.contact = contact
connection1.save()
connection_profile = ConnectionProfile.lookup_by_connection(connection1)
self.assertEqual(user, connection_profile.user)
def test_should_return_none_if_no_user_for_connection(self):
self.assertIsNone(ConnectionProfile.lookup_by_connection(Connection()))