Issue 1296 - Add SetUpdateAvailableCallback

test/FilterEngine.cpp

 /*
  * This file is part of Adblock Plus <http://adblockplus.org/>,
  * Copyright (C) 2006-2014 Eyeo GmbH
  *
  * Adblock Plus is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License version 3 as
  * published by the Free Software Foundation.
  *
  * Adblock Plus is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
@@ skipping 30 matching lines @@
       BaseJsTest::SetUp();
       jsEngine->SetFileSystem(AdblockPlus::FileSystemPtr(new FileSystem));
       jsEngine->SetWebRequest(AdblockPlus::WebRequestPtr(new LazyWebRequest));
       jsEngine->SetLogSystem(AdblockPlus::LogSystemPtr(new LogSystem));
       filterEngine = FilterEnginePtr(new AdblockPlus::FilterEngine(jsEngine));
     }
   };
 
   typedef FilterEngineTestGeneric<LazyFileSystem, AdblockPlus::DefaultLogSystem> FilterEngineTest;
   typedef FilterEngineTestGeneric<VeryLazyFileSystem, LazyLogSystem> FilterEngineTestNoData;
+
+  class UpdaterTest : public ::testing::Test
+  {
+  protected:
+    class MockWebRequest : public AdblockPlus::WebRequest
+    {
+    public:
+      AdblockPlus::ServerResponse response;
+
+      AdblockPlus::ServerResponse GET(const std::string& url,
+          const AdblockPlus::HeaderList& requestHeaders) const
+      {
+        return response;
+      }
+    };
+
+    MockWebRequest* mockWebRequest;
+    FilterEnginePtr filterEngine;
+
+    void SetUp()
+    {
+      AdblockPlus::AppInfo appInfo;
+      appInfo.name = "test";
+      appInfo.version = "1.0.1";
+      AdblockPlus::JsEnginePtr jsEngine = AdblockPlus::JsEngine::New(appInfo);
+      jsEngine->SetFileSystem(AdblockPlus::FileSystemPtr(new LazyFileSystem));
+      mockWebRequest = new MockWebRequest;
+      jsEngine->SetWebRequest(AdblockPlus::WebRequestPtr(mockWebRequest));
+      filterEngine = FilterEnginePtr(new AdblockPlus::FilterEngine(jsEngine));
+    }
+  };
+
+  struct MockUpdateAvailableCallback
+  {
+    MockUpdateAvailableCallback(int& timesCalled) : timesCalled(timesCalled) {}
+
+    void operator()(const std::string&)
+    {
+      timesCalled++;
+    }
+
+  private:
+    // We currently cannot store timesCalled in the functor, see:
+    // https://issues.adblockplus.org/ticket/1378.
+    int& timesCalled;
+  };
+
+  // Workaround for https://issues.adblockplus.org/ticket/1397.
+  void NoOpUpdaterCallback(const std::string&) {}
 }
 
 TEST_F(FilterEngineTest, FilterCreation)
 {
   AdblockPlus::FilterPtr filter1 = filterEngine->GetFilter("foo");
   ASSERT_EQ(AdblockPlus::Filter::TYPE_BLOCKING, filter1->GetType());
   AdblockPlus::FilterPtr filter2 = filterEngine->GetFilter("@@foo");
   ASSERT_EQ(AdblockPlus::Filter::TYPE_EXCEPTION, filter2->GetType());
   AdblockPlus::FilterPtr filter3 = filterEngine->GetFilter("example.com##foo");
   ASSERT_EQ(AdblockPlus::Filter::TYPE_ELEMHIDE, filter3->GetType());
@@ skipping 243 matching lines @@
 
 TEST_F(FilterEngineTest, FirstRunFlag)
 {
   ASSERT_FALSE(filterEngine->IsFirstRun());
 }
 
 TEST_F(FilterEngineTestNoData, FirstRunFlag)
 {
   ASSERT_TRUE(filterEngine->IsFirstRun());
 }
+
+TEST_F(UpdaterTest, SetRemoveUpdateAvailableCallback)
+{
+  mockWebRequest->response.status = 0;
+  mockWebRequest->response.responseStatus = 200;
+  mockWebRequest->response.responseText = "\
+{\
+  \"test\": {\
+    \"version\": \"1.0.2\",\
+    \"url\": \"https://downloads.adblockplus.org/test-1.0.2.tar.gz?update\"\
+  }\
+}";
+
+  int timesCalled = 0;
+  MockUpdateAvailableCallback mockUpdateAvailableCallback(timesCalled);
+
+  filterEngine->SetUpdateAvailableCallback(mockUpdateAvailableCallback);
+  filterEngine->ForceUpdateCheck(&NoOpUpdaterCallback);
+  AdblockPlus::Sleep(100);
+  ASSERT_EQ(1, timesCalled);
+
+  filterEngine->RemoveUpdateAvailableCallback();
+  filterEngine->ForceUpdateCheck(&NoOpUpdaterCallback);
+  AdblockPlus::Sleep(100);
+  ASSERT_EQ(1, timesCalled);
+}