file_data_source_unittest.cc 2.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889
  1. // Copyright (c) 2012 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include <stdint.h>
  5. #include <string>
  6. #include "base/base_paths.h"
  7. #include "base/bind.h"
  8. #include "base/files/file_path.h"
  9. #include "base/path_service.h"
  10. #include "base/strings/utf_string_conversions.h"
  11. #include "media/base/test_helpers.h"
  12. #include "media/filters/file_data_source.h"
  13. using ::testing::NiceMock;
  14. using ::testing::StrictMock;
  15. namespace media {
  16. class ReadCBHandler {
  17. public:
  18. ReadCBHandler() = default;
  19. ReadCBHandler(const ReadCBHandler&) = delete;
  20. ReadCBHandler& operator=(const ReadCBHandler&) = delete;
  21. MOCK_METHOD1(ReadCB, void(int size));
  22. };
  23. // Returns a path to the test file which contains the string "0123456789"
  24. // without the quotes or any trailing space or null termination. The file lives
  25. // under the media/test/data directory. Under Windows, strings for the
  26. // FilePath class are unicode, and the pipeline wants char strings. Convert
  27. // the string to UTF8 under Windows. For Mac and Linux, file paths are already
  28. // chars so just return the string from the base::FilePath.
  29. base::FilePath TestFileURL() {
  30. base::FilePath data_dir;
  31. EXPECT_TRUE(base::PathService::Get(base::DIR_SOURCE_ROOT, &data_dir));
  32. data_dir = data_dir.Append(FILE_PATH_LITERAL("media"))
  33. .Append(FILE_PATH_LITERAL("test"))
  34. .Append(FILE_PATH_LITERAL("data"))
  35. .Append(FILE_PATH_LITERAL("ten_byte_file"));
  36. return data_dir;
  37. }
  38. // Use the mock filter host to directly call the Read and GetPosition methods.
  39. TEST(FileDataSourceTest, ReadData) {
  40. int64_t size;
  41. uint8_t ten_bytes[10];
  42. // Create our mock filter host and initialize the data source.
  43. FileDataSource data_source;
  44. EXPECT_TRUE(data_source.Initialize(TestFileURL()));
  45. EXPECT_TRUE(data_source.GetSize(&size));
  46. EXPECT_EQ(10, size);
  47. ReadCBHandler handler;
  48. EXPECT_CALL(handler, ReadCB(10));
  49. data_source.Read(
  50. 0, 10, ten_bytes,
  51. base::BindOnce(&ReadCBHandler::ReadCB, base::Unretained(&handler)));
  52. EXPECT_EQ('0', ten_bytes[0]);
  53. EXPECT_EQ('5', ten_bytes[5]);
  54. EXPECT_EQ('9', ten_bytes[9]);
  55. EXPECT_CALL(handler, ReadCB(1));
  56. data_source.Read(
  57. 9, 1, ten_bytes,
  58. base::BindOnce(&ReadCBHandler::ReadCB, base::Unretained(&handler)));
  59. EXPECT_EQ('9', ten_bytes[0]);
  60. EXPECT_CALL(handler, ReadCB(0));
  61. data_source.Read(
  62. 10, 10, ten_bytes,
  63. base::BindOnce(&ReadCBHandler::ReadCB, base::Unretained(&handler)));
  64. EXPECT_CALL(handler, ReadCB(5));
  65. data_source.Read(
  66. 5, 10, ten_bytes,
  67. base::BindOnce(&ReadCBHandler::ReadCB, base::Unretained(&handler)));
  68. EXPECT_EQ('5', ten_bytes[0]);
  69. data_source.Stop();
  70. }
  71. } // namespace media