Chromium Code Reviews| Index: ash/common/system/chromeos/palette/tools/create_note_unittest.cc |
| diff --git a/ash/common/system/chromeos/palette/tools/create_note_unittest.cc b/ash/common/system/chromeos/palette/tools/create_note_unittest.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..3050c0fc510b12afbfe319d1f1f534c91dc6dd04 |
| --- /dev/null |
| +++ b/ash/common/system/chromeos/palette/tools/create_note_unittest.cc |
| @@ -0,0 +1,85 @@ |
| +// Copyright 2016 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 "ash/common/system/chromeos/palette/palette_ids.h" |
| +#include "ash/common/system/chromeos/palette/palette_tool.h" |
| +#include "ash/common/system/chromeos/palette/tools/create_note_action.h" |
| +#include "ash/common/test/test_palette_delegate.h" |
| +#include "ash/common/wm_shell.h" |
| +#include "ash/test/ash_test_base.h" |
| +#include "base/macros.h" |
| +#include "base/memory/ptr_util.h" |
| +#include "testing/gmock/include/gmock/gmock.h" |
| + |
| +using namespace ash; |
|
James Cook
2016/08/16 22:08:44
nit: just wrap the file in
namespace ash {
}
unl
jdufault
2016/08/16 22:45:52
Done.
|
| + |
| +namespace { |
| + |
| +// Mock PaletteTool::Delegate class. |
| +class MockPaletteToolDelegate : public PaletteTool::Delegate { |
| + public: |
| + MockPaletteToolDelegate() {} |
| + ~MockPaletteToolDelegate() {} |
|
James Cook
2016/08/16 22:08:45
override
jdufault
2016/08/16 22:45:52
Done.
|
| + |
| + MOCK_METHOD1(EnableTool, void(PaletteToolId tool_id)); |
| + MOCK_METHOD1(DisableTool, void(PaletteToolId tool_id)); |
| + MOCK_METHOD0(HidePalette, void()); |
| + MOCK_METHOD0(GetWindow, WmWindow*()); |
| +}; |
| + |
| +// Base class for all create note ash tests. |
| +class CreateNoteTest : public test::AshTestBase { |
| + public: |
| + CreateNoteTest() {} |
| + ~CreateNoteTest() override {} |
| + |
| + void SetUp() override { |
| + test::AshTestBase::SetUp(); |
| + |
| + WmShell::Get()->SetPaletteDelegateForTesting( |
| + base::MakeUnique<TestPaletteDelegate>()); |
| + |
| + palette_tool_delegate = base::MakeUnique<MockPaletteToolDelegate>(); |
| + tool = base::MakeUnique<CreateNoteAction>(palette_tool_delegate.get()); |
| + } |
| + |
| + TestPaletteDelegate* test_palette_delegate() { |
| + return static_cast<TestPaletteDelegate*>( |
| + WmShell::Get()->palette_delegate()); |
| + } |
| + |
| + std::unique_ptr<MockPaletteToolDelegate> palette_tool_delegate; |
|
James Cook
2016/08/16 22:08:44
palette_tool_delegate_
jdufault
2016/08/16 22:45:51
Done; moved to protected as well.
|
| + std::unique_ptr<PaletteTool> tool; |
|
James Cook
2016/08/16 22:08:44
tool_
jdufault
2016/08/16 22:45:52
Done; moved to protected as well.
|
| + |
| + private: |
| + DISALLOW_COPY_AND_ASSIGN(CreateNoteTest); |
| +}; |
| + |
| +} // namespace |
| + |
| +// The note tool is only visible when there is a note-taking app available. |
| +TEST_F(CreateNoteTest, ViewOnlyCreatedWhenNoteAppIsAvailable) { |
| + test_palette_delegate()->set_has_note_app(false); |
| + EXPECT_FALSE(tool->CreateView()); |
| + tool->OnViewDestroyed(); |
| + |
| + test_palette_delegate()->set_has_note_app(true); |
| + std::unique_ptr<views::View> view = base::WrapUnique(tool->CreateView()); |
| + EXPECT_TRUE(view); |
| + tool->OnViewDestroyed(); |
| +} |
| + |
| +// Activating the note tool both creates a note via the delegate and also |
| +// disables the tool and hides the palette. |
| +TEST_F(CreateNoteTest, EnablingToolCreatesNewNoteAndDisablesTool) { |
| + test_palette_delegate()->set_has_note_app(true); |
| + std::unique_ptr<views::View> view = base::WrapUnique(tool->CreateView()); |
| + |
| + EXPECT_CALL(*palette_tool_delegate.get(), |
| + DisableTool(PaletteToolId::CREATE_NOTE)); |
| + EXPECT_CALL(*palette_tool_delegate.get(), HidePalette()); |
| + |
| + tool->OnEnable(); |
| + EXPECT_EQ(1, test_palette_delegate()->create_note_count()); |
| +} |