Bug 1577040 - Part 6: Fix a couple of tests, r=Ehsan

Differential Revision: https://phabricator.services.mozilla.com/D63304

toolkit/components/url-classifier/tests/unit/test_shouldclassify.js

@@ -128,6 +128,7 @@ add_task(async function testShouldClassify() {
     "privacy.trackingprotection.annotate_channels",
     true
   );
+  Services.prefs.setBoolPref("network.dns.native-is-localhost", true);
 
   setupHttpServer();
 
@@ -141,9 +142,11 @@ add_task(async function testShouldClassify() {
       channel.asyncOpen({
         onStartRequest: (request, context) => {
           Assert.equal(
-            request
-              .QueryInterface(Ci.nsIClassifiedChannel)
-              .isTrackingResource(),
+            !!(
+              request.QueryInterface(Ci.nsIClassifiedChannel)
+                .classificationFlags &
+              Ci.nsIClassifiedChannel.CLASSIFIED_ANY_BASIC_TRACKING
+            ),
             getExpectedResult(params)
           );
           request.cancel(Cr.NS_ERROR_ABORT);