juice-shop

1
/*
2
 * Copyright (c) 2014-2024 Bjoern Kimminich & the OWASP Juice Shop contributors.
3
 * SPDX-License-Identifier: MIT
4
 */
5

6
import { type ComponentFixture, TestBed, waitForAsync } from '@angular/core/testing'
7

8
import { TwoFactorAuthEnterComponent } from './two-factor-auth-enter.component'
9
import { SearchResultComponent } from '../search-result/search-result.component'
10
import { UserService } from '../Services/user.service'
11
import { WindowRefService } from '../Services/window-ref.service'
12

13
import { ReactiveFormsModule } from '@angular/forms'
14
import { HttpClientTestingModule } from '@angular/common/http/testing'
15
import { RouterTestingModule } from '@angular/router/testing'
16
import { BrowserAnimationsModule } from '@angular/platform-browser/animations'
17

18
import { TranslateModule } from '@ngx-translate/core'
19
import { CookieModule, CookieService } from 'ngx-cookie'
20

21
import { MatCardModule } from '@angular/material/card'
22
import { MatFormFieldModule } from '@angular/material/form-field'
23
import { MatButtonModule } from '@angular/material/button'
24
import { MatInputModule } from '@angular/material/input'
25
import { MatCheckboxModule } from '@angular/material/checkbox'
26
import { MatIconModule } from '@angular/material/icon'
27
import { MatTableModule } from '@angular/material/table'
28
import { MatPaginatorModule } from '@angular/material/paginator'
29
import { MatDialogModule } from '@angular/material/dialog'
30
import { MatDividerModule } from '@angular/material/divider'
31
import { MatGridListModule } from '@angular/material/grid-list'
32
import { MatSnackBarModule } from '@angular/material/snack-bar'
33
import { MatTooltipModule } from '@angular/material/tooltip'
34

35
import { of } from 'rxjs'
36
import { TwoFactorAuthService } from '../Services/two-factor-auth-service'
37

38
describe('TwoFactorAuthEnterComponent', () => {
39
  let component: TwoFactorAuthEnterComponent
40
  let fixture: ComponentFixture<TwoFactorAuthEnterComponent>
41
  let cookieService: any
42
  let userService: any
43
  let twoFactorAuthService: any
44

45
  beforeEach(waitForAsync(() => {
46
    userService = jasmine.createSpyObj('UserService', ['login'])
47
    userService.login.and.returnValue(of({}))
48
    userService.isLoggedIn = jasmine.createSpyObj('userService.isLoggedIn', ['next'])
49
    userService.isLoggedIn.next.and.returnValue({})
50
    twoFactorAuthService = jasmine.createSpyObj('TwoFactorAuthService', ['verify'])
51
    twoFactorAuthService.verify.and.returnValue(of({ }))
52

53
    TestBed.configureTestingModule({
54
      imports: [
55
        HttpClientTestingModule,
56
        RouterTestingModule.withRoutes([
57
          { path: 'search', component: SearchResultComponent }
58
        ]),
59
        ReactiveFormsModule,
60
        CookieModule.forRoot(),
61
        TranslateModule.forRoot(),
62
        BrowserAnimationsModule,
63
        MatCheckboxModule,
64
        MatFormFieldModule,
65
        MatCardModule,
66
        MatIconModule,
67
        MatInputModule,
68
        MatTableModule,
69
        MatPaginatorModule,
70
        MatDialogModule,
71
        MatDividerModule,
72
        MatButtonModule,
73
        MatGridListModule,
74
        MatSnackBarModule,
75
        MatTooltipModule
76
      ],
77
      declarations: [TwoFactorAuthEnterComponent, SearchResultComponent],
78
      providers: [
79
        { provide: UserService, useValue: userService },
80
        { provide: TwoFactorAuthService, useValue: twoFactorAuthService },
81
        CookieService,
82
        WindowRefService,
83
        CookieService
84
      ]
85
    })
86
      .compileComponents()
87
    cookieService = TestBed.inject(CookieService)
88
  }))
89

90
  beforeEach(() => {
91
    fixture = TestBed.createComponent(TwoFactorAuthEnterComponent)
92
    component = fixture.componentInstance
93
    fixture.detectChanges()
94
  })
95

96
  it('should create', () => {
97
    expect(component).toBeTruthy()
98
  })
99

100
  it('should store authentication token in cookie', () => {
101
    twoFactorAuthService.verify.and.returnValue(of({ token: 'TOKEN' }))
102
    component.verify()
103

104
    expect(cookieService.get('token')).toBe('TOKEN')
105
  })
106

107
  it('should store authentication token in local storage', () => {
108
    twoFactorAuthService.verify.and.returnValue(of({ token: 'TOKEN' }))
109
    component.verify()
110

111
    expect(localStorage.getItem('token')).toBe('TOKEN')
112
  })
113

114
  it('should store basket ID in session storage', () => {
115
    twoFactorAuthService.verify.and.returnValue(of({ bid: 42 }))
116
    component.verify()
117

118
    expect(sessionStorage.getItem('bid')).toBe('42')
119
  })
120

121
  xit('should notify about user login after 2FA verification', () => { // FIXME Spy call is not registered at all
122
    component.verify()
123

124
    expect(userService.isLoggedIn.next).toHaveBeenCalledWith(true)
125
  })
126
})
127