github.com/ismailbayram/bigpicture@v0.0.0-20231225173155-e4b21f5efcff/internal/browser/pyproject/base/test.py (about)

     1  from model_mommy import mommy
     2  
     3  from users.enums import GroupType
     4  from users.service import UserService
     5  
     6  
     7  class BaseTestViewMixin:
     8      def init_users(self):
     9          service = UserService()
    10          self.superuser = mommy.make('users.User', is_staff=True)
    11          self.superuser_token = service._create_token(self.superuser)
    12  
    13          data = {
    14              "first_name": "Customer 1",
    15              "last_name": "CusLast",
    16              "phone_number": "555111",
    17              "group_type": GroupType.customer
    18          }
    19          self.customer, self.customer_token = service.get_or_create_user(**data)
    20          self.customer_profile = self.customer.customer_profile
    21  
    22          data = {
    23              "first_name": "Customer 2",
    24              "last_name": "CusLast 2",
    25              "phone_number": "5551112",
    26              "group_type": GroupType.customer
    27          }
    28          self.customer2, self.customer2_token = service.get_or_create_user(**data)
    29          self.customer2_profile = self.customer2.customer_profile
    30  
    31          data = {
    32              "first_name": "Worker 1",
    33              "last_name": "WorkLast",
    34              "phone_number": "555222",
    35              "group_type": GroupType.worker
    36          }
    37          self.worker, self.worker_token = service.get_or_create_user(**data)
    38          self.worker_profile = self.worker.worker_profile
    39  
    40          data = {
    41              "first_name": "Worker 2",
    42              "last_name": "WorkLast",
    43              "phone_number": "555223",
    44              "group_type": GroupType.worker
    45          }
    46          self.worker2, self.worker2_token = service.get_or_create_user(**data)
    47          self.worker2_profile = self.worker2.worker_profile
    48  
    49          data = {
    50              "first_name": "Washer 1",
    51              "last_name": "WashLast",
    52              "phone_number": "555333",
    53              "group_type": GroupType.washer
    54          }
    55          self.washer, self.washer_token = service.get_or_create_user(**data)
    56          self.washer_profile = self.washer.washer_profile
    57  
    58          data = {
    59              "first_name": "Washer 2",
    60              "last_name": "WashLast",
    61              "phone_number": "555334",
    62              "group_type": GroupType.washer
    63          }
    64          self.washer2, self.washer2_token = service.get_or_create_user(**data)
    65          self.washer2_profile = self.washer2.washer_profile