From 7e28ee31f156bfa5b027d004af47805a51a3b55b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Petr=20Dlouh=C3=BD?= <petr.dlouhy@email.cz>
Date: Thu, 14 Jul 2022 11:56:27 +0200
Subject: [PATCH] add custom identity test

---
 tests/unit/test_utils.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/tests/unit/test_utils.py b/tests/unit/test_utils.py
index 8a9bd13..816acda 100644
--- a/tests/unit/test_utils.py
+++ b/tests/unit/test_utils.py
@@ -46,6 +46,19 @@ class GetIdentityTestCase(TestCase):
         get_id = get_identity(Context({}), user=MyUser(identity='fake_id'))
         assert get_id == 'fake_id'
 
+    def test_custom_identity_specific_provider(self):
+        get_id = get_identity(Context({
+            'foo_provider_identity': 'bar',
+            'analytical_identity': 'baz',
+        }), prefix='foo_provider')
+        assert get_id == 'bar'
+
+    def test_custom_identity_general(self):
+        get_id = get_identity(Context({
+            'analytical_identity': 'baz',
+        }), prefix='foo_provider')
+        assert get_id == 'baz'
+
 
 @override_settings(ANALYTICAL_DOMAIN="example.org")
 class GetDomainTestCase(TestCase):