fixed password update not working

core/tests.py

@@ -26,12 +26,8 @@ class CustomUserViewSetTest(APITestCase):
         self.endpoint = '/api/v1/users/'
         self.factory = factories.CustomUserFactory
         self.model = models.CustomUser
-        # create admin user
-        self.admin_email = f"admin_user@mail.com"
-        self.password = ''.join(random.choices(string.ascii_uppercase, k = 10))
-        self.user = self.factory(email=self.admin_email, password=self.password, is_active=True)
         # create regular user
-        self.reg_email = f"regular_user@mail.com"
+        self.reg_email = f"user@mail.com"
         self.password = ''.join(random.choices(string.ascii_uppercase, k = 10))
         self.user = self.factory(email=self.reg_email, password=self.password, is_active=True)
 
@@ -129,7 +125,20 @@ class CustomUserViewSetTest(APITestCase):
         response = self.client.put(url, data=data, format='json')
 
         # Assert forbidden code
-        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
+
+        # assert new password hash properly updated
+        # assert fields exist, and data matches
+        updated_user = self.model.objects.get(pk=self.user.id)
+        stored_value = updated_user.__dict__['password']
+        hash_type, iteration, salt, stored_password_hash = stored_value.split('$')
+        new_password_hash = hashlib.pbkdf2_hmac(
+                hash_name='sha256',
+                password=data['password'].encode(),
+                salt=salt.encode(),
+                iterations=int(iteration),
+                )
+        self.assertEqual(stored_password_hash, base64.b64encode(new_password_hash).decode())
 
     def test_regular_user_cannot_modify_existing_instance(self):
         """Regular user cannot modify existing instance