tcp_connection_for_testing.h

Go to the documentation of this file.

// Copyright 2016 Google Inc.

#ifndef PAGESPEED_SYSTEM_TCP_CONNECTION_FOR_TESTING_H_
#define PAGESPEED_SYSTEM_TCP_CONNECTION_FOR_TESTING_H_

#include "apr_pools.h"
#include "apr_network_io.h"
#include "pagespeed/kernel/base/basictypes.h"
#include "pagespeed/kernel/base/string.h"
#include "pagespeed/kernel/base/string_util.h"

namespace net_instaweb {

class TcpConnectionForTesting {
 public:
  TcpConnectionForTesting();
  ~TcpConnectionForTesting();

  bool Connect(const GoogleString& hostname, int port);

  void Send(StringPiece data);

  GoogleString ReadBytes(int length);

  GoogleString ReadLine() {
    return ReadUntil("\n");
  }

  GoogleString ReadLineCrLf() {
    return ReadUntil("\r\n");
  }

  GoogleString ReadUntil(StringPiece marker);

 private:
  apr_pool_t* pool_;
  apr_socket_t* socket_;

  
};

}  

#endif