Changed TestShellWebMimeRegistryImpl to blacklist rather than whitelist containers and codecs.

webkit/tools/test_shell/test_shell_webmimeregistry_impl.cc

 // Copyright (c) 2010 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 "webkit/tools/test_shell/test_shell_webmimeregistry_impl.h"
 
+#include "base/basictypes.h"
 #include "base/string_util.h"
 #include "net/base/mime_util.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebString.h"
 
 using WebKit::WebString;
 using WebKit::WebMimeRegistry;
 
 namespace {
 
 // Convert a WebString to ASCII, falling back on an empty string in the case
 // of a non-ASCII string.
 std::string ToASCIIOrEmpty(const WebString& string) {
   if (!IsStringASCII(string))
     return std::string();
   return UTF16ToASCII(string);
 }
 
+// Contains the types in mime_util.cc:supported_media_types that are
+// conditionally compiled based on defined(GOOGLE_CHROME_BUILD) or
+// defined(USE_PROPRIETARY_CODECS).
+static const char* const conditional_media_types[] = {

[scherkus (not reviewing), 2012/04/02 19:22:25]

s/conditional/blacklisted/ ?

also the comments should describe why we're blacklisting these codecs (it's to
make sure Google Chrome tests run the same as Chromium)

[ddorwin, 2012/04/02 19:44:42]

Done.

+  // MPEG-4.
+  "video/mp4",
+  "video/x-m4v",
+  "audio/mp4",
+  "audio/x-m4a",
+
+  // MP3.
+  "audio/mp3",
+  "audio/x-mp3",
+  "audio/mpeg",
+};
+
+// Contains the types in mime_util.cc:supported_media_codecs that are
+// conditionally compiled based on defined(GOOGLE_CHROME_BUILD) or
+// defined(USE_PROPRIETARY_CODECS).
+static const char* const conditional_media_codecs[] = {
+    "avc1",
+    "mp4a",
+};
+
 }  // namespace
 
 TestShellWebMimeRegistryImpl::TestShellWebMimeRegistryImpl() {
-  // Claim we support Ogg+Theora/Vorbis.
-  media_map_.insert("video/ogg");
-  media_map_.insert("audio/ogg");
-  media_map_.insert("application/ogg");
-  codecs_map_.insert("theora");
-  codecs_map_.insert("vorbis");
+  for (size_t i = 0; i < arraysize(conditional_media_types); ++i)
+    conditional_media_map_.insert(conditional_media_types[i]);
 
-  // Claim we support WAV.
-  media_map_.insert("audio/wav");
-  media_map_.insert("audio/x-wav");
-  codecs_map_.insert("1");  // PCM for WAV.
+  for (size_t i = 0; i < arraysize(conditional_media_codecs); ++i)
+    conditional_codecs_map_.insert(conditional_media_codecs[i]);
 }
 
 TestShellWebMimeRegistryImpl::~TestShellWebMimeRegistryImpl() {}
 
+// Returns IsNotSupported if mime_type or any of the codecs are not supported.
+// Otherwse, defers to the real registry.
 WebMimeRegistry::SupportsType
 TestShellWebMimeRegistryImpl::supportsMediaMIMEType(
     const WebString& mime_type, const WebString& codecs) {
-  // Not supporting the container is a flat-out no.
-  if (!IsSupportedMediaMimeType(ToASCIIOrEmpty(mime_type).c_str()))
+  if (IsConditionalMediaMimeType(ToASCIIOrEmpty(mime_type).c_str()))
     return IsNotSupported;
 
-  // If we don't recognize the codec, it's possible we support it.
   std::vector<std::string> parsed_codecs;
   net::ParseCodecString(ToASCIIOrEmpty(codecs).c_str(), &parsed_codecs, true);
-  if (!AreSupportedMediaCodecs(parsed_codecs))
-    return MayBeSupported;
+  if (HasConditionalMediaCodecs(parsed_codecs))
+    return IsNotSupported;
 
-  // Otherwise we have a perfect match.
-  return IsSupported;
+  return SimpleWebMimeRegistryImpl::supportsMediaMIMEType(mime_type, codecs);
 }
 
-bool TestShellWebMimeRegistryImpl::IsSupportedMediaMimeType(
+bool TestShellWebMimeRegistryImpl::IsConditionalMediaMimeType(
     const std::string& mime_type) {
-  return media_map_.find(mime_type) != media_map_.end();
+  return conditional_media_map_.find(mime_type) != conditional_media_map_.end();
 }
 
-bool TestShellWebMimeRegistryImpl::AreSupportedMediaCodecs(
+bool TestShellWebMimeRegistryImpl::HasConditionalMediaCodecs(
     const std::vector<std::string>& codecs) {
   for (size_t i = 0; i < codecs.size(); ++i) {
-    if (codecs_map_.find(codecs[i]) == codecs_map_.end()) {
-      return false;
+    if (conditional_codecs_map_.find(codecs[i]) !=
+        conditional_codecs_map_.end()) {
+      return true;
     }
   }
-  return true;
+  return false;
 }