media/formats/mp2t/es_parser_adts_unittest.cc

   1 // Copyright 2014 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
   5 #include <vector>
   6
   7 #include "base/bind.h"
   8 #include "base/logging.h"
   9 #include "base/time/time.h"
  10 #include "media/base/buffers.h"
  11 #include "media/base/stream_parser_buffer.h"
  12 #include "media/formats/mp2t/es_parser_adts.h"
  13 #include "media/formats/mp2t/es_parser_test_base.h"
  14 #include "testing/gtest/include/gtest/gtest.h"
  15
  16 namespace media {
  17 class AudioDecoderConfig;
  18
  19 namespace mp2t {
  20
  21 class EsParserAdtsTest : public EsParserTestBase,
  22                          public testing::Test {
  23  public:
  24   EsParserAdtsTest();
  25   virtual ~EsParserAdtsTest() {}
  26
  27  protected:
  28   bool Process(const std::vector<Packet>& pes_packets, bool force_timing);
  29
  30  private:
  31   DISALLOW_COPY_AND_ASSIGN(EsParserAdtsTest);
  32 };
  33
  34 EsParserAdtsTest::EsParserAdtsTest() {
  35 }
  36
  37 bool EsParserAdtsTest::Process(
  38     const std::vector<Packet>& pes_packets,
  39     bool force_timing) {
  40   EsParserAdts es_parser(
  41       base::Bind(&EsParserAdtsTest::NewAudioConfig, base::Unretained(this)),
  42       base::Bind(&EsParserAdtsTest::EmitBuffer, base::Unretained(this)),
  43       false);
  44   return ProcessPesPackets(&es_parser, pes_packets, force_timing);
  45 }
  46
  47 TEST_F(EsParserAdtsTest, NoInitialPts) {
  48   LoadStream("bear.adts");
  49   std::vector<Packet> pes_packets = GenerateFixedSizePesPacket(512);
  50   // Process should succeed even without timing info, we should just skip the
  51   // audio frames without timing info, but still should be able to parse and
  52   // play the stream after that.
  53   EXPECT_TRUE(Process(pes_packets, false));
  54   EXPECT_EQ(1u, config_count_);
  55   EXPECT_EQ(0u, buffer_count_);
  56 }
  57
  58 TEST_F(EsParserAdtsTest, SinglePts) {
  59   LoadStream("bear.adts");
  60
  61   std::vector<Packet> pes_packets = GenerateFixedSizePesPacket(512);
  62   pes_packets.front().pts = base::TimeDelta::FromSeconds(10);
  63
  64   EXPECT_TRUE(Process(pes_packets, false));
  65   EXPECT_EQ(1u, config_count_);
  66   EXPECT_EQ(45u, buffer_count_);
  67 }
  68
  69 }  // namespace mp2t
  70 }  // namespace media
  71