Created a new class SpdyTestStateHelper to serve as a helper to manage the state of a number of SPD…

net/spdy/spdy_websocket_stream_spdy2_unittest.cc

 // Copyright (c) 2012 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 "net/spdy/spdy_websocket_stream.h"
 
 #include <string>
 #include <vector>
 
 #include "base/bind.h"
@@ skipping 171 matching lines @@
 
   void DoSync(SpdyWebSocketStreamEvent* event) {
     sync_callback_.SetResult(OK);
   }
 
  protected:
   SpdyWebSocketStreamSpdy2Test() {}
   virtual ~SpdyWebSocketStreamSpdy2Test() {}
 
   virtual void SetUp() {
-    EnableCompression(false);
     SpdySession::set_default_protocol(SSLClientSocket::kProtoSPDY2);
 
     host_port_pair_.set_host("example.com");
     host_port_pair_.set_port(80);
     host_port_proxy_pair_.first = host_port_pair_;
     host_port_proxy_pair_.second = ProxyServer::Direct();
 
     const size_t max_concurrent_streams = 1;
     spdy::SettingsFlagsAndId id(spdy::SETTINGS_FLAG_PLEASE_PERSIST,
                                 spdy::SETTINGS_MAX_CONCURRENT_STREAMS);
     spdy_settings_to_set_.push_back(
         spdy::SpdySetting(id, max_concurrent_streams));
 
     spdy::SettingsFlagsAndId id1(spdy::SETTINGS_FLAG_PERSISTED,
                                  spdy::SETTINGS_MAX_CONCURRENT_STREAMS);
     spdy_settings_to_send_.push_back(
         spdy::SpdySetting(id1, max_concurrent_streams));
   }
 
   virtual void TearDown() {
     MessageLoop::current()->RunAllPending();
   }
 
-  void EnableCompression(bool enabled) {
-    spdy::SpdyFramer::set_enable_compression_default(enabled);
-  }
   void Prepare(spdy::SpdyStreamId stream_id) {
     stream_id_ = stream_id;
 
     const char* const request_headers[] = {
       "url", "ws://example.com/echo",
       "origin", "http://example.com/wsdemo",
     };
 
     int request_header_count = arraysize(request_headers) / 2;
 
@@ skipping 81 matching lines @@
   scoped_ptr<spdy::SpdyFrame> closing_frame_;
   HostPortPair host_port_pair_;
   HostPortProxyPair host_port_proxy_pair_;
   TestCompletionCallback completion_callback_;
   TestCompletionCallback sync_callback_;
 
   static const char kMessageFrame[];
   static const char kClosingFrame[];
   static const size_t kMessageFrameLength;
   static const size_t kClosingFrameLength;
+
+ private:
+  SpdyTestStateHelper spdy_state_;
 };
 
 const char SpdyWebSocketStreamSpdy2Test::kMessageFrame[] = "\0hello\xff";
 const char SpdyWebSocketStreamSpdy2Test::kClosingFrame[] = "\xff\0";
 const size_t SpdyWebSocketStreamSpdy2Test::kMessageFrameLength =
     arraysize(SpdyWebSocketStreamSpdy2Test::kMessageFrame) - 1;
 const size_t SpdyWebSocketStreamSpdy2Test::kClosingFrameLength =
     arraysize(SpdyWebSocketStreamSpdy2Test::kClosingFrame) - 1;
 
 TEST_F(SpdyWebSocketStreamSpdy2Test, Basic) {
@@ skipping 291 matching lines @@
   EXPECT_EQ(OK, events[7].result);
 
   // EOF close SPDY session.
   EXPECT_TRUE(!http_session_->spdy_session_pool()->HasSession(
       host_port_proxy_pair_));
   EXPECT_TRUE(data()->at_read_eof());
   EXPECT_TRUE(data()->at_write_eof());
 }
 
 }  // namespace net