| Index: content/browser/media/dtls_identity_store_unittest.cc
|
| diff --git a/content/browser/media/dtls_identity_store_unittest.cc b/content/browser/media/dtls_identity_store_unittest.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..24f9c2ec6da6d73592f6a8602b6e096b6c957974
|
| --- /dev/null
|
| +++ b/content/browser/media/dtls_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/threading/sequenced_worker_pool.h"
|
| +#include "content/browser/media/dtls_identity_store.h"
|
| +#include "content/public/test/test_browser_thread_bundle.h"
|
| +#include "googleurl/src/gurl.h"
|
| +#include "net/base/net_errors.h"
|
| +#include "testing/gtest/include/gtest/gtest.h"
|
| +
|
| +namespace content {
|
| +namespace {
|
| +
|
| +class DTLSIdentityStoreForTest : public DTLSIdentityStore {
|
| + public:
|
| + DTLSIdentityStoreForTest(const scoped_refptr<base::TaskRunner>& task_runner)
|
| + : DTLSIdentityStore(task_runner) {}
|
| +
|
| + virtual ~DTLSIdentityStoreForTest() {}
|
| +};
|
| +
|
| +class DTLSIdentityStoreTest : public testing::Test {
|
| + public:
|
| + DTLSIdentityStoreTest()
|
| + : browser_thread_bundle_(TestBrowserThreadBundle::IO_MAINLOOP),
|
| + sequenced_worker_pool_(
|
| + new base::SequencedWorkerPool(3, "ServerBoundCertServiceTest")) {
|
| + dtls_identity_store_.reset(
|
| + new DTLSIdentityStoreForTest(sequenced_worker_pool_));
|
| + }
|
| +
|
| + virtual ~DTLSIdentityStoreTest() { sequenced_worker_pool_->Shutdown(); }
|
| +
|
| + protected:
|
| + TestBrowserThreadBundle browser_thread_bundle_;
|
| + scoped_refptr<base::SequencedWorkerPool> sequenced_worker_pool_;
|
| + scoped_ptr<DTLSIdentityStoreForTest> dtls_identity_store_;
|
| +};
|
| +
|
| +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(DTLSIdentityStoreTest, RequestIdentity) {
|
| + scoped_ptr<bool> completed(new bool(false));
|
| + dtls_identity_store_->RequestIdentity(
|
| + GURL("http://google.com"),
|
| + "a",
|
| + "b",
|
| + base::Bind(&OnRequestCompleted, base::Unretained(completed.get())));
|
| + sequenced_worker_pool_->FlushForTesting();
|
| + base::RunLoop().RunUntilIdle();
|
| + EXPECT_EQ(true, *completed);
|
| +}
|
| +
|
| +TEST_F(DTLSIdentityStoreTest, CancelRequest) {
|
| + scoped_ptr<bool> completed(new bool(false));
|
| + base::Closure cancel_closure =
|
| + dtls_identity_store_->RequestIdentity(
|
| + GURL("http://google.com"),
|
| + "a",
|
| + "b",
|
| + base::Bind(&OnRequestCompleted, base::Unretained(completed.get())));
|
| + cancel_closure.Run();
|
| + sequenced_worker_pool_->FlushForTesting();
|
| + base::RunLoop().RunUntilIdle();
|
| + EXPECT_EQ(false, *completed);
|
| +}
|
| +
|
| +TEST_F(DTLSIdentityStoreTest, MultipleRequests) {
|
| + scoped_ptr<bool> completed_1(new bool(false));
|
| + scoped_ptr<bool> completed_2(new bool(false));
|
| +
|
| + dtls_identity_store_->RequestIdentity(
|
| + GURL("http://foo.com"),
|
| + "a",
|
| + "b",
|
| + base::Bind(&OnRequestCompleted, base::Unretained(completed_1.get())));
|
| +
|
| + dtls_identity_store_->RequestIdentity(
|
| + GURL("http://bar.com"),
|
| + "a",
|
| + "b",
|
| + base::Bind(&OnRequestCompleted, base::Unretained(completed_2.get())));
|
| +
|
| + sequenced_worker_pool_->FlushForTesting();
|
| + base::RunLoop().RunUntilIdle();
|
| + EXPECT_EQ(true, *completed_1);
|
| + EXPECT_EQ(true, *completed_2);
|
| +}
|
| +
|
| +} // namespace
|
| +} // namespace content
|
|
|