Issue #1820 - Part 4: Add a gtest for testing the extraction function.

2 files changed, 112 insertions, 0 deletions

diff --git a/dom/media/mediasource/gtest/TestExtractVPXCodecDetails.cpp b/dom/media/mediasource/gtest/TestExtractVPXCodecDetails.cpp
new file mode 100644
index 0000000000..be8a120f61
--- /dev/null
+++ b/dom/media/mediasource/gtest/TestExtractVPXCodecDetails.cpp
@@ -0,0 +1,111 @@
+/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#include <gtest/gtest.h>
+#include <stdint.h>
+
+#include "VideoUtils.h"
+
+using namespace mozilla;
+
+struct TestData
+{
+  const char16_t* const mCodecParameterString;
+  const bool mExpectedValue;
+};
+
+TEST(ExtractVPXCodecDetails, TestDataLength) {
+  TestData tests[] =
+  {
+    {u"vp09.00.11.08", true }, // valid case
+    {u"vp09.00.11.08.00", true }, // valid case, have extra optional field
+    {u"vp09.02.10.10.01.09.16.09.01", true}, // maximum length valid case
+    {u"vp09", false }, // lack of mandatory fields
+    {u"vp09.00", false }, // lack of mandatory fields
+    {u"vp09.00.11", false }, // lack of mandatory fields
+    {u"vp09.02.10.10.01.09.16.09.01.00", false} // more than 9 fields, invalid case.
+  };
+
+  for (const auto& data : tests) {
+    uint8_t profile = 0;
+    uint8_t level = 0;
+    uint8_t bitDepth = 0;
+    bool result = ExtractVPXCodecDetails(nsString(data.mCodecParameterString), profile, level, bitDepth);
+    EXPECT_EQ(result, data.mExpectedValue) << NS_ConvertUTF16toUTF8(data.mCodecParameterString).get();
+  }
+}
+
+TEST(ExtractVPXCodecDetails, TestInputData) {
+  TestData tests[] =
+  {
+    {u"vp09.02..08", false}, // malformed
+    {u"vp9.02.10.08", false}, // invalid 4CC
+    {u"vp09.03.11.08", false }, // profile should < 3
+    {u"vp09.00.63.08.00", false }, // invalid level
+    {u"vp09.02.10.13", false}, // invalid bitDepth
+    {u"vp09.02.10.10.04", false}, // invalid chromasubsampling, should < 4
+    {u"vp09.02.10.10.01.00", false}, // invalid Colour primaries, should not be 0,3 or < 23.
+    {u"vp09.02.10.10.01.03", false}, // invalid Colour primaries.
+    {u"vp09.02.10.10.01.23", false}, // invalid Colour primaries.
+    {u"vp09.02.10.10.01.09.00", false}, // invalid Transfer characteristics, should not be 0,3 or < 19.
+    {u"vp09.02.10.10.01.09.03", false}, // invalid Transfer characteristics.
+    {u"vp09.02.10.10.01.09.19", false}, // invalid Transfer characteristics.
+    {u"vp09.02.10.10.01.09.16.12", false}, // invalid Matrix coefficients, should not be 3 or < 12.
+    {u"vp09.02.10.10.01.09.16.03", false}, // invalid matrix.
+    {u"vp09.02.10.10.01.09.16.09.02", false}, // invalid range, should < 2.
+    // Test if matrixCoefficients is 0 (RGB), then chroma subsampling MUST be 3 (4:4:4).
+    {u"vp09.02.10.08.03.09.16.00.00", true} // invalid combination.
+  };
+
+  for (const auto& data : tests) {
+    uint8_t profile = 0;
+    uint8_t level = 0;
+    uint8_t bitDepth = 0;
+    bool result = ExtractVPXCodecDetails(nsString(data.mCodecParameterString), profile, level, bitDepth);
+    EXPECT_EQ(result, data.mExpectedValue) << NS_ConvertUTF16toUTF8(data.mCodecParameterString).get();
+  }
+}
+
+TEST(ExtractVPXCodecDetails, TestParsingOutput) {
+    uint8_t profile = 0;
+    uint8_t level = 0;
+    uint8_t bitDepth = 0;
+    uint8_t chromaSubsampling = 0;
+    VideoColorSpace colorSpace;
+    auto data = u"vp09.01.11.08";
+    bool result = ExtractVPXCodecDetails(nsString(data),
+                                         profile,
+                                         level,
+                                         bitDepth,
+                                         chromaSubsampling,
+                                         colorSpace);
+    EXPECT_EQ(result, true);
+    EXPECT_EQ(profile, 1);
+    EXPECT_EQ(level, 11);
+    EXPECT_EQ(bitDepth, 8);
+    // Should keep spec defined default value.
+    EXPECT_EQ(chromaSubsampling, 1);
+    EXPECT_EQ(colorSpace.mPrimaryId, 1);
+    EXPECT_EQ(colorSpace.mTransferId, 1);
+    EXPECT_EQ(colorSpace.mMatrixId, 1);
+    EXPECT_EQ(colorSpace.mRangeId, 0);
+
+    data = u"vp09.02.10.10.01.09.16.09.01";
+    result = ExtractVPXCodecDetails(nsString(data),
+                                    profile,
+                                    level,
+                                    bitDepth,
+                                    chromaSubsampling,
+                                    colorSpace);
+    EXPECT_EQ(result, true);
+    EXPECT_EQ(profile, 2);
+    EXPECT_EQ(level, 10);
+    EXPECT_EQ(bitDepth, 10);
+    EXPECT_EQ(chromaSubsampling, 1);
+    EXPECT_EQ(colorSpace.mPrimaryId, 9);
+    EXPECT_EQ(colorSpace.mTransferId, 16);
+    EXPECT_EQ(colorSpace.mMatrixId, 9);
+    EXPECT_EQ(colorSpace.mRangeId, 1);
+}
diff --git a/dom/media/mediasource/gtest/moz.build b/dom/media/mediasource/gtest/moz.build
index 2cccc80a21..2a59a8343a 100644
--- a/dom/media/mediasource/gtest/moz.build
+++ b/dom/media/mediasource/gtest/moz.build
@@ -5,6 +5,7 @@
 
 UNIFIED_SOURCES += [
     'TestContainerParser.cpp',
+    'TestExtractVPXCodecDetails.cpp',
 ]
 
 LOCAL_INCLUDES += [