Add unittest for H264BitReader.

content/common/gpu/media/h264_parser.h

 // 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.
 //
 // This file contains an implementation of an H264 Annex-B video stream parser.
 
 #ifndef CONTENT_COMMON_GPU_MEDIA_H264_PARSER_H_
 #define CONTENT_COMMON_GPU_MEDIA_H264_PARSER_H_
 
 #include <sys/types.h>
 
 #include <map>
 
 #include "base/basictypes.h"
 #include "content/common/content_export.h"
 
+class H264BitReaderTest;
+
 namespace content {
 
 // For explanations of each struct and its members, see H.264 specification
 // at http://www.itu.int/rec/T-REC-H.264.
 struct CONTENT_EXPORT H264NALU {
   H264NALU();
 
   enum Type {
     kUnspecified = 0,
     kNonIDRSlice = 1,
@@ skipping 267 matching lines @@
 
   // Parse a slice header, returning it in |*shdr|. |*nalu| must be set to
   // the NALU returned from AdvanceToNextNALU() and corresponding to |*shdr|.
   Result ParseSliceHeader(const H264NALU& nalu, H264SliceHeader* shdr);
 
   // Parse a SEI message, returning it in |*sei_msg|, provided and managed
   // by the caller.
   Result ParseSEI(H264SEIMessage* sei_msg);
 
  private:
+  friend class ::H264BitReaderTest;
+
   // A class to provide bit-granularity reading of H.264 streams.
   // This is not a generic bit reader class, as it takes into account
   // H.264 stream-specific constraints, such as skipping emulation-prevention
   // bytes and stop bits. See spec for more details.
-  // TODO(posciak): need separate unittests for this class.
   class H264BitReader {
    public:
     H264BitReader();
     ~H264BitReader();
 
     // Initialize the reader to start reading at |data|, |size| being size
     // of |data| in bytes.
     // Return false on insufficient size of stream..
     // TODO(posciak,fischman): consider replacing Initialize() with
     // heap-allocating and creating bit readers on demand instead.
@@ skipping 80 matching lines @@
   typedef std::map<int, H264PPS*> PPSById;
   SPSById active_SPSes_;
   PPSById active_PPSes_;
 
   DISALLOW_COPY_AND_ASSIGN(H264Parser);
 };
 
 }  // namespace content
 
 #endif  // CONTENT_COMMON_GPU_MEDIA_H264_PARSER_H_