app/src/test/java/dev/msfjarvis/aps/util/viewmodel/StrictDomainRegexTest.kt


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60

/*
 * Copyright © 2014-2021 The Android Password Store Authors. All Rights Reserved.
 * SPDX-License-Identifier: GPL-3.0-only
 */
package dev.msfjarvis.aps.util.viewmodel

import kotlin.test.assertFalse
import kotlin.test.assertNull
import kotlin.test.assertTrue
import org.junit.Test
import org.junit.runner.RunWith
import org.robolectric.RobolectricTestRunner
import org.robolectric.annotation.Config

private infix fun String.matchedForDomain(domain: String) =
  SearchableRepositoryViewModel.generateStrictDomainRegex(domain)?.containsMatchIn(this) == true

@RunWith(RobolectricTestRunner::class)
@Config(sdk = [23])
class StrictDomainRegexTest {

  @Test
  fun acceptsLiteralDomain() {
    assertTrue("work/example.org/john.doe@example.org.gpg" matchedForDomain "example.org")
    assertTrue("example.org/john.doe@example.org.gpg" matchedForDomain "example.org")
    assertTrue("example.org.gpg" matchedForDomain "example.org")
  }

  @Test
  fun acceptsSubdomains() {
    assertTrue("work/www.example.org/john.doe@example.org.gpg" matchedForDomain "example.org")
    assertTrue("www2.example.org/john.doe@example.org.gpg" matchedForDomain "example.org")
    assertTrue("www.login.example.org.gpg" matchedForDomain "example.org")
  }

  @Test
  fun rejectsPhishingAttempts() {
    assertFalse("example.org.gpg" matchedForDomain "xample.org")
    assertFalse("login.example.org.gpg" matchedForDomain "xample.org")
    assertFalse("example.org/john.doe@exmple.org.gpg" matchedForDomain "xample.org")
    assertFalse("example.org.gpg" matchedForDomain "e/xample.org")
  }

  @Test
  fun rejectNonGpgComponentMatches() {
    assertFalse("work/example.org" matchedForDomain "example.org")
  }

  @Test
  fun rejectsEmailAddresses() {
    assertFalse("work/notexample.org/john.doe@example.org.gpg" matchedForDomain "example.org")
    assertFalse("work/notexample.org/john.doe@www.example.org.gpg" matchedForDomain "example.org")
    assertFalse("work/john.doe@www.example.org/foo.org" matchedForDomain "example.org")
  }

  @Test
  fun rejectsPathSeparators() {
    assertNull(SearchableRepositoryViewModel.generateStrictDomainRegex("ex/ample.org"))
  }
}