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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
#include "gtest/gtest.h"
#include "nsCOMPtr.h"
#include "nsNetCID.h"
#include "nsIURIMutator.h"
#include "nsIURL.h"
#include "nsThreadPool.h"
#include "nsNetUtil.h"
TEST(TestURIMutator, Mutator)
{
nsAutoCString out;
// This test instantiates a new nsStandardURL::Mutator (via contractID)
// and uses it to create a new URI.
nsCOMPtr<nsIURI> uri;
nsresult rv = NS_MutateURI(NS_STANDARDURLMUTATOR_CONTRACTID)
.SetSpec("http://example.com"_ns)
.Finalize(uri);
ASSERT_EQ(rv, NS_OK);
ASSERT_EQ(uri->GetSpec(out), NS_OK);
ASSERT_TRUE(out == "http://example.com/"_ns);
// This test verifies that we can use NS_MutateURI to change a URI
rv = NS_MutateURI(uri)
.SetScheme("ftp"_ns)
.SetHost("mozilla.org"_ns)
.SetPathQueryRef("/path?query#ref"_ns)
.Finalize(uri);
ASSERT_EQ(rv, NS_OK);
ASSERT_EQ(uri->GetSpec(out), NS_OK);
ASSERT_TRUE(out == "ftp://mozilla.org/path?query#ref"_ns);
// This test verifies that we can pass nsIURL to Finalize, and
nsCOMPtr<nsIURL> url;
rv = NS_MutateURI(uri).SetScheme("https"_ns).Finalize(url);
ASSERT_EQ(rv, NS_OK);
ASSERT_EQ(url->GetSpec(out), NS_OK);
ASSERT_TRUE(out == "https://mozilla.org/path?query#ref"_ns);
// This test verifies that we can pass nsIURL** to Finalize.
// We need to use the explicit template because it's actually passing
// getter_AddRefs
nsCOMPtr<nsIURL> url2;
rv = NS_MutateURI(url)
.SetRef("newref"_ns)
.Finalize<nsIURL>(getter_AddRefs(url2));
ASSERT_EQ(rv, NS_OK);
ASSERT_EQ(url2->GetSpec(out), NS_OK);
ASSERT_TRUE(out == "https://mozilla.org/path?query#newref"_ns);
// This test verifies that we can pass nsIURI** to Finalize.
// No need to be explicit.
auto functionSetRef = [](nsIURI* aURI, nsIURI** aResult) -> nsresult {
return NS_MutateURI(aURI).SetRef("originalRef"_ns).Finalize(aResult);
};
nsCOMPtr<nsIURI> newURI;
rv = functionSetRef(url2, getter_AddRefs(newURI));
ASSERT_EQ(rv, NS_OK);
ASSERT_EQ(newURI->GetSpec(out), NS_OK);
ASSERT_TRUE(out == "https://mozilla.org/path?query#originalRef"_ns);
// This test verifies that we can pass nsIURI** to Finalize.
nsCOMPtr<nsIURI> uri2;
rv =
NS_MutateURI(url2).SetQuery("newquery"_ns).Finalize(getter_AddRefs(uri2));
ASSERT_EQ(rv, NS_OK);
ASSERT_EQ(uri2->GetSpec(out), NS_OK);
ASSERT_TRUE(out == "https://mozilla.org/path?newquery#newref"_ns);
// This test verifies that we can pass nsIURI** to Finalize.
// No need to be explicit.
auto functionSetQuery = [](nsIURI* aURI, nsIURL** aResult) -> nsresult {
return NS_MutateURI(aURI).SetQuery("originalQuery"_ns).Finalize(aResult);
};
nsCOMPtr<nsIURL> newURL;
rv = functionSetQuery(uri2, getter_AddRefs(newURL));
ASSERT_EQ(rv, NS_OK);
ASSERT_EQ(newURL->GetSpec(out), NS_OK);
ASSERT_TRUE(out == "https://mozilla.org/path?originalQuery#newref"_ns);
// Check that calling Finalize twice will fail.
NS_MutateURI mutator(newURL);
rv = mutator.SetQuery(""_ns).Finalize(uri2);
ASSERT_EQ(rv, NS_OK);
ASSERT_EQ(uri2->GetSpec(out), NS_OK);
ASSERT_TRUE(out == "https://mozilla.org/path#newref"_ns);
nsCOMPtr<nsIURI> uri3;
rv = mutator.Finalize(uri3);
ASSERT_EQ(rv, NS_ERROR_NOT_AVAILABLE);
ASSERT_TRUE(uri3 == nullptr);
}
extern MOZ_THREAD_LOCAL(uint32_t) gTlsURLRecursionCount;
TEST(TestURIMutator, OnAnyThread)
{
nsCOMPtr<nsIThreadPool> pool = new nsThreadPool();
pool->SetThreadLimit(60);
pool = new nsThreadPool();
for (int i = 0; i < 1000; ++i) {
nsCOMPtr<nsIRunnable> task =
NS_NewRunnableFunction("gtest-OnAnyThread", []() {
nsCOMPtr<nsIURI> uri;
nsresult rv = NS_NewURI(getter_AddRefs(uri), "http://example.com"_ns);
ASSERT_EQ(rv, NS_OK);
nsAutoCString out;
ASSERT_EQ(uri->GetSpec(out), NS_OK);
ASSERT_TRUE(out == "http://example.com/"_ns);
});
EXPECT_TRUE(task);
pool->Dispatch(task, NS_DISPATCH_NORMAL);
}
nsCOMPtr<nsIURI> uri;
nsresult rv = NS_NewURI(getter_AddRefs(uri), "http://example.com"_ns);
ASSERT_EQ(rv, NS_OK);
nsAutoCString out;
ASSERT_EQ(uri->GetSpec(out), NS_OK);
ASSERT_TRUE(out == "http://example.com/"_ns);
pool->Shutdown();
ASSERT_EQ(gTlsURLRecursionCount.get(), 0u);
}
|
