Index: content/browser/media/webrtc_identity_store_unittest.cc |
diff --git a/content/browser/media/webrtc_identity_store_unittest.cc b/content/browser/media/webrtc_identity_store_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..1a972b34829b2b17f27d822f4e412291a4ce9079 |
--- /dev/null |
+++ b/content/browser/media/webrtc_identity_store_unittest.cc |
@@ -0,0 +1,104 @@ |
+// Copyright (c) 2013 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include "base/bind.h" |
+#include "base/memory/scoped_ptr.h" |
+#include "base/message_loop.h" |
+#include "base/run_loop.h" |
+#include "base/synchronization/waitable_event.h" |
+#include "base/threading/sequenced_worker_pool.h" |
+#include "content/browser/media/webrtc_identity_store.h" |
+#include "content/browser/renderer_host/media/webrtc_identity_service_host.h" |
+#include "content/common/media/webrtc_identity_messages.h" |
+#include "content/public/test/test_browser_thread_bundle.h" |
+#include "content/public/test/test_utils.h" |
+#include "googleurl/src/gurl.h" |
+#include "net/base/net_errors.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+namespace content { |
+ |
+class WebRTCIdentityStoreTest : public testing::Test { |
+ public: |
+ WebRTCIdentityStoreTest() |
+ : browser_thread_bundle_(TestBrowserThreadBundle::IO_MAINLOOP), |
+ sequenced_worker_pool_( |
+ new base::SequencedWorkerPool(3, "WebRTCIdentityStoreTest")), |
+ webrtc_identity_store_(new WebRTCIdentityStore()) { |
+ webrtc_identity_store_->SetTaskRunnerForTesting(sequenced_worker_pool_); |
+ } |
+ |
+ virtual ~WebRTCIdentityStoreTest() { sequenced_worker_pool_->Shutdown(); } |
+ |
+ protected: |
+ TestBrowserThreadBundle browser_thread_bundle_; |
+ scoped_refptr<base::SequencedWorkerPool> sequenced_worker_pool_; |
+ scoped_ptr<WebRTCIdentityStore> webrtc_identity_store_; |
+}; |
+ |
+static void OnRequestCompleted(bool* completed, |
+ int error, |
+ const std::string& certificate, |
+ const std::string& private_key) { |
+ ASSERT_EQ(net::OK, error); |
+ ASSERT_NE("", certificate); |
+ ASSERT_NE("", private_key); |
+ *completed = true; |
+} |
+ |
+TEST_F(WebRTCIdentityStoreTest, RequestIdentity) { |
+ bool completed = false; |
+ base::Closure cancel_callback = |
+ webrtc_identity_store_->RequestIdentity( |
+ GURL("http://google.com"), |
+ "a", |
+ "b", |
+ base::Bind(&OnRequestCompleted, base::Unretained(&completed))); |
Ami GONE FROM CHROMIUM
2013/06/27 21:30:32
Is Unretained really necessary here??
jiayl
2013/06/27 21:48:28
I thought it won't compile without it because it w
Ami GONE FROM CHROMIUM
2013/06/27 22:13:08
That concern only applies to the |this| argument b
|
+ ASSERT_FALSE(cancel_callback.is_null()); |
+ sequenced_worker_pool_->FlushForTesting(); |
+ base::RunLoop().RunUntilIdle(); |
+ EXPECT_EQ(true, completed); |
+} |
+ |
+TEST_F(WebRTCIdentityStoreTest, CancelRequest) { |
+ bool completed = false; |
+ base::Closure cancel_callback = |
+ webrtc_identity_store_->RequestIdentity( |
+ GURL("http://google.com"), |
+ "a", |
+ "b", |
+ base::Bind(&OnRequestCompleted, base::Unretained(&completed))); |
+ ASSERT_FALSE(cancel_callback.is_null()); |
+ cancel_callback.Run(); |
+ sequenced_worker_pool_->FlushForTesting(); |
+ base::RunLoop().RunUntilIdle(); |
+ EXPECT_EQ(false, completed); |
+} |
+ |
+TEST_F(WebRTCIdentityStoreTest, MultipleRequests) { |
+ bool completed_1 = false; |
+ bool completed_2 = false; |
+ base::Closure cancel_callback_1 = |
+ webrtc_identity_store_->RequestIdentity( |
+ GURL("http://foo.com"), |
+ "a", |
+ "b", |
+ base::Bind(&OnRequestCompleted, base::Unretained(&completed_1))); |
+ ASSERT_FALSE(cancel_callback_1.is_null()); |
+ |
+ base::Closure cancel_callback_2 = |
+ webrtc_identity_store_->RequestIdentity( |
+ GURL("http://bar.com"), |
+ "a", |
+ "b", |
+ base::Bind(&OnRequestCompleted, base::Unretained(&completed_2))); |
+ ASSERT_FALSE(cancel_callback_2.is_null()); |
+ |
+ sequenced_worker_pool_->FlushForTesting(); |
+ base::RunLoop().RunUntilIdle(); |
+ EXPECT_EQ(true, completed_1); |
+ EXPECT_EQ(true, completed_2); |
+} |
+ |
+} // namespace content |