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/non_blocking_invalidator.h" 6 7 #include "base/bind_helpers.h" 8 #include "base/memory/ref_counted.h" 9 #include "base/memory/scoped_ptr.h" 10 #include "base/message_loop/message_loop.h" 11 #include "base/run_loop.h" 12 #include "base/threading/thread.h" 13 #include "google/cacheinvalidation/types.pb.h" 14 #include "jingle/notifier/base/fake_base_task.h" 15 #include "net/url_request/url_request_test_util.h" 16 #include "sync/internal_api/public/util/weak_handle.h" 17 #include "sync/notifier/fake_invalidation_handler.h" 18 #include "sync/notifier/invalidation_state_tracker.h" 19 #include "sync/notifier/invalidator_test_template.h" 20 #include "testing/gtest/include/gtest/gtest.h" 21 22 namespace syncer { 23 24 namespace { 25 26 class NonBlockingInvalidatorTestDelegate { 27 public: NonBlockingInvalidatorTestDelegate()28 NonBlockingInvalidatorTestDelegate() : io_thread_("IO thread") {} 29 ~NonBlockingInvalidatorTestDelegate()30 ~NonBlockingInvalidatorTestDelegate() { 31 DestroyInvalidator(); 32 } 33 CreateInvalidator(const std::string & invalidator_client_id,const std::string & initial_state,const base::WeakPtr<InvalidationStateTracker> & invalidation_state_tracker)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 base::Thread::Options options; 41 options.message_loop_type = base::MessageLoop::TYPE_IO; 42 io_thread_.StartWithOptions(options); 43 request_context_getter_ = 44 new net::TestURLRequestContextGetter(io_thread_.message_loop_proxy()); 45 notifier::NotifierOptions invalidator_options; 46 invalidator_options.request_context_getter = request_context_getter_; 47 invalidator_.reset( 48 new NonBlockingInvalidator( 49 invalidator_options, 50 invalidator_client_id, 51 UnackedInvalidationsMap(), 52 initial_state, 53 MakeWeakHandle(invalidation_state_tracker), 54 "fake_client_info")); 55 } 56 GetInvalidator()57 Invalidator* GetInvalidator() { 58 return invalidator_.get(); 59 } 60 DestroyInvalidator()61 void DestroyInvalidator() { 62 invalidator_.reset(); 63 request_context_getter_ = NULL; 64 io_thread_.Stop(); 65 message_loop_.RunUntilIdle(); 66 } 67 WaitForInvalidator()68 void WaitForInvalidator() { 69 base::RunLoop run_loop; 70 ASSERT_TRUE( 71 io_thread_.message_loop_proxy()->PostTaskAndReply( 72 FROM_HERE, 73 base::Bind(&base::DoNothing), 74 run_loop.QuitClosure())); 75 run_loop.Run(); 76 } 77 TriggerOnInvalidatorStateChange(InvalidatorState state)78 void TriggerOnInvalidatorStateChange(InvalidatorState state) { 79 invalidator_->OnInvalidatorStateChange(state); 80 } 81 TriggerOnIncomingInvalidation(const ObjectIdInvalidationMap & invalidation_map)82 void TriggerOnIncomingInvalidation( 83 const ObjectIdInvalidationMap& invalidation_map) { 84 invalidator_->OnIncomingInvalidation(invalidation_map); 85 } 86 87 private: 88 base::MessageLoop message_loop_; 89 base::Thread io_thread_; 90 scoped_refptr<net::URLRequestContextGetter> request_context_getter_; 91 scoped_ptr<NonBlockingInvalidator> invalidator_; 92 }; 93 94 INSTANTIATE_TYPED_TEST_CASE_P( 95 NonBlockingInvalidatorTest, InvalidatorTest, 96 NonBlockingInvalidatorTestDelegate); 97 98 } // namespace 99 100 } // namespace syncer 101