ash/common/system/chromeos/palette/tools/screenshot_unittest.cc - Issue 2732813002: chromeos: Move files in //ash/common to //ash, part 1

Side by Side Diff: ash/common/system/chromeos/palette/tools/screenshot_unittest.cc

Issue 2732813002: chromeos: Move files in //ash/common to //ash, part 1 (Closed)

Patch Set: rebase Created 3 years, 9 months ago

Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.

Jump to:

OLD	NEW
	(Empty)
1 // Copyright 2016 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 "ash/common/system/chromeos/palette/mock_palette_tool_delegate.h"

6 #include "ash/common/system/chromeos/palette/palette_ids.h"

7 #include "ash/common/system/chromeos/palette/palette_tool.h"

8 #include "ash/common/system/chromeos/palette/tools/capture_region_mode.h"

9 #include "ash/common/system/chromeos/palette/tools/capture_screen_action.h"

10 #include "ash/common/test/test_palette_delegate.h"

11 #include "ash/common/wm_shell.h"

12 #include "ash/test/ash_test_base.h"

13 #include "base/macros.h"

14 #include "base/memory/ptr_util.h"

15

16 namespace ash {

17

18 namespace {

19

20 // Base class for all create note ash tests.

21 class ScreenshotToolTest : public test::AshTestBase {

22 public:

23 ScreenshotToolTest() {}

24 ~ScreenshotToolTest() override {}

25

26 void SetUp() override {

27 test::AshTestBase::SetUp();

28

29 WmShell::Get()->SetPaletteDelegateForTesting(

30 base::MakeUnique<TestPaletteDelegate>());

31

32 palette_tool_delegate_ = base::MakeUnique<MockPaletteToolDelegate>();

33 }

34

35 TestPaletteDelegate* test_palette_delegate() {

36 return static_cast<TestPaletteDelegate*>(

37 WmShell::Get()->palette_delegate());

38 }

39

40 protected:

41 std::unique_ptr<MockPaletteToolDelegate> palette_tool_delegate_;

42

43 private:

44 DISALLOW_COPY_AND_ASSIGN(ScreenshotToolTest);

45 };

46

47 } // namespace

48

49 // Verifies that capturing a region triggers the partial screenshot delegate

50 // method. Invoking the callback passed to the delegate disables the tool.

51 TEST_F(ScreenshotToolTest, EnablingCaptureRegionCallsDelegateAndDisablesTool) {

52 std::unique_ptr<PaletteTool> tool =

53 base::MakeUnique<CaptureRegionMode>(palette_tool_delegate_.get());

54

55 // Starting a partial screenshot calls the calls the palette delegate to start

56 // a screenshot session and hides the palette.

57 EXPECT_CALL(*palette_tool_delegate_.get(), HidePalette());

58 tool->OnEnable();

59 EXPECT_EQ(1, test_palette_delegate()->take_partial_screenshot_count());

60 testing::Mock::VerifyAndClearExpectations(palette_tool_delegate_.get());

61

62 // Calling the associated callback (partial screenshot finished) will disable

63 // the tool.

64 EXPECT_CALL(*palette_tool_delegate_.get(),

65 DisableTool(PaletteToolId::CAPTURE_REGION));

66 test_palette_delegate()->partial_screenshot_done().Run();

67 }

68

69 // Verifies that capturing the screen triggers the screenshot delegate method,

70 // disables the tool, and hides the palette.

71 TEST_F(ScreenshotToolTest, EnablingCaptureScreenCallsDelegateAndDisablesTool) {

72 std::unique_ptr<PaletteTool> tool =

73 base::MakeUnique<CaptureScreenAction>(palette_tool_delegate_.get());

74 EXPECT_CALL(*palette_tool_delegate_.get(),

75 DisableTool(PaletteToolId::CAPTURE_SCREEN));

76 EXPECT_CALL(*palette_tool_delegate_.get(), HidePaletteImmediately());

77 tool->OnEnable();

78 EXPECT_EQ(1, test_palette_delegate()->take_screenshot_count());

79 }

80

81 } // namespace ash

OLD	NEW