forked from mirrors/gecko-dev
1a53231595
MozReview-Commit-ID: Ft1deSNKuVB --HG-- extra : rebase_source : 0cf746b1d2def35606f012c994ddbda0857ccf2d
98 lines
3.1 KiB
C++
98 lines
3.1 KiB
C++
#include "LookupCache.h"
|
|
#include "LookupCacheV4.h"
|
|
#include "HashStore.h"
|
|
#include "gtest/gtest.h"
|
|
#include "nsAppDirectoryServiceDefs.h"
|
|
|
|
namespace mozilla {
|
|
namespace safebrowsing {
|
|
|
|
class PerProviderDirectoryTestUtils {
|
|
public:
|
|
template<typename T>
|
|
static nsIFile* InspectStoreDirectory(const T& aT)
|
|
{
|
|
return aT.mStoreDirectory;
|
|
}
|
|
};
|
|
|
|
} // end of namespace safebrowsing
|
|
} // end of namespace mozilla
|
|
|
|
using namespace mozilla;
|
|
using namespace mozilla::safebrowsing;
|
|
|
|
template<typename T>
|
|
void VerifyPrivateStorePath(const char* aTableName,
|
|
const char* aProvider,
|
|
nsIFile* aRootDir,
|
|
bool aUsePerProviderStore)
|
|
{
|
|
nsString rootStorePath;
|
|
nsresult rv = aRootDir->GetPath(rootStorePath);
|
|
EXPECT_EQ(rv, NS_OK);
|
|
|
|
T target(nsCString(aTableName), nsCString(aProvider), aRootDir);
|
|
|
|
nsIFile* privateStoreDirectory =
|
|
PerProviderDirectoryTestUtils::InspectStoreDirectory(target);
|
|
|
|
nsString privateStorePath;
|
|
rv = privateStoreDirectory->GetPath(privateStorePath);
|
|
ASSERT_EQ(rv, NS_OK);
|
|
|
|
nsString expectedPrivateStorePath = rootStorePath;
|
|
|
|
if (aUsePerProviderStore) {
|
|
// Use API to append "provider" to the root directoy path
|
|
nsCOMPtr<nsIFile> expectedPrivateStoreDir;
|
|
rv = aRootDir->Clone(getter_AddRefs(expectedPrivateStoreDir));
|
|
ASSERT_EQ(rv, NS_OK);
|
|
|
|
expectedPrivateStoreDir->AppendNative(nsCString(aProvider));
|
|
rv = expectedPrivateStoreDir->GetPath(expectedPrivateStorePath);
|
|
ASSERT_EQ(rv, NS_OK);
|
|
}
|
|
|
|
printf("table: %s\nprovider: %s\nroot path: %s\nprivate path: %s\n\n",
|
|
aTableName,
|
|
aProvider,
|
|
NS_ConvertUTF16toUTF8(rootStorePath).get(),
|
|
NS_ConvertUTF16toUTF8(privateStorePath).get());
|
|
|
|
ASSERT_TRUE(privateStorePath == expectedPrivateStorePath);
|
|
}
|
|
|
|
TEST(PerProviderDirectory, LookupCache)
|
|
{
|
|
RunTestInNewThread([] () -> void {
|
|
nsCOMPtr<nsIFile> rootDir;
|
|
NS_GetSpecialDirectory(NS_APP_USER_PROFILE_50_DIR, getter_AddRefs(rootDir));
|
|
|
|
// For V2 tables (NOT ending with '-proto'), root directory should be
|
|
// used as the private store.
|
|
VerifyPrivateStorePath<LookupCacheV2>("goog-phish-shavar", "google", rootDir, false);
|
|
|
|
// For V4 tables, if provider is found, use per-provider subdirectory;
|
|
// If not found, use root directory.
|
|
VerifyPrivateStorePath<LookupCacheV4>("goog-noprovider-proto", "", rootDir, false);
|
|
VerifyPrivateStorePath<LookupCacheV4>("goog-phish-proto", "google4", rootDir, true);
|
|
});
|
|
}
|
|
|
|
TEST(PerProviderDirectory, HashStore)
|
|
{
|
|
RunTestInNewThread([] () -> void {
|
|
nsCOMPtr<nsIFile> rootDir;
|
|
NS_GetSpecialDirectory(NS_APP_USER_PROFILE_50_DIR, getter_AddRefs(rootDir));
|
|
|
|
// For V2 tables (NOT ending with '-proto'), root directory should be
|
|
// used as the private store.
|
|
VerifyPrivateStorePath<HashStore>("goog-phish-shavar", "google", rootDir, false);
|
|
|
|
// For V4 tables, if provider is found, use per-provider subdirectory;
|
|
// If not found, use root directory.
|
|
VerifyPrivateStorePath<HashStore>("goog-noprovider-proto", "", rootDir, false);
|
|
VerifyPrivateStorePath<HashStore>("goog-phish-proto", "google4", rootDir, true);
|
|
});
|
|
}
|