Add: account views, tests, override user model

logs_collector/collector/api/tests/test_views.py

@@ -1,12 +1,13 @@
 from pathlib import Path
 from django.core.files.base import ContentFile
 from django.conf import settings
-from django.contrib.auth.models import User
 from django.urls import reverse
 
 from rest_framework import status
 from rest_framework.test import APITestCase
 
+from account.models import User
+
 from collector.models import Archive, Platform, Ticket
 
 

logs_collector/collector/models.py

@@ -2,10 +2,11 @@ import uuid
 import hashlib
 
 from django.core.validators import MaxValueValidator, MinValueValidator
-from django.contrib.auth.models import User
 from django.db import models
 from django.urls import reverse
 
+from account.models import User
+
 from .utils.helpers import logs_dir_path
 
 

logs_collector/collector/tests/test_models.py

@@ -1,9 +1,9 @@
 from pathlib import Path
 from django.test import TestCase
-from django.contrib.auth.models import User
 from django.core.files.base import ContentFile
 from django.conf import settings
 
+from account.models import User
 from collector.models import Platform, Ticket, Archive
 
 

logs_collector/collector/tests/test_urls.py

@@ -1,6 +1,7 @@
 from django.test import TestCase
 from django.urls import resolve, reverse
-from django.contrib.auth.models import User
+
+from account.models import User
 
 from collector import views
 from collector.models import Ticket, Platform

logs_collector/collector/tests/test_views.py

@@ -1,6 +1,7 @@
 from django.test import TestCase, Client
 from django.urls import reverse
-from django.contrib.auth.models import User
+
+from account.models import User
 
 from collector.models import Ticket, Platform