mirror of
https://github.com/mozilla/gecko-dev.git
synced 2025-11-12 06:08:24 +02:00
808d8be324
Differential Revision: https://phabricator.services.mozilla.com/D28535 --HG-- extra : moz-landing-system : lando
272 lines
10 KiB
C++
272 lines
10 KiB
C++
/* vim:set ts=2 sw=2 sts=2 et: */
|
|
/* Any copyright is dedicated to the Public Domain.
|
|
* http://creativecommons.org/publicdomain/zero/1.0/
|
|
*/
|
|
|
|
#include "core/TelemetryOrigin.h"
|
|
#include "gtest/gtest.h"
|
|
#include "gmock/gmock.h"
|
|
#include "mozilla/Services.h"
|
|
#include "mozilla/Telemetry.h"
|
|
#include "mozilla/Unused.h"
|
|
#include "nsIObserverService.h"
|
|
#include "TelemetryFixture.h"
|
|
#include "TelemetryTestHelpers.h"
|
|
|
|
using namespace mozilla;
|
|
using namespace TelemetryTestHelpers;
|
|
using mozilla::Telemetry::OriginMetricID;
|
|
using ::testing::_;
|
|
using ::testing::AtLeast;
|
|
using ::testing::StrEq;
|
|
|
|
// Test that we can properly record origin stuff using the C++ API.
|
|
TEST_F(TelemetryTestFixture, RecordOrigin) {
|
|
AutoJSContextWithGlobal cx(mCleanGlobal);
|
|
JSContext* aCx = cx.GetJSContext();
|
|
|
|
Unused << mTelemetry->ClearOrigins();
|
|
|
|
const nsLiteralCString doubleclick("doubleclick.net");
|
|
const nsLiteralCString telemetryTest1("telemetry.test_test1");
|
|
|
|
Telemetry::RecordOrigin(OriginMetricID::TelemetryTest_Test1, doubleclick);
|
|
|
|
JS::RootedValue originSnapshot(aCx);
|
|
GetOriginSnapshot(aCx, &originSnapshot);
|
|
|
|
ASSERT_FALSE(originSnapshot.isNullOrUndefined())
|
|
<< "Origin snapshot must not be null/undefined.";
|
|
|
|
JS::RootedValue origins(aCx);
|
|
JS::RootedObject snapshotObj(aCx, &originSnapshot.toObject());
|
|
ASSERT_TRUE(JS_GetProperty(aCx, snapshotObj, telemetryTest1.get(), &origins))
|
|
<< "telemetry.test_test1 must be in the snapshot.";
|
|
|
|
JS::RootedObject originsObj(aCx, &origins.toObject());
|
|
JS::RootedValue count(aCx);
|
|
ASSERT_TRUE(JS_GetProperty(aCx, originsObj, doubleclick.get(), &count));
|
|
ASSERT_TRUE(count.isInt32() && count.toInt32() == 1)
|
|
<< "Must have recorded the origin exactly once.";
|
|
|
|
// Now test that the snapshot didn't clear things out.
|
|
GetOriginSnapshot(aCx, &originSnapshot);
|
|
ASSERT_FALSE(originSnapshot.isNullOrUndefined());
|
|
JS::RootedObject unemptySnapshotObj(aCx, &originSnapshot.toObject());
|
|
JS::Rooted<JS::IdVector> ids(aCx, JS::IdVector(aCx));
|
|
ASSERT_TRUE(JS_Enumerate(aCx, unemptySnapshotObj, &ids));
|
|
ASSERT_GE(ids.length(), (unsigned)0) << "Returned object must not be empty.";
|
|
}
|
|
|
|
TEST_F(TelemetryTestFixture, RecordOriginTwiceAndClear) {
|
|
AutoJSContextWithGlobal cx(mCleanGlobal);
|
|
JSContext* aCx = cx.GetJSContext();
|
|
|
|
Unused << mTelemetry->ClearOrigins();
|
|
|
|
const nsLiteralCString doubleclick("doubleclick.net");
|
|
const nsLiteralCString telemetryTest1("telemetry.test_test1");
|
|
|
|
Telemetry::RecordOrigin(OriginMetricID::TelemetryTest_Test1, doubleclick);
|
|
Telemetry::RecordOrigin(OriginMetricID::TelemetryTest_Test1, doubleclick);
|
|
|
|
JS::RootedValue originSnapshot(aCx);
|
|
GetOriginSnapshot(aCx, &originSnapshot, true /* aClear */);
|
|
|
|
ASSERT_FALSE(originSnapshot.isNullOrUndefined())
|
|
<< "Origin snapshot must not be null/undefined.";
|
|
|
|
JS::RootedValue origins(aCx);
|
|
JS::RootedObject snapshotObj(aCx, &originSnapshot.toObject());
|
|
ASSERT_TRUE(JS_GetProperty(aCx, snapshotObj, telemetryTest1.get(), &origins))
|
|
<< "telemetry.test_test1 must be in the snapshot.";
|
|
|
|
JS::RootedObject originsObj(aCx, &origins.toObject());
|
|
JS::RootedValue count(aCx);
|
|
ASSERT_TRUE(JS_GetProperty(aCx, originsObj, doubleclick.get(), &count));
|
|
ASSERT_TRUE(count.isInt32() && count.toInt32() == 2)
|
|
<< "Must have recorded the origin exactly twice.";
|
|
|
|
// Now check that snapshotting with clear actually cleared it.
|
|
GetOriginSnapshot(aCx, &originSnapshot);
|
|
ASSERT_FALSE(originSnapshot.isNullOrUndefined());
|
|
JS::RootedObject emptySnapshotObj(aCx, &originSnapshot.toObject());
|
|
JS::Rooted<JS::IdVector> ids(aCx, JS::IdVector(aCx));
|
|
ASSERT_TRUE(JS_Enumerate(aCx, emptySnapshotObj, &ids));
|
|
ASSERT_EQ(ids.length(), (unsigned)0) << "Returned object must be empty.";
|
|
}
|
|
|
|
TEST_F(TelemetryTestFixture, RecordOriginTwiceMixed) {
|
|
AutoJSContextWithGlobal cx(mCleanGlobal);
|
|
JSContext* aCx = cx.GetJSContext();
|
|
|
|
Unused << mTelemetry->ClearOrigins();
|
|
|
|
const nsLiteralCString doubleclick("doubleclick.net");
|
|
const nsLiteralCString doubleclickHash(
|
|
"uXNT1PzjAVau8b402OMAIGDejKbiXfQX5iXvPASfO/s=");
|
|
const nsLiteralCString telemetryTest1("telemetry.test_test1");
|
|
|
|
Telemetry::RecordOrigin(OriginMetricID::TelemetryTest_Test1, doubleclick);
|
|
Telemetry::RecordOrigin(OriginMetricID::TelemetryTest_Test1, doubleclickHash);
|
|
|
|
JS::RootedValue originSnapshot(aCx);
|
|
GetOriginSnapshot(aCx, &originSnapshot, true /* aClear */);
|
|
|
|
ASSERT_FALSE(originSnapshot.isNullOrUndefined())
|
|
<< "Origin snapshot must not be null/undefined.";
|
|
|
|
JS::RootedValue origins(aCx);
|
|
JS::RootedObject snapshotObj(aCx, &originSnapshot.toObject());
|
|
ASSERT_TRUE(JS_GetProperty(aCx, snapshotObj, telemetryTest1.get(), &origins))
|
|
<< "telemetry.test_test1 must be in the snapshot.";
|
|
|
|
JS::RootedObject originsObj(aCx, &origins.toObject());
|
|
JS::RootedValue count(aCx);
|
|
ASSERT_TRUE(JS_GetProperty(aCx, originsObj, doubleclick.get(), &count));
|
|
ASSERT_TRUE(count.isInt32() && count.toInt32() == 2)
|
|
<< "Must have recorded the origin exactly twice.";
|
|
}
|
|
|
|
TEST_F(TelemetryTestFixture, RecordUnknownOrigin) {
|
|
AutoJSContextWithGlobal cx(mCleanGlobal);
|
|
JSContext* aCx = cx.GetJSContext();
|
|
|
|
Unused << mTelemetry->ClearOrigins();
|
|
|
|
const nsLiteralCString telemetryTest1("telemetry.test_test1");
|
|
const nsLiteralCString unknown("this origin isn't known to Origin Telemetry");
|
|
const nsLiteralCString unknown2("neither is this one");
|
|
|
|
Telemetry::RecordOrigin(OriginMetricID::TelemetryTest_Test1, unknown);
|
|
|
|
JS::RootedValue originSnapshot(aCx);
|
|
GetOriginSnapshot(aCx, &originSnapshot);
|
|
|
|
ASSERT_FALSE(originSnapshot.isNullOrUndefined())
|
|
<< "Origin snapshot must not be null/undefined.";
|
|
|
|
JS::RootedValue origins(aCx);
|
|
JS::RootedObject snapshotObj(aCx, &originSnapshot.toObject());
|
|
ASSERT_TRUE(JS_GetProperty(aCx, snapshotObj, telemetryTest1.get(), &origins))
|
|
<< "telemetry.test_test1 must be in the snapshot.";
|
|
|
|
JS::RootedObject originsObj(aCx, &origins.toObject());
|
|
JS::RootedValue count(aCx);
|
|
ASSERT_TRUE(JS_GetProperty(aCx, originsObj, "__UNKNOWN__", &count));
|
|
ASSERT_TRUE(count.isInt32() && count.toInt32() == 1)
|
|
<< "Must have recorded the unknown origin exactly once.";
|
|
|
|
// Record a second, different unknown origin and ensure only one is stored.
|
|
Telemetry::RecordOrigin(OriginMetricID::TelemetryTest_Test1, unknown2);
|
|
|
|
GetOriginSnapshot(aCx, &originSnapshot);
|
|
|
|
ASSERT_FALSE(originSnapshot.isNullOrUndefined())
|
|
<< "Origin snapshot must not be null/undefined.";
|
|
|
|
JS::RootedObject snapshotObj2(aCx, &originSnapshot.toObject());
|
|
ASSERT_TRUE(JS_GetProperty(aCx, snapshotObj2, telemetryTest1.get(), &origins))
|
|
<< "telemetry.test_test1 must be in the snapshot.";
|
|
|
|
JS::RootedObject originsObj2(aCx, &origins.toObject());
|
|
JS::RootedValue count2(aCx);
|
|
ASSERT_TRUE(JS_GetProperty(aCx, originsObj2, "__UNKNOWN__", &count2));
|
|
ASSERT_TRUE(count2.isInt32() && count2.toInt32() == 1)
|
|
<< "Must have recorded the unknown origin exactly once.";
|
|
}
|
|
|
|
TEST_F(TelemetryTestFixture, EncodedSnapshot) {
|
|
AutoJSContextWithGlobal cx(mCleanGlobal);
|
|
JSContext* aCx = cx.GetJSContext();
|
|
|
|
Unused << mTelemetry->ClearOrigins();
|
|
|
|
const nsLiteralCString doubleclick("doubleclick.net");
|
|
const nsLiteralCString telemetryTest1("telemetry.test_test1");
|
|
|
|
Telemetry::RecordOrigin(OriginMetricID::TelemetryTest_Test1, doubleclick);
|
|
|
|
// Properly prepare the prio prefs
|
|
// (Sourced from PrioEncoder.cpp from when it was being prototyped)
|
|
const nsLiteralCString prioKeyA(
|
|
"35AC1C7576C7C6EDD7FED6BCFC337B34D48CB4EE45C86BEEFB40BD8875707733");
|
|
const nsLiteralCString prioKeyB(
|
|
"26E6674E65425B823F1F1D5F96E3BB3EF9E406EC7FBA7DEF8B08A35DD135AF50");
|
|
Preferences::SetCString("prio.publicKeyA", prioKeyA);
|
|
Preferences::SetCString("prio.publicKeyB", prioKeyB);
|
|
|
|
nsTArray<Tuple<nsCString, nsCString>> firstStrings;
|
|
GetEncodedOriginStrings(aCx, telemetryTest1 + NS_LITERAL_CSTRING("-%u"),
|
|
firstStrings);
|
|
|
|
// Now snapshot a second time and ensure the encoded payloads change.
|
|
nsTArray<Tuple<nsCString, nsCString>> secondStrings;
|
|
GetEncodedOriginStrings(aCx, telemetryTest1 + NS_LITERAL_CSTRING("-%u"),
|
|
secondStrings);
|
|
|
|
const auto sizeOfPrioDatasPerMetric =
|
|
TelemetryOrigin::SizeOfPrioDatasPerMetric();
|
|
ASSERT_EQ(sizeOfPrioDatasPerMetric, firstStrings.Length());
|
|
ASSERT_EQ(sizeOfPrioDatasPerMetric, secondStrings.Length());
|
|
|
|
for (size_t i = 0; i < sizeOfPrioDatasPerMetric; ++i) {
|
|
auto& aStr = Get<0>(firstStrings[i]);
|
|
auto& bStr = Get<1>(firstStrings[i]);
|
|
auto& secondAStr = Get<0>(secondStrings[i]);
|
|
auto& secondBStr = Get<1>(secondStrings[i]);
|
|
|
|
ASSERT_TRUE(aStr != secondAStr)
|
|
<< "aStr (" << aStr.get() << ") must not equal secondAStr ("
|
|
<< secondAStr.get() << ")";
|
|
ASSERT_TRUE(bStr != secondBStr)
|
|
<< "bStr (" << bStr.get() << ") must not equal secondBStr ("
|
|
<< secondBStr.get() << ")";
|
|
}
|
|
}
|
|
|
|
class MockObserver final : public nsIObserver {
|
|
public:
|
|
NS_DECL_ISUPPORTS
|
|
|
|
MOCK_METHOD1(Mobserve, void(const char* aTopic));
|
|
NS_IMETHOD Observe(nsISupports* aSubject, const char* aTopic,
|
|
const char16_t* aData) override {
|
|
Mobserve(aTopic);
|
|
return NS_OK;
|
|
};
|
|
|
|
MockObserver() = default;
|
|
|
|
private:
|
|
~MockObserver() = default;
|
|
};
|
|
|
|
NS_IMPL_ISUPPORTS(MockObserver, nsIObserver);
|
|
|
|
TEST_F(TelemetryTestFixture, OriginTelemetryNotifiesTopic) {
|
|
Unused << mTelemetry->ClearOrigins();
|
|
|
|
const char* kTopic = "origin-telemetry-storage-limit-reached";
|
|
NS_NAMED_LITERAL_CSTRING(doubleclick, "doubleclick.net");
|
|
NS_NAMED_LITERAL_CSTRING(fb, "fb.com");
|
|
|
|
MockObserver* mo = new MockObserver();
|
|
nsCOMPtr<nsIObserver> nsMo(mo);
|
|
EXPECT_CALL(*mo, Mobserve(StrEq(kTopic))).Times(1);
|
|
|
|
nsCOMPtr<nsIObserverService> os = mozilla::services::GetObserverService();
|
|
ASSERT_TRUE(os);
|
|
os->AddObserver(nsMo, kTopic, false);
|
|
|
|
const size_t size = ceil(10.0 / TelemetryOrigin::SizeOfPrioDatasPerMetric());
|
|
for (size_t i = 0; i < size; ++i) {
|
|
if (i < size - 1) {
|
|
// Let's ensure we only notify the once.
|
|
Telemetry::RecordOrigin(OriginMetricID::TelemetryTest_Test1, fb);
|
|
}
|
|
Telemetry::RecordOrigin(OriginMetricID::TelemetryTest_Test1, doubleclick);
|
|
}
|
|
|
|
os->RemoveObserver(nsMo, kTopic);
|
|
}
|