Add UMA stats for audio/video containers.

media/base/container_names_unittest.cc

+// Copyright (c) 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.
+
+#include "base/file_util.h"
+#include "media/base/container_names.h"
+#include "media/base/test_data_util.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace media {
+
+namespace container_names {
+
+// Using a macros to simplify tests. Since EXPECT_EQ outputs the second argument
+// as a string when it fails, this lets the output identify what item actually
+// failed.
+#define VERIFY(buffer, name)                                           \
+  EXPECT_EQ(name,                                                      \
+            DetermineContainer(reinterpret_cast<const uint8*>(buffer), \
+                               sizeof(buffer)))
+
+// Test that small buffers are handled correctly.
+TEST(ContainerNamesTest, CheckSmallBuffer) {
+  // Empty buffer.
+  char buffer[0];
+  VERIFY(buffer, CONTAINER_UNKNOWN);
+
+  // Try a simple SRT file.
+  char buffer1[] =
+      "1\n"
+      "00:03:23,550 --> 00:03:24,375\n"
+      "You always had a hard time finding your place in this world.\n"
+      "\n"
+      "2\n"
+      "00:03:24,476 --> 00:03:25,175\n"
+      "What are you talking about?\n";
+  VERIFY(buffer1, CONTAINER_SRT);
+
+  // HLS has it's own loop.
+  char buffer2[] = "#EXTM3U"
+                   "some other random stuff"
+                   "#EXT-X-MEDIA-SEQUENCE:";
+  VERIFY(buffer2, CONTAINER_HLS);
+
+  // Try a large buffer all zeros.
+  char buffer3[4096];
+  memset(buffer3, 0, sizeof(buffer3));
+  VERIFY(buffer3, CONTAINER_UNKNOWN);
+
+  // Reuse buffer, but all \n this time.
+  memset(buffer3, '\n', sizeof(buffer3));
+  VERIFY(buffer3, CONTAINER_UNKNOWN);
+}
+
+#define BYTE_ORDER_MARK "\xef\xbb\xbf"
+
+// Note that the comparisons need at least 12 bytes, so make sure the buffer is
+// at least that size.
+const char kAmrBuffer[12] = "#!AMR";
+uint8 kAsfBuffer[] = { 0x30, 0x26, 0xb2, 0x75, 0x8e, 0x66, 0xcf, 0x11, 0xa6,
+                       0xd9, 0x00, 0xaa, 0x00, 0x62, 0xce, 0x6c };
+const char kAss1Buffer[] = "[Script Info]";
+const char kAss2Buffer[] = BYTE_ORDER_MARK "[Script Info]";
+uint8 kCafBuffer[] = { 'c', 'a', 'f', 'f', 0, 1, 0, 0, 'd', 'e', 's', 'c', 0, 0,
+                       0, 0, 0, 0, 0, 32, 64, 229, 136, 128, 0, 0, 0, 0, 'a',
+                       'a', 'c', ' ', 0, 0, 0, 2, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0,
+                       0, 2, 0, 0, 0, 0 };
+const char kDtshdBuffer[12] = "DTSHDHDR";
+const char kDxaBuffer[16] = "DEXA";
+const char kFlacBuffer[12] = "fLaC";
+uint8 kFlvBuffer[12] = { 'F', 'L', 'V', 0, 0, 0, 0, 1, 0, 0, 0, 0 };
+uint8 kIrcamBuffer[] = { '\x64', '\xa3', 1, 0, 0, 0, 0, 1, 0, 0, 0, 1 };
+const char kRm1Buffer[12] = ".RMF\0\0";
+const char kRm2Buffer[12] = ".ra\xfd";
+uint8 kWtvBuffer[] = { 0xb7, 0xd8, 0x00, 0x20, 0x37, 0x49, 0xda, 0x11, 0xa6,
+                       0x4e, 0x00, 0x07, 0xe9, 0x5e, 0xad, 0x8d };
+
+// Test that containers that start with fixed strings are handled correctly.
+// This is to verify that the TAG matches the first 4 characters of the string.
+TEST(ContainerNamesTest, CheckFixedStrings) {
+  VERIFY(kAmrBuffer, CONTAINER_AMR);
+  VERIFY(kAsfBuffer, CONTAINER_ASF);
+  VERIFY(kAss1Buffer, CONTAINER_ASS);
+  VERIFY(kAss2Buffer, CONTAINER_ASS);
+  VERIFY(kCafBuffer, CONTAINER_CAF);
+  VERIFY(kDtshdBuffer, CONTAINER_DTSHD);
+  VERIFY(kDxaBuffer, CONTAINER_DXA);
+  VERIFY(kFlacBuffer, CONTAINER_FLAC);
+  VERIFY(kFlvBuffer, CONTAINER_FLV);
+  VERIFY(kIrcamBuffer, CONTAINER_IRCAM);
+  VERIFY(kRm1Buffer, CONTAINER_RM);
+  VERIFY(kRm2Buffer, CONTAINER_RM);
+  VERIFY(kWtvBuffer, CONTAINER_WTV);
+}
+
+// Determine the container type of a specified file.
+void TestFile(FFmpegContainerName expected, const base::FilePath& filename) {
+  char buffer[8192];
+  size_t read = file_util::ReadFile(filename, buffer, sizeof(buffer));
+
+  // Now verify the type.
+  EXPECT_EQ(expected,
+            DetermineContainer(reinterpret_cast<const uint8*>(buffer), read))
+      << "Failure with file " << filename.value();
+}
+
+// Test several OGG files to ensure that the container is detected properly.
+TEST(ContainerNamesTest, FileCheckOGG) {
+  TestFile(CONTAINER_OGG, GetTestDataFilePath("bear.ogv"));
+  TestFile(CONTAINER_OGG, GetTestDataFilePath("9ch.ogg"));
+}
+
+// Test several WAV files to ensure that the container is detected properly.
+TEST(ContainerNamesTest, FileCheckWAV) {
+  TestFile(CONTAINER_WAV, GetTestDataFilePath("4ch.wav"));
+  TestFile(CONTAINER_WAV, GetTestDataFilePath("sfx_f32le.wav"));
+  TestFile(CONTAINER_WAV, GetTestDataFilePath("sfx_s16le.wav"));
+}
+
+// Test several MOV files to ensure that the container is detected properly.
+TEST(ContainerNamesTest, FileCheckMOV) {
+  TestFile(CONTAINER_MOV, GetTestDataFilePath("bear-1280x720.mp4"));
+  TestFile(CONTAINER_MOV, GetTestDataFilePath("sfx.m4a"));
+}
+
+// Test several WEBM files to ensure that the container is detected properly.
+TEST(ContainerNamesTest, FileCheckWEBM) {
+  TestFile(CONTAINER_WEBM, GetTestDataFilePath("bear-320x240.webm"));
+  TestFile(CONTAINER_WEBM, GetTestDataFilePath("no_streams.webm"));
+  TestFile(CONTAINER_WEBM, GetTestDataFilePath("webm_ebml_element"));
+}
+
+// Test several MP3 files to ensure that the container is detected properly.
+TEST(ContainerNamesTest, FileCheckMP3) {
+  TestFile(CONTAINER_MP3, GetTestDataFilePath("id3_test.mp3"));
+  TestFile(CONTAINER_MP3, GetTestDataFilePath("sfx.mp3"));
+}
+
+// Try a few non containers.
+TEST(ContainerNamesTest, FileCheckUNKNOWN) {
+  TestFile(CONTAINER_UNKNOWN, GetTestDataFilePath("ten_byte_file"));
+  TestFile(CONTAINER_UNKNOWN, GetTestDataFilePath("README"));
+  TestFile(CONTAINER_UNKNOWN, GetTestDataFilePath("bali_640x360_P422.yuv"));
+  TestFile(CONTAINER_UNKNOWN, GetTestDataFilePath("bali_640x360_RGB24.rgb"));
+  TestFile(CONTAINER_UNKNOWN, GetTestDataFilePath("webm_vp8_track_entry"));
+}
+
+}  // namespace container_names
+
+}  // namespace media