summaryrefslogtreecommitdiffstats
path: root/mobile/android/geckoview/src/androidTest/java/org/mozilla/geckoview/test/PrivateModeTest.kt
blob: 7df55b1ccb3b509b566b540361fdc4416948dc09 (plain)
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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
/* -*- Mode: Java; c-basic-offset: 4; tab-width: 4; indent-tabs-mode: nil; -*-
 * Any copyright is dedicated to the Public Domain.
   http://creativecommons.org/publicdomain/zero/1.0/ */

package org.mozilla.geckoview.test

import androidx.test.ext.junit.runners.AndroidJUnit4
import androidx.test.filters.MediumTest
import org.hamcrest.Matchers
import org.junit.Test
import org.junit.runner.RunWith
import org.mozilla.geckoview.GeckoSessionSettings

@RunWith(AndroidJUnit4::class)
@MediumTest
class PrivateModeTest : BaseSessionTest() {
    @Test
    fun privateDataNotShared() {
        mainSession.loadUri("https://example.com")
        mainSession.waitForPageStop()

        mainSession.evaluateJS(
            """
            localStorage.setItem('ctx', 'regular');
        """,
        )

        val privateSession = sessionRule.createOpenSession(
            GeckoSessionSettings.Builder(mainSession.settings)
                .usePrivateMode(true)
                .build(),
        )
        privateSession.loadUri("https://example.com")
        privateSession.waitForPageStop()
        var localStorage = privateSession.evaluateJS(
            """
            localStorage.getItem('ctx') || 'null'
        """,
        ) as String

        // Ensure that the regular session's data hasn't leaked into the private session.
        assertThat(
            "Private mode local storage value should be empty",
            localStorage,
            Matchers.equalTo("null"),
        )

        privateSession.evaluateJS(
            """
            localStorage.setItem('ctx', 'private');
        """,
        )

        localStorage = mainSession.evaluateJS(
            """
            localStorage.getItem('ctx') || 'null'
        """,
        ) as String

        // Conversely, ensure private data hasn't leaked into the regular session.
        assertThat(
            "Regular mode storage value should be unchanged",
            localStorage,
            Matchers.equalTo("regular"),
        )
    }

    @Test
    fun privateModeStorageShared() {
        // Two private mode sessions should share the same storage (bug 1533406).
        val privateSession1 = sessionRule.createOpenSession(
            GeckoSessionSettings.Builder(mainSession.settings)
                .usePrivateMode(true)
                .build(),
        )
        privateSession1.loadUri("https://example.com")
        privateSession1.waitForPageStop()

        privateSession1.evaluateJS(
            """
            localStorage.setItem('ctx', 'private');
        """,
        )

        val privateSession2 = sessionRule.createOpenSession(
            GeckoSessionSettings.Builder(mainSession.settings)
                .usePrivateMode(true)
                .build(),
        )
        privateSession2.loadUri("https://example.com")
        privateSession2.waitForPageStop()

        val localStorage = privateSession2.evaluateJS(
            """
            localStorage.getItem('ctx') || 'null'
        """,
        ) as String

        assertThat(
            "Private mode storage value still set",
            localStorage,
            Matchers.equalTo("private"),
        )
    }
}