added custom fields to user model

apps/auth_zero/tests.py

@@ -1,12 +1,14 @@
-from django.test import TestCase
 from django.contrib.auth import get_user_model
-from common.config import ConfigFileLoader, CredentialsLoader
+from django.test import TestCase
+
+from apps.common.config import ConfigFileLoader
+from apps.common.config import CredentialsLoader
 
 
 class UserTestCase(TestCase):
     def setUp(self):
         self.user_model = get_user_model()
-        config_file = "apps/auth_zero/config/test_config.ini"
+        config_file = "apps/auth_zero/config/data/test/test_config.ini"
         self.creds_obj = CredentialsLoader(config_file, ConfigFileLoader)
         self.users = {
             verification_status: self.user_model.objects.create_user(

common/config.py → apps/common/config.py

@@ -1,20 +1,47 @@
+import abc
 from configparser import ConfigParser
 from typing import TypeVar
+
 from django.utils.encoding import force_text
 
 
+class InterfaceConfigLoader(metaclass=abc.ABCMeta):
+    """Enforce abstract interface for class capability."""
+
+    impl_methods = ("read_config",)
+
+    @classmethod
+    def __subclasshook__(cls, subclass):
+        return all(map(cls.get_implementation, cls.impl_methods)) or NotImplemented
+
+    @staticmethod
+    def get_implementation(subclass, attr_name: str) -> bool:
+        """Fetch an implemented method."""
+        return hasattr(subclass, attr_name)
+
+    @abc.abstractmethod
+    def read_config(self, verification_status: str) -> dict:
+        raise NotImplementedError
+
+
+IConfigLoader_T = TypeVar("IConfigLoader_T", bound=InterfaceConfigLoader)
+
+
 class CredentialsLoader:
     """Loads credentials from config file using custom loader class"""
 
-    def __init__(self, config_file: str, loader_class: TypeVar("ConfigFileLoader")):
+    def __init__(
+        self, config_file: str, loader_class: TypeVar("ConfigFileLoader")
+    ) -> IConfigLoader_T:
         """Initialises custom loader class."""
         self.loader_class = loader_class(config_file)
 
-    def read_config(self, verification_status: str) -> dict:
-        return self.loader_class.read_config(verification_status)
+    @property
+    def loader_class(self):
+        return self.loader_class
 
 
-class ConfigFileLoader:
+class ConfigFileLoader(InterfaceConfigLoader):
     """Loads data from .ini file"""
 
     def __init__(self, config_file: str):