components/test_runner/mock_color_chooser.h

// Copyright 2014 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.

#ifndef COMPONENTS_TEST_RUNNER_MOCK_COLOR_CHOOSER_H_
#define COMPONENTS_TEST_RUNNER_MOCK_COLOR_CHOOSER_H_

#include "base/basictypes.h"
#include "components/test_runner/web_task.h"
#include "third_party/WebKit/public/web/WebColorChooser.h"
#include "third_party/WebKit/public/web/WebColorChooserClient.h"

namespace test_runner {

class WebTestDelegate;
class WebTestProxyBase;

class MockColorChooser : public blink::WebColorChooser {
public:
    MockColorChooser(blink::WebColorChooserClient* client,
                     WebTestDelegate* delegate,
                     WebTestProxyBase* proxy);
    ~MockColorChooser() override;

    // blink::WebColorChooser implementation.
    void setSelectedColor(const blink::WebColor color) override;
    void endChooser() override;

    void InvokeDidEndChooser();
    WebTaskList* mutable_task_list() { return &task_list_; }

private:
    blink::WebColorChooserClient* client_;
    WebTestDelegate* delegate_;
    WebTestProxyBase* proxy_;
    WebTaskList task_list_;

    DISALLOW_COPY_AND_ASSIGN(MockColorChooser);
};

}  // namespace test_runner

#endif  // COMPONENTS_TEST_RUNNER_MOCK_COLOR_CHOOSER_H_