| /* |
| * Copyright 2016 Google Inc. All Rights Reserved. |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #include "cobalt/loader/blob_fetcher.h" |
| |
| #include <map> |
| #include <string> |
| #include <vector> |
| |
| #include "base/bind.h" |
| #include "base/message_loop.h" |
| #include "base/run_loop.h" |
| #include "cobalt/loader/fetcher_test.h" |
| #include "testing/gmock/include/gmock/gmock.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| using ::testing::AtLeast; |
| using ::testing::InSequence; |
| using ::testing::StrictMock; |
| using ::testing::_; |
| |
| namespace cobalt { |
| namespace loader { |
| namespace { |
| |
| typedef std::map<std::string, std::vector<char> > TestRegistry; |
| |
| bool TestResolver(const TestRegistry& registry, const GURL& url, |
| const char** data, size_t* size) { |
| DCHECK(data); |
| DCHECK(size); |
| |
| *size = 0; |
| *data = NULL; |
| |
| TestRegistry::const_iterator match = registry.find(url.spec()); |
| if (match != registry.end()) { |
| const std::vector<char>& buffer = match->second; |
| *size = buffer.size(); |
| |
| if (!buffer.empty()) { |
| *data = &buffer[0]; |
| } |
| |
| return true; |
| } else { |
| return false; |
| } |
| } |
| |
| TEST(BlobFetcherTest, NonExistentBlobURL) { |
| MessageLoop message_loop(MessageLoop::TYPE_DEFAULT); |
| base::RunLoop run_loop; |
| |
| StrictMock<MockFetcherHandler> fetcher_handler_mock(&run_loop); |
| EXPECT_CALL(fetcher_handler_mock, OnError(_, _)); |
| |
| TestRegistry registry; |
| |
| scoped_ptr<BlobFetcher> blob_fetcher = make_scoped_ptr( |
| new loader::BlobFetcher(GURL("blob:sd98sdfuh8sdh"), &fetcher_handler_mock, |
| base::Bind(&TestResolver, registry))); |
| |
| run_loop.Run(); |
| |
| EXPECT_EQ(blob_fetcher.get(), fetcher_handler_mock.fetcher()); |
| } |
| |
| TEST(BlobFetcherTest, EmptyBlob) { |
| MessageLoop message_loop(MessageLoop::TYPE_DEFAULT); |
| base::RunLoop run_loop; |
| InSequence dummy; |
| |
| StrictMock<MockFetcherHandler> fetcher_handler_mock(&run_loop); |
| EXPECT_CALL(fetcher_handler_mock, OnDone(_)); |
| |
| const char* url = "blob:28y3-fsdaf-dsfa"; |
| |
| TestRegistry registry; |
| registry[url]; |
| |
| scoped_ptr<BlobFetcher> blob_fetcher = make_scoped_ptr( |
| new loader::BlobFetcher(GURL(url), &fetcher_handler_mock, |
| base::Bind(&TestResolver, registry))); |
| |
| run_loop.Run(); |
| |
| EXPECT_EQ(0, fetcher_handler_mock.data().size()); |
| |
| EXPECT_EQ(blob_fetcher.get(), fetcher_handler_mock.fetcher()); |
| } |
| |
| TEST(BlobFetcherTest, ValidBlob) { |
| MessageLoop message_loop(MessageLoop::TYPE_DEFAULT); |
| base::RunLoop run_loop; |
| InSequence dummy; |
| |
| StrictMock<MockFetcherHandler> fetcher_handler_mock(&run_loop); |
| EXPECT_CALL(fetcher_handler_mock, OnReceived(_, _, _)).Times(AtLeast(1)); |
| EXPECT_CALL(fetcher_handler_mock, OnDone(_)); |
| |
| const char* url = "blob:28y3-fsdaf-dsfa"; |
| |
| TestRegistry registry; |
| std::vector<char>& buffer = registry[url]; |
| buffer.push_back('a'); |
| buffer.push_back(0); |
| buffer.push_back(7); |
| |
| scoped_ptr<BlobFetcher> blob_fetcher = make_scoped_ptr( |
| new loader::BlobFetcher(GURL(url), &fetcher_handler_mock, |
| base::Bind(&TestResolver, registry))); |
| |
| run_loop.Run(); |
| |
| const std::string& data = fetcher_handler_mock.data(); |
| ASSERT_EQ(3, data.size()); |
| EXPECT_EQ('a', data[0]); |
| EXPECT_EQ(0, data[1]); |
| EXPECT_EQ(7, data[2]); |
| |
| EXPECT_EQ(blob_fetcher.get(), fetcher_handler_mock.fetcher()); |
| } |
| |
| } // namespace |
| } // namespace loader |
| } // namespace cobalt |