密码策略模块测试

password_security/tests/test_password_security_session.py

@@ -0,0 +1,59 @@
+# Copyright 2016 LasLabs Inc.
+# License LGPL-3.0 or later (http://www.gnu.org/licenses/lgpl.html).
+
+from contextlib import contextmanager
+from unittest import mock
+
+from odoo.http import _request_stack
+from odoo.tests.common import TransactionCase
+
+from ..controllers import main
+
+IMPORT = "odoo.addons.password_security.controllers.main"
+
+
+class EndTestException(Exception):
+    """It allows for isolation of resources by raise"""
+
+
+class TestPasswordSecuritySession(TransactionCase):
+    def setUp(self):
+        super(TestPasswordSecuritySession, self).setUp()
+        self.PasswordSecuritySession = main.PasswordSecuritySession
+        self.password_security_session = self.PasswordSecuritySession()
+        self.passwd = "I am a password!"
+        self.fields = [
+            {"name": "new_password", "value": self.passwd},
+        ]
+        _request_stack.push(
+            mock.Mock(
+                env=self.env,
+            )
+        )
+        self.addCleanup(_request_stack.pop)
+
+    @contextmanager
+    def mock_assets(self):
+        """It mocks and returns assets used by this controller"""
+        with mock.patch("%s.request" % IMPORT) as request:
+            yield {
+                "request": request,
+            }
+
+    def test_change_password_check(self):
+        """It should check password on request user"""
+        with self.mock_assets() as assets:
+            check_password = assets["request"].env.user._check_password
+            check_password.side_effect = EndTestException
+            with self.assertRaises(EndTestException):
+                self.password_security_session.change_password(self.fields)
+            check_password.assert_called_once_with(
+                self.passwd,
+            )
+
+    def test_change_password_return(self):
+        """It should return result of super"""
+        with self.mock_assets():
+            with mock.patch.object(main.Session, "change_password") as chg:
+                res = self.password_security_session.change_password(self.fields)
+                self.assertEqual(chg(), res)