remoting/host/security_key/fake_security_key_message_writer.h

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

#ifndef REMOTING_HOST_SECURITY_KEY_FAKE_SECURITY_KEY_MESSAGE_WRITER_H_
#define REMOTING_HOST_SECURITY_KEY_FAKE_SECURITY_KEY_MESSAGE_WRITER_H_

#include <string>

#include "base/callback.h"
#include "base/macros.h"
#include "base/memory/weak_ptr.h"
#include "remoting/host/security_key/security_key_message.h"
#include "remoting/host/security_key/security_key_message_writer.h"

namespace remoting {

// Simulates the SecurityKeyMessageWriter and provides access to data
// members for testing.
class FakeSecurityKeyMessageWriter : public SecurityKeyMessageWriter {
 public:
  explicit FakeSecurityKeyMessageWriter(const base::Closure& write_callback);
  ~FakeSecurityKeyMessageWriter() override;

  // SecurityKeyMessageWriter interface.
  bool WriteMessage(SecurityKeyMessageType message_type) override;
  bool WriteMessageWithPayload(SecurityKeyMessageType message_type,
                               const std::string& message_payload) override;

  base::WeakPtr<FakeSecurityKeyMessageWriter> AsWeakPtr();

  SecurityKeyMessageType last_message_type() { return last_message_type_; }

  const std::string& last_message_payload() { return last_message_payload_; }

  void set_write_request_succeeded(bool should_succeed) {
    write_request_succeeded_ = should_succeed;
  }

 private:
  // Tracks the last message_type value written.
  SecurityKeyMessageType last_message_type_ = SecurityKeyMessageType::INVALID;

  // Tracks the last message_payload value written.
  std::string last_message_payload_;

  // This value is returned by the WriteMessage* functions above.
  bool write_request_succeeded_ = true;

  // Signaled whenever a write is requested.
  base::Closure write_callback_;

  base::WeakPtrFactory<FakeSecurityKeyMessageWriter> weak_factory_;

  DISALLOW_COPY_AND_ASSIGN(FakeSecurityKeyMessageWriter);
};

}  // namespace remoting

#endif  // REMOTING_HOST_SECURITY_KEY_FAKE_SECURITY_KEY_MESSAGE_WRITER_H_