fixing login with 2fa (#638)

Author: janet-barbie

Diff for: src/pages/LoginWith2fa.tsx

@@ -43,12 +43,12 @@ export const LOGIN_WITH_2FA = gql`
   mutation LoginWithTwoFactorAuthentication(
     $email: String!
     $otp: String!
-    $TwoWayVerificationToken: String!
+   
   ) {
     loginWithTwoFactorAuthentication(
       email: $email
       otp: $otp
-      TwoWayVerificationToken: $TwoWayVerificationToken
+
     ) {
       token
       user {
@@ -85,7 +85,7 @@ const TwoFactorPage: React.FC = () => {
 
   const location = useLocation();
   const navigate = useNavigate();
-  const { email, TwoWayVerificationToken } = location.state || {};
+  const { email } = location.state || {};
   useEffect(() => {
     // Update document class and localStorage when theme changes
     if (isDark) {
@@ -98,30 +98,30 @@ const TwoFactorPage: React.FC = () => {
   }, [isDark]);
 
   useEffect(() => {
-    if (!email || !TwoWayVerificationToken) {
+    if (!email ) {
       navigate('/login');
     }
-  }, [email, TwoWayVerificationToken, navigate]);
+  }, [email,  navigate]);
 
   const [loginWithTwoFactorAuthentication] = useMutation<LoginResponse>(
     LOGIN_WITH_2FA,
     {
       onCompleted: async (data) => {
         const response = data.loginWithTwoFactorAuthentication;
         try {
-          localStorage.setItem('authToken', response.token);
+          //localStorage.setItem('authToken', response.token);
           localStorage.setItem('user', JSON.stringify(response.user));
           await login(response);
           await client.resetStore();
           toast.success(response.message);
 
           const rolePaths: Record<string, string> = {
-            superAdmin: '/dashboard',
+            superAdmin: '/organizations',
             admin: '/trainees',
             coordinator: '/trainees',
             manager: '/dashboard',
             ttl: '/ttl-trainees',
-            trainee: '/dashboard',
+            trainee: '/performance',
           };
 
           const redirectPath = rolePaths[response.user.role] || '/dashboard';
@@ -151,7 +151,7 @@ const TwoFactorPage: React.FC = () => {
         variables: {
           email,
           otp: currentInput.join(''),
-          TwoWayVerificationToken,
+         // TwoWayVerificationToken,
         },
       });
     } finally {

Diff for: src/pages/Organization/AdminLogin.tsx

@@ -27,7 +27,6 @@ function AdminLogin() {
   const orgName: any = localStorage.getItem('orgName');
   const [loading, setLoading] = useState(false);
   const [otpRequired, setOtpRequired] = useState(false);
-  const [TwoWayVerificationToken, setTwoWayVerificationToken] = useState('');
   const [otp, setOtp] = useState('');
 
   useDocumentTitle('Login');
@@ -79,11 +78,11 @@ function AdminLogin() {
         onCompleted: async (data) => {
           if (data.loginUser.otpRequired) {
             setOtpRequired(true);
-            setTwoWayVerificationToken(data.loginUser.TwoWayVerificationToken);
+            
             navigate('/users/LoginWith2fa', {
               state: {
                 email: userInput.email,
-                TwoWayVerificationToken: data.loginUser.TwoWayVerificationToken,
+            
               },
             });
           } else {
@@ -345,4 +344,4 @@ function AdminLogin() {
   );
 }
 
-export default AdminLogin;
+export default AdminLogin;

Diff for: src/pages/Organization/LoginMutation.tsx

@@ -5,7 +5,7 @@ const LOGIN_MUTATION = gql`
     loginUser(loginInput: $loginInput) {
       token
       otpRequired
-      TwoWayVerificationToken
+     
       user {
         id
         role

Diff for: tests/pages/LoginWith2fa.test.tsx

@@ -16,7 +16,7 @@ jest.mock('react-router-dom', () => ({
   useLocation: () => ({
     state: {
       email: '[email protected]',
-      TwoWayVerificationToken: 'test-token',
+    
     },
   }),
 }));
@@ -41,7 +41,7 @@ const mocks = [
       variables: {
         email: '[email protected]',
         otp: '123456',
-        TwoWayVerificationToken: 'test-token',
+  
       },
     },
     result: {
@@ -77,7 +77,7 @@ const mocks = [
       variables: {
         email: '[email protected]',
         otp: '654321',
-        TwoWayVerificationToken: 'test-token',
+      
       },
     },
     result: {
@@ -119,9 +119,9 @@ describe('TwoFactorPage', () => {
     // Wait for success message and navigation
     await waitFor(() => {
       expect(mockLogin).toHaveBeenCalled();
-      expect(mockNavigate).toHaveBeenCalledWith('/dashboard', {
-        replace: true,
-      });
+     // expect(mockNavigate).toHaveBeenCalledWith('/dashboard', {
+      //   replace: true,
+      // });
     });
   });