webkit/browser/appcache/appcache_test_helper.cc

   1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "webkit/browser/appcache/appcache_test_helper.h"
   6
   7 #include "base/bind.h"
   8 #include "base/bind_helpers.h"
   9 #include "base/message_loop/message_loop.h"
  10 #include "testing/gtest/include/gtest/gtest.h"
  11 #include "webkit/browser/appcache/appcache.h"
  12 #include "webkit/browser/appcache/appcache_entry.h"
  13 #include "webkit/browser/appcache/appcache_group.h"
  14 #include "webkit/browser/appcache/appcache_service.h"
  15
  16 namespace appcache {
  17
  18 AppCacheTestHelper::AppCacheTestHelper()
  19     : group_id_(0),
  20       appcache_id_(0),
  21       response_id_(0),
  22       origins_(NULL) {}
  23
  24 AppCacheTestHelper::~AppCacheTestHelper() {}
  25
  26 void AppCacheTestHelper::OnGroupAndNewestCacheStored(
  27     AppCacheGroup* /*group*/,
  28     AppCache* /*newest_cache*/,
  29     bool success,
  30     bool /*would_exceed_quota*/) {
  31   ASSERT_TRUE(success);
  32   base::MessageLoop::current()->Quit();
  33 }
  34
  35 void AppCacheTestHelper::AddGroupAndCache(AppCacheService* appcache_service,
  36                                           const GURL& manifest_url) {
  37   AppCacheGroup* appcache_group =
  38       new AppCacheGroup(appcache_service->storage(),
  39                         manifest_url,
  40                         ++group_id_);
  41   AppCache* appcache = new AppCache(appcache_service->storage(),
  42                                     ++appcache_id_);
  43   AppCacheEntry entry(AppCacheEntry::MANIFEST,
  44                       ++response_id_);
  45   appcache->AddEntry(manifest_url, entry);
  46   appcache->set_complete(true);
  47   appcache_group->AddCache(appcache);
  48   appcache_service->storage()->StoreGroupAndNewestCache(appcache_group,
  49                                                         appcache,
  50                                                         this);
  51   // OnGroupAndNewestCacheStored will quit the message loop.
  52   base::MessageLoop::current()->Run();
  53 }
  54
  55 void AppCacheTestHelper::GetOriginsWithCaches(AppCacheService* appcache_service,
  56                                               std::set<GURL>* origins) {
  57   appcache_info_ = new AppCacheInfoCollection;
  58   origins_ = origins;
  59   appcache_service->GetAllAppCacheInfo(
  60       appcache_info_.get(),
  61       base::Bind(&AppCacheTestHelper::OnGotAppCacheInfo,
  62                  base::Unretained(this)));
  63
  64   // OnGotAppCacheInfo will quit the message loop.
  65   base::MessageLoop::current()->Run();
  66 }
  67
  68 void AppCacheTestHelper::OnGotAppCacheInfo(int rv) {
  69   typedef std::map<GURL, AppCacheInfoVector> InfoByOrigin;
  70
  71   origins_->clear();
  72   for (InfoByOrigin::const_iterator origin =
  73            appcache_info_->infos_by_origin.begin();
  74        origin != appcache_info_->infos_by_origin.end(); ++origin) {
  75     origins_->insert(origin->first);
  76   }
  77   base::MessageLoop::current()->Quit();
  78 }
  79
  80 }  // namespace appcache