ios/chrome/browser/download/browser_download_service_factory_unittest.mm

// Copyright 2017 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 "ios/chrome/browser/download/browser_download_service_factory.h"

#include "ios/chrome/browser/browser_state/test_chrome_browser_state.h"
#include "ios/chrome/browser/download/browser_download_service.h"
#import "ios/web/public/download/download_controller.h"
#include "ios/web/public/test/test_web_thread_bundle.h"
#include "testing/platform_test.h"

#if !defined(__has_feature) || !__has_feature(objc_arc)
#error "This file requires ARC support."
#endif

// Test fixture for testing BrowserDownloadServiceFactory class.
class BrowserDownloadServiceFactoryTest : public PlatformTest {
 protected:
  BrowserDownloadServiceFactoryTest()
      : browser_state_(browser_state_builder_.Build()) {}
  web::TestWebThreadBundle thread_bundle_;  // ChromeBrowserState needs thread.
  TestChromeBrowserState::Builder browser_state_builder_;
  std::unique_ptr<TestChromeBrowserState> browser_state_;
};

// Tests that BrowserDownloadServiceFactory creates BrowserDownloadService and
// sets it as DownloadControllerDelegate.
TEST_F(BrowserDownloadServiceFactoryTest, Delegate) {
  web::DownloadController* download_controller =
      web::DownloadController::FromBrowserState(browser_state_.get());
  ASSERT_TRUE(download_controller);

  BrowserDownloadService* service =
      BrowserDownloadServiceFactory::GetForBrowserState(browser_state_.get());
  EXPECT_EQ(service, download_controller->GetDelegate());
}