sync/notifier/invalidation_notifier_unittest.cc

   1 // Copyright (c) 2012 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 "sync/notifier/invalidation_notifier.h"
   6
   7 #include "base/memory/scoped_ptr.h"
   8 #include "base/message_loop/message_loop.h"
   9 #include "jingle/notifier/base/fake_base_task.h"
  10 #include "jingle/notifier/base/notifier_options.h"
  11 #include "jingle/notifier/listener/fake_push_client.h"
  12 #include "net/url_request/url_request_test_util.h"
  13 #include "sync/internal_api/public/base/model_type.h"
  14 #include "sync/internal_api/public/util/weak_handle.h"
  15 #include "sync/notifier/fake_invalidation_handler.h"
  16 #include "sync/notifier/fake_invalidation_state_tracker.h"
  17 #include "sync/notifier/invalidation_state_tracker.h"
  18 #include "sync/notifier/invalidator_test_template.h"
  19 #include "sync/notifier/object_id_invalidation_map_test_util.h"
  20 #include "testing/gtest/include/gtest/gtest.h"
  21
  22 namespace syncer {
  23
  24 namespace {
  25
  26 class InvalidationNotifierTestDelegate {
  27  public:
  28   InvalidationNotifierTestDelegate() {}
  29
  30   ~InvalidationNotifierTestDelegate() {
  31     DestroyInvalidator();
  32   }
  33
  34   void CreateInvalidator(
  35       const std::string& invalidator_client_id,
  36       const std::string& initial_state,
  37       const base::WeakPtr<InvalidationStateTracker>&
  38           invalidation_state_tracker) {
  39     DCHECK(!invalidator_.get());
  40     invalidator_.reset(
  41         new InvalidationNotifier(
  42             scoped_ptr<notifier::PushClient>(new notifier::FakePushClient()),
  43             invalidator_client_id,
  44             InvalidationStateMap(),
  45             initial_state,
  46             MakeWeakHandle(invalidation_state_tracker),
  47             "fake_client_info"));
  48   }
  49
  50   Invalidator* GetInvalidator() {
  51     return invalidator_.get();
  52   }
  53
  54   void DestroyInvalidator() {
  55     // Stopping the invalidation notifier stops its scheduler, which deletes
  56     // any pending tasks without running them.  Some tasks "run and delete"
  57     // another task, so they must be run in order to avoid leaking the inner
  58     // task.  Stopping does not schedule any tasks, so it's both necessary and
  59     // sufficient to drain the task queue before stopping the notifier.
  60     message_loop_.RunUntilIdle();
  61     invalidator_.reset();
  62   }
  63
  64   void WaitForInvalidator() {
  65     message_loop_.RunUntilIdle();
  66   }
  67
  68   void TriggerOnInvalidatorStateChange(InvalidatorState state) {
  69     invalidator_->OnInvalidatorStateChange(state);
  70   }
  71
  72   void TriggerOnIncomingInvalidation(
  73       const ObjectIdInvalidationMap& invalidation_map) {
  74     invalidator_->OnInvalidate(invalidation_map);
  75   }
  76
  77  private:
  78   base::MessageLoop message_loop_;
  79   scoped_ptr<InvalidationNotifier> invalidator_;
  80 };
  81
  82 INSTANTIATE_TYPED_TEST_CASE_P(
  83     InvalidationNotifierTest, InvalidatorTest,
  84     InvalidationNotifierTestDelegate);
  85
  86 }  // namespace
  87
  88 }  // namespace syncer