xref: /external/chromium_org/net/tools/flip_server/flip_test_utils.h

// Copyright 2013 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 NET_TOOLS_FLIP_SERVER_FLIP_TEST_UTILS_H_
#define NET_TOOLS_FLIP_SERVER_FLIP_TEST_UTILS_H_

#include <string>

#include "net/tools/flip_server/sm_interface.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace net {

class MockSMInterface : public SMInterface {
 public:
  MockSMInterface();
  virtual ~MockSMInterface();

  MOCK_METHOD2(InitSMInterface, void(SMInterface*, int32));
  MOCK_METHOD8(InitSMConnection,
               void(SMConnectionPoolInterface*,
                    SMInterface*,
                    EpollServer*,
                    int,
                    std::string,
                    std::string,
                    std::string,
                    bool));
  MOCK_METHOD2(ProcessReadInput, size_t(const char*, size_t));
  MOCK_METHOD2(ProcessWriteInput, size_t(const char*, size_t));
  MOCK_METHOD1(SetStreamID, void(uint32 stream_id));
  MOCK_CONST_METHOD0(MessageFullyRead, bool());
  MOCK_CONST_METHOD0(Error, bool());
  MOCK_CONST_METHOD0(ErrorAsString, const char*());
  MOCK_METHOD0(Reset, void());
  MOCK_METHOD1(ResetForNewInterface, void(int32 server_idx));
  MOCK_METHOD0(ResetForNewConnection, void());
  MOCK_METHOD0(Cleanup, void());
  MOCK_METHOD0(PostAcceptHook, int());
  MOCK_METHOD3(NewStream, void(uint32, uint32, const std::string&));
  MOCK_METHOD1(SendEOF, void(uint32 stream_id));
  MOCK_METHOD1(SendErrorNotFound, void(uint32 stream_id));
  MOCK_METHOD2(SendSynStream, size_t(uint32, const BalsaHeaders&));
  MOCK_METHOD2(SendSynReply, size_t(uint32, const BalsaHeaders&));
  MOCK_METHOD5(SendDataFrame, void(uint32, const char*, int64, uint32, bool));
  MOCK_METHOD0(GetOutput, void());
  MOCK_METHOD0(set_is_request, void());
};

}  // namespace net

#endif  // NET_TOOLS_FLIP_SERVER_FLIP_TEST_UTILS_H_