def testLoginFailure(self, mock_ucs, mock_request): mock_ucs.return_value.login.return_value = False mock_ucs.return_value.cookie = EXPECTED_COOKIE mock_request.headers = MOCK_HEADER cookie = controler.login_get() mock_ucs.assert_called_with(HOST, USER, PASS, secure=False) mock_ucs.return_value.login.assert_called_once() self.assertIsNone(cookie, 'returned cookie does not equal None')
def testLoginSuccess(self, mock_ucs, mock_request): mock_ucs.return_value.login.return_value = True mock_ucs.return_value.cookie = EXPECTED_COOKIE mock_request.headers = MOCK_HEADER cookie = controler.login_get() mock_ucs.assert_called_with(HOST, USER, PASS, secure=False) mock_ucs.return_value.login.assert_called_once() self.assertEqual(cookie, EXPECTED_COOKIE, 'returned cookie %s does not equal %s' % (cookie, EXPECTED_COOKIE))