mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-10-19 08:15:31 +00:00
5dc21d568c
The inclusions were removed with the following very crude script and the resulting breakage was fixed up by hand. The manual fixups did either revert the changes done by the script, replace a generic header with a more specific one or replace a header with a forward declaration. find . -name "*.idl" | grep -v web-platform | grep -v third_party | while read path; do interfaces=$(grep "^\(class\|interface\).*:.*" "$path" | cut -d' ' -f2) if [ -n "$interfaces" ]; then if [[ "$interfaces" == *$'\n'* ]]; then regexp="\(" for i in $interfaces; do regexp="$regexp$i\|"; done regexp="${regexp%%\\\|}\)" else regexp="$interfaces" fi interface=$(basename "$path") rg -l "#include.*${interface%%.idl}.h" . | while read path2; do hits=$(grep -v "#include.*${interface%%.idl}.h" "$path2" | grep -c "$regexp" ) if [ $hits -eq 0 ]; then echo "Removing ${interface} from ${path2}" grep -v "#include.*${interface%%.idl}.h" "$path2" > "$path2".tmp mv -f "$path2".tmp "$path2" fi done fi done Differential Revision: https://phabricator.services.mozilla.com/D55444 --HG-- extra : moz-landing-system : lando
185 lines
5.2 KiB
C++
185 lines
5.2 KiB
C++
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2; -*- */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
#include "mozilla/scache/StartupCache.h"
|
|
#include "mozilla/scache/StartupCacheUtils.h"
|
|
|
|
#include "nsDirectoryServiceDefs.h"
|
|
#include "nsIOutputStream.h"
|
|
#include "nsISupports.h"
|
|
#include "nsIStorageStream.h"
|
|
#include "nsIObjectInputStream.h"
|
|
#include "nsIObjectOutputStream.h"
|
|
#include "nsIURI.h"
|
|
#include "nsThreadUtils.h"
|
|
#include "prenv.h"
|
|
#include "prio.h"
|
|
#include "prprf.h"
|
|
#include "mozilla/Maybe.h"
|
|
#include "mozilla/Printf.h"
|
|
#include "mozilla/UniquePtr.h"
|
|
#include "nsNetCID.h"
|
|
#include "nsIURIMutator.h"
|
|
|
|
using namespace JS;
|
|
|
|
using namespace mozilla::scache;
|
|
using mozilla::UniquePtr;
|
|
|
|
void WaitForStartupTimer() {
|
|
StartupCache* sc = StartupCache::GetSingleton();
|
|
PR_Sleep(10 * PR_TicksPerSecond());
|
|
|
|
while (true) {
|
|
NS_ProcessPendingEvents(nullptr);
|
|
if (sc->StartupWriteComplete()) {
|
|
return;
|
|
}
|
|
PR_Sleep(1 * PR_TicksPerSecond());
|
|
}
|
|
}
|
|
|
|
class TestStartupCache : public ::testing::Test {
|
|
protected:
|
|
TestStartupCache();
|
|
~TestStartupCache();
|
|
|
|
nsCOMPtr<nsIFile> mSCFile;
|
|
};
|
|
|
|
TestStartupCache::TestStartupCache() {
|
|
NS_GetSpecialDirectory(NS_OS_TEMP_DIR, getter_AddRefs(mSCFile));
|
|
mSCFile->AppendNative(NS_LITERAL_CSTRING("test-startupcache.tmp"));
|
|
#ifdef XP_WIN
|
|
nsAutoString env(NS_LITERAL_STRING("MOZ_STARTUP_CACHE="));
|
|
env.Append(mSCFile->NativePath());
|
|
_wputenv(env.get());
|
|
#else
|
|
nsAutoCString path;
|
|
mSCFile->GetNativePath(path);
|
|
char* env = mozilla::Smprintf("MOZ_STARTUP_CACHE=%s", path.get()).release();
|
|
PR_SetEnv(env);
|
|
// We intentionally leak `env` here because it is required by PR_SetEnv
|
|
MOZ_LSAN_INTENTIONALLY_LEAK_OBJECT(env);
|
|
#endif
|
|
StartupCache::GetSingleton()->InvalidateCache();
|
|
}
|
|
TestStartupCache::~TestStartupCache() {
|
|
PR_SetEnv("MOZ_STARTUP_CACHE=");
|
|
StartupCache::GetSingleton()->InvalidateCache();
|
|
}
|
|
|
|
TEST_F(TestStartupCache, StartupWriteRead) {
|
|
nsresult rv;
|
|
StartupCache* sc = StartupCache::GetSingleton();
|
|
|
|
const char* buf = "Market opportunities for BeardBook";
|
|
const char* id = "id";
|
|
const char* outbuf;
|
|
uint32_t len;
|
|
|
|
rv = sc->PutBuffer(id, UniquePtr<char[]>(strdup(buf)), strlen(buf) + 1);
|
|
EXPECT_TRUE(NS_SUCCEEDED(rv));
|
|
|
|
rv = sc->GetBuffer(id, &outbuf, &len);
|
|
EXPECT_TRUE(NS_SUCCEEDED(rv));
|
|
EXPECT_STREQ(buf, outbuf);
|
|
|
|
rv = sc->ResetStartupWriteTimer();
|
|
EXPECT_TRUE(NS_SUCCEEDED(rv));
|
|
WaitForStartupTimer();
|
|
|
|
rv = sc->GetBuffer(id, &outbuf, &len);
|
|
EXPECT_TRUE(NS_SUCCEEDED(rv));
|
|
EXPECT_STREQ(buf, outbuf);
|
|
}
|
|
|
|
TEST_F(TestStartupCache, WriteInvalidateRead) {
|
|
nsresult rv;
|
|
const char* buf = "BeardBook competitive analysis";
|
|
const char* id = "id";
|
|
const char* outbuf;
|
|
uint32_t len;
|
|
StartupCache* sc = StartupCache::GetSingleton();
|
|
ASSERT_TRUE(sc);
|
|
|
|
rv = sc->PutBuffer(id, UniquePtr<char[]>(strdup(buf)), strlen(buf) + 1);
|
|
EXPECT_TRUE(NS_SUCCEEDED(rv));
|
|
|
|
sc->InvalidateCache();
|
|
|
|
rv = sc->GetBuffer(id, &outbuf, &len);
|
|
EXPECT_EQ(rv, NS_ERROR_NOT_AVAILABLE);
|
|
}
|
|
|
|
TEST_F(TestStartupCache, WriteObject) {
|
|
nsresult rv;
|
|
|
|
nsCOMPtr<nsIURI> obj;
|
|
|
|
NS_NAMED_LITERAL_CSTRING(spec, "http://www.mozilla.org");
|
|
rv = NS_MutateURI(NS_SIMPLEURIMUTATOR_CONTRACTID).SetSpec(spec).Finalize(obj);
|
|
EXPECT_TRUE(NS_SUCCEEDED(rv));
|
|
|
|
StartupCache* sc = StartupCache::GetSingleton();
|
|
|
|
// Create an object stream. Usually this is done with
|
|
// NewObjectOutputWrappedStorageStream, but that uses
|
|
// StartupCache::GetSingleton in debug builds, and we
|
|
// don't have access to that here. Obviously.
|
|
const char* id = "id";
|
|
nsCOMPtr<nsIStorageStream> storageStream =
|
|
do_CreateInstance("@mozilla.org/storagestream;1");
|
|
ASSERT_TRUE(storageStream);
|
|
|
|
rv = storageStream->Init(256, (uint32_t)-1);
|
|
EXPECT_TRUE(NS_SUCCEEDED(rv));
|
|
|
|
nsCOMPtr<nsIObjectOutputStream> objectOutput =
|
|
do_CreateInstance("@mozilla.org/binaryoutputstream;1");
|
|
ASSERT_TRUE(objectOutput);
|
|
|
|
nsCOMPtr<nsIOutputStream> outputStream = do_QueryInterface(storageStream);
|
|
|
|
rv = objectOutput->SetOutputStream(outputStream);
|
|
EXPECT_TRUE(NS_SUCCEEDED(rv));
|
|
|
|
nsCOMPtr<nsISupports> objQI(do_QueryInterface(obj));
|
|
rv = objectOutput->WriteObject(objQI, true);
|
|
EXPECT_TRUE(NS_SUCCEEDED(rv));
|
|
|
|
UniquePtr<char[]> buf;
|
|
uint32_t len;
|
|
NewBufferFromStorageStream(storageStream, &buf, &len);
|
|
|
|
// Since this is a post-startup write, it should be written and
|
|
// available.
|
|
rv = sc->PutBuffer(id, std::move(buf), len);
|
|
EXPECT_TRUE(NS_SUCCEEDED(rv));
|
|
|
|
const char* buf2;
|
|
uint32_t len2;
|
|
nsCOMPtr<nsIObjectInputStream> objectInput;
|
|
rv = sc->GetBuffer(id, &buf2, &len2);
|
|
EXPECT_TRUE(NS_SUCCEEDED(rv));
|
|
|
|
rv = NewObjectInputStreamFromBuffer(buf2, len2, getter_AddRefs(objectInput));
|
|
EXPECT_TRUE(NS_SUCCEEDED(rv));
|
|
|
|
nsCOMPtr<nsISupports> deserialized;
|
|
rv = objectInput->ReadObject(true, getter_AddRefs(deserialized));
|
|
EXPECT_TRUE(NS_SUCCEEDED(rv));
|
|
|
|
nsCOMPtr<nsIURI> uri(do_QueryInterface(deserialized));
|
|
ASSERT_TRUE(uri);
|
|
|
|
nsCString outSpec;
|
|
rv = uri->GetSpec(outSpec);
|
|
EXPECT_TRUE(NS_SUCCEEDED(rv));
|
|
ASSERT_TRUE(outSpec.Equals(spec));
|
|
}
|