fix test

PermanentOrg · Dec 3, 2024 · 67e08d2 · 67e08d2
1 parent 9185b50
commit 67e08d2
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/src/app/onboarding/components/onboarding/onboarding.component.spec.ts b/src/app/onboarding/components/onboarding/onboarding.component.spec.ts
@@ -211,16 +211,17 @@ describe('OnboardingComponent #onboarding', () => {
     expect(removeItemSpy).toHaveBeenCalledWith('shareToken');
   });
 
-  it('should navigate to /app if shareToken is not in localStorage', async () => {
+  it('should navigate to /app/welcome if shareToken is not in localStorage and isGlam is false', async () => {
     const { instance, fixture } = await shallow.render();
 
     spyOn(localStorage, 'getItem').and.returnValue(null);
-
+    instance.isGlam = false;
+    instance.acceptedInvite = false;
     instance.setScreen(OnboardingScreen.done);
     instance.selectedPendingArchive = null;
     fixture.detectChanges();
     await fixture.whenStable();
 
-    expect(mockRouter.navigate).toHaveBeenCalledWith(['/app']);
+    expect(mockRouter.navigate).toHaveBeenCalledWith(['/app', 'welcome']);
   });
 });
diff --git a/src/app/onboarding/components/onboarding/onboarding.component.ts b/src/app/onboarding/components/onboarding/onboarding.component.ts
@@ -88,9 +88,14 @@ export class OnboardingComponent implements OnInit {
       this.selectedPendingArchive = null;
     }
     if (screen === OnboardingScreen.done) {
+      if (!this.isGlam && this.acceptedInvite) {
+        this.router.navigate(['/app', 'welcome-invitation']);
+      }
       if (localStorage.getItem('shareToken')) {
         localStorage.removeItem('shareToken');
         this.router.navigate(['/app', 'shares']);
+      } else if (!this.isGlam) {
+        this.router.navigate(['/app', 'welcome']);
       } else {
         this.router.navigate(['/app']);
       }