Chromium Code Reviews| OLD | NEW |
|---|---|
| (Empty) | |
| 1 // Copyright (c) 2013 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 "base/memory/scoped_ptr.h" | |
| 6 #include "base/message_loop.h" | |
| 7 #include "base/threading/sequenced_worker_pool.h" | |
| 8 #include "content/browser/media/dtls_identity_store.h" | |
| 9 #include "content/public/test/test_browser_thread.h" | |
| 10 #include "googleurl/src/gurl.h" | |
| 11 #include "testing/gtest/include/gtest/gtest.h" | |
| 12 | |
| 13 | |
| 14 namespace content { | |
| 15 namespace { | |
| 16 | |
| 17 class DTLSIdentityStoreTest : public testing::Test { | |
| 18 public: | |
| 19 DTLSIdentityStoreTest() : | |
| 20 ui_thread_(BrowserThread::UI, &message_loop_), | |
| 21 sequenced_worker_pool_(new base::SequencedWorkerPool( | |
| 22 3, "ServerBoundCertServiceTest")), | |
| 23 completed_(false) {} | |
| 24 | |
| 25 virtual ~DTLSIdentityStoreTest() { | |
| 26 sequenced_worker_pool_->Shutdown(); | |
| 27 } | |
| 28 | |
| 29 void OnGetOrGenerateCompleted(const std::string& certificate, | |
| 30 const std::string& private_key) { | |
| 31 ASSERT_NE("", certificate); | |
| 32 ASSERT_NE("", private_key); | |
| 33 completed_ = true; | |
| 34 } | |
| 35 | |
| 36 protected: | |
| 37 virtual void SetUp() { | |
| 38 completed_ = false; | |
| 39 } | |
|
Ryan Sleevi
2013/06/04 19:13:52
There is no need to use SetUp/TearDown
see https:
jiayl
2013/06/06 05:10:24
Done.
| |
| 40 | |
| 41 base::MessageLoopForUI message_loop_; | |
| 42 TestBrowserThread ui_thread_; | |
| 43 scoped_refptr<base::SequencedWorkerPool> sequenced_worker_pool_; | |
| 44 bool completed_; | |
| 45 }; | |
| 46 | |
| 47 TEST_F(DTLSIdentityStoreTest, GetOrGenerateIdentity) { | |
| 48 DTLSIdentityStore* store = new DTLSIdentityStore(sequenced_worker_pool_); | |
| 49 store->GetOrGenerateIdentity( | |
| 50 GURL("http://google.com"), "a", "b", | |
| 51 base::Bind(&DTLSIdentityStoreTest::OnGetOrGenerateCompleted, | |
| 52 base::Unretained(this))); | |
| 53 sequenced_worker_pool_->FlushForTesting(); | |
| 54 message_loop_.RunUntilIdle(); | |
| 55 | |
| 56 EXPECT_EQ(true, completed_); | |
| 57 } | |
| 58 | |
| 59 } // namespace | |
| 60 } // namespace content | |
| OLD | NEW |