1 // Copyright 2014 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 "components/invalidation/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 "components/invalidation/fake_invalidation_handler.h" 14 #include "components/invalidation/invalidation_state_tracker.h" 15 #include "components/invalidation/invalidator_test_template.h" 16 #include "google/cacheinvalidation/types.pb.h" 17 #include "jingle/notifier/base/fake_base_task.h" 18 #include "net/url_request/url_request_test_util.h" 19 #include "testing/gtest/include/gtest/gtest.h" 20 21 namespace syncer { 22 23 class NonBlockingInvalidatorTestDelegate { 24 public: NonBlockingInvalidatorTestDelegate()25 NonBlockingInvalidatorTestDelegate() : io_thread_("IO thread") {} 26 ~NonBlockingInvalidatorTestDelegate()27 ~NonBlockingInvalidatorTestDelegate() { 28 DestroyInvalidator(); 29 } 30 CreateInvalidator(const std::string & invalidator_client_id,const std::string & initial_state,const base::WeakPtr<InvalidationStateTracker> & invalidation_state_tracker)31 void CreateInvalidator( 32 const std::string& invalidator_client_id, 33 const std::string& initial_state, 34 const base::WeakPtr<InvalidationStateTracker>& 35 invalidation_state_tracker) { 36 DCHECK(!invalidator_.get()); 37 base::Thread::Options options; 38 options.message_loop_type = base::MessageLoop::TYPE_IO; 39 io_thread_.StartWithOptions(options); 40 request_context_getter_ = 41 new net::TestURLRequestContextGetter(io_thread_.message_loop_proxy()); 42 notifier::NotifierOptions notifier_options; 43 notifier_options.request_context_getter = request_context_getter_; 44 NetworkChannelCreator network_channel_creator = 45 NonBlockingInvalidator::MakePushClientChannelCreator(notifier_options); 46 invalidator_.reset( 47 new NonBlockingInvalidator( 48 network_channel_creator, 49 invalidator_client_id, 50 UnackedInvalidationsMap(), 51 initial_state, 52 invalidation_state_tracker.get(), 53 "fake_client_info", 54 request_context_getter_)); 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 syncer 99