From 67572618e55e08c4c45158dfc502564828e4b82b Mon Sep 17 00:00:00 2001
From: Alexander Shalamov <alexander.shalamov@mapbox.com>
Date: Fri, 6 Sep 2019 11:51:00 +0300
Subject: [core] Add unit test to verify that Offline usage tag is not set

New unit test checks that Offline usage flag is not set for invalidated
offline region resource requests.
---
 test/storage/offline_download.test.cpp | 80 +++++++++++++++++++++++++++++++++-
 1 file changed, 79 insertions(+), 1 deletion(-)

diff --git a/test/storage/offline_download.test.cpp b/test/storage/offline_download.test.cpp
index 2256e04bf1..4e967c297c 100644
--- a/test/storage/offline_download.test.cpp
+++ b/test/storage/offline_download.test.cpp
@@ -1,8 +1,12 @@
+#include <mbgl/test/map_adapter.hpp>
 #include <mbgl/test/stub_file_source.hpp>
 #include <mbgl/test/fake_file_source.hpp>
+#include <mbgl/test/stub_map_observer.hpp>
 #include <mbgl/test/fixture_log_observer.hpp>
 #include <mbgl/test/sqlite3_test_fs.hpp>
 
+#include <mbgl/gfx/headless_frontend.hpp>
+#include <mbgl/storage/default_file_source.hpp>
 #include <mbgl/storage/offline.hpp>
 #include <mbgl/storage/offline_database.hpp>
 #include <mbgl/storage/offline_download.hpp>
@@ -14,7 +18,6 @@
 
 #include <mbgl/storage/sqlite3.hpp>
 #include <gtest/gtest.h>
-#include <iostream>
 
 using namespace mbgl;
 using namespace std::literals::string_literals;
@@ -71,6 +74,10 @@ public:
         return db.createRegion(definition, metadata);
     }
 
+    auto invalidateRegion(int64_t region) {
+        return db.invalidateRegion(region);
+    }
+
     Response response(const std::string& path) {
         Response result;
         result.data = std::make_shared<std::string>(util::read_file("test/fixtures/offline_download/"s + path));
@@ -868,4 +875,75 @@ TEST(OfflineDownload, DiskFull) {
     test.loop.run();
     EXPECT_EQ(0u, log.uncheckedCount());
 }
+
+// Test verifies that resource requests for invalidated region don't
+// have Resource::Usage::Offline tag set.
+TEST(OfflineDownload, ResourceOfflineUsageUnset) {
+    deleteDatabaseFiles();
+    test::SQLite3TestFS fs;
+
+    OfflineTest test{ filename_test_fs };
+    auto region = test.createRegion();
+    ASSERT_TRUE(region);
+
+    OfflineDownload download(
+        region->getID(),
+        OfflineTilePyramidRegionDefinition("http://127.0.0.1:3000/inline_source.style.json",
+                                            LatLngBounds::world(), 0.0, 0.0, 1.0, false),
+        test.db, test.fileSource);
+
+    test.fileSource.styleResponse = [&] (const Resource& resource) {
+        EXPECT_TRUE(resource.priority == Resource::Priority::Low);
+        EXPECT_TRUE(resource.usage == Resource::Usage::Offline);
+        return test.response("inline_source.style.json");
+    };
+
+    test.fileSource.tileResponse = [&] (const Resource& resource) {
+        EXPECT_TRUE(resource.priority == Resource::Priority::Low);
+        EXPECT_TRUE(resource.usage == Resource::Usage::Offline);
+        return test.response("0-0-0.vector.pbf");
+    };
+
+    auto observer = std::make_unique<MockObserver>();
+    observer->statusChangedFn = [&] (OfflineRegionStatus status) {
+        if (status.complete()) {
+            // Once download completes, invalidate region and try rendering map.
+            // Resource requests must not have Offline usage tag.
+            ASSERT_FALSE(test.invalidateRegion(region->getID()));
+            test.loop.stop();
+        }
+    };
+
+    download.setObserver(std::move(observer));
+    download.setState(OfflineRegionDownloadState::Active);
+    test.loop.run();
+
+    std::shared_ptr<StubFileSource> stubfileSource = std::make_shared<StubFileSource>();
+    stubfileSource->styleResponse = [&] (const Resource& resource) {
+        EXPECT_TRUE(resource.usage != Resource::Usage::Offline);
+        return test.response("inline_source.style.json");
+    };
+
+    stubfileSource->tileResponse = [&] (const Resource& resource) {
+        EXPECT_TRUE(resource.usage != Resource::Usage::Offline);
+        return test.response("0-0-0.vector.pbf");
+    };
+
+    StubMapObserver mapObserver;
+    mapObserver.didFinishRenderingFrameCallback = [&] (MapObserver::RenderMode mode) {
+        if (mode == MapObserver::RenderMode::Full) {
+            test.loop.stop();
+        }
+    };
+
+    HeadlessFrontend frontend { 1 };
+    MapAdapter map { frontend, mapObserver, stubfileSource,
+            MapOptions()
+            .withMapMode(MapMode::Continuous)
+            .withSize(frontend.getSize())};
+
+    map.getStyle().loadURL("http://127.0.0.1:3000/inline_source.style.json");
+    map.jumpTo(CameraOptions().withCenter(LatLng{0.0, 0.0}).withZoom(0));
+    test.loop.run();
+}
 #endif // __QT__
-- 
cgit v1.2.1