PicasaAlbumTableReader for Media Galleries API

webkit/fileapi/media/picasa/pmp_test_helper.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 WEBKIT_FILEAPI_MEDIA_PICASA_PMP_TEST_HELPER_H_
 #define WEBKIT_FILEAPI_MEDIA_PICASA_PMP_TEST_HELPER_H_
 
 #include <vector>
 
 #include "base/basictypes.h"
 #include "base/files/scoped_temp_dir.h"
+#include "webkit/fileapi/media/picasa/pmp_constants.h"
 
 namespace base {
 
 class FilePath;
 
 }  // namespace base
 
 namespace picasaimport {
 
 class PmpColumnReader;
 
 // A helper class used for unit tests only
 class PmpTestHelper {
  public:
   PmpTestHelper();
 
   bool Init();
 
   base::FilePath GetTempDirPath();
 
   template<class T>
   bool WriteColumnFileFromVector(std::string table_name,
                                  std::string column_name,
-                                 const uint16 field_type,
+                                 const PmpFieldType field_type,
                                  std::vector<T> elements_vector);
 
   bool InitColumnReaderFromBytes(PmpColumnReader* const reader,
-                                 std::vector<uint8> data, uint32* rows_read);
+                                 std::vector<uint8> data,
+                                 const PmpFieldType expected_type,
+                                 uint32* rows_read);
 
-  static std::vector<uint8> MakeHeader(const uint16 field_type,
+  static std::vector<uint8> MakeHeader(const PmpFieldType field_type,
                                        const uint32 row_count);
 
   template<class T>
-  static std::vector<uint8> MakeHeaderAndBody(const uint16 field_type,
+  static std::vector<uint8> MakeHeaderAndBody(const PmpFieldType field_type,
                                               const uint32 row_count,
                                               const std::vector<T>& elems);
 
  private:
   base::ScopedTempDir temp_dir_;
 };
 
 }  // namespace picasaimport
 
 #endif  // WEBKIT_FILEAPI_MEDIA_PICASA_PMP_TEST_HELPER_H_