net/dcsctp/packet/error_cause/protocol_violation_cause_test.cc

/*
 *  Copyright (c) 2021 The WebRTC project authors. All Rights Reserved.
 *
 *  Use of this source code is governed by a BSD-style license
 *  that can be found in the LICENSE file in the root of the source
 *  tree. An additional intellectual property rights grant can be found
 *  in the file PATENTS.  All contributing project authors may
 *  be found in the AUTHORS file in the root of the source tree.
 */
#include "net/dcsctp/packet/error_cause/protocol_violation_cause.h"

#include <stdint.h>

#include <type_traits>
#include <vector>

#include "api/array_view.h"
#include "net/dcsctp/packet/error_cause/error_cause.h"
#include "net/dcsctp/packet/tlv_trait.h"
#include "net/dcsctp/testing/testing_macros.h"
#include "rtc_base/gunit.h"
#include "test/gmock.h"

namespace dcsctp {
namespace {
using ::testing::SizeIs;

TEST(ProtocolViolationCauseTest, EmptyReason) {
  Parameters causes =
      Parameters::Builder().Add(ProtocolViolationCause("")).Build();

  ASSERT_HAS_VALUE_AND_ASSIGN(Parameters deserialized,
                              Parameters::Parse(causes.data()));
  ASSERT_THAT(deserialized.descriptors(), SizeIs(1));
  EXPECT_EQ(deserialized.descriptors()[0].type, ProtocolViolationCause::kType);

  ASSERT_HAS_VALUE_AND_ASSIGN(
      ProtocolViolationCause cause,
      ProtocolViolationCause::Parse(deserialized.descriptors()[0].data));

  EXPECT_EQ(cause.additional_information(), "");
}

TEST(ProtocolViolationCauseTest, SetReason) {
  Parameters causes = Parameters::Builder()
                          .Add(ProtocolViolationCause("Reason goes here"))
                          .Build();

  ASSERT_HAS_VALUE_AND_ASSIGN(Parameters deserialized,
                              Parameters::Parse(causes.data()));
  ASSERT_THAT(deserialized.descriptors(), SizeIs(1));
  EXPECT_EQ(deserialized.descriptors()[0].type, ProtocolViolationCause::kType);

  ASSERT_HAS_VALUE_AND_ASSIGN(
      ProtocolViolationCause cause,
      ProtocolViolationCause::Parse(deserialized.descriptors()[0].data));

  EXPECT_EQ(cause.additional_information(), "Reason goes here");
}
}  // namespace
}  // namespace dcsctp