opentelemetry-dotnet/test/OpenTelemetry.Tests/Trace/SamplingResultTest.cs

// <copyright file="SamplingResultTest.cs" company="OpenTelemetry Authors">
// Copyright The OpenTelemetry Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// </copyright>
using Xunit;

namespace OpenTelemetry.Trace.Tests
{
    public class SamplingResultTest
    {
        [Theory]
        [InlineData(SamplingDecision.Drop)]
        [InlineData(SamplingDecision.RecordAndSample)]
        [InlineData(SamplingDecision.RecordOnly)]
        public void VerifyCtor_SamplingDecision(SamplingDecision decision)
        {
            var samplingResult = new SamplingResult(decision);
            Assert.Equal(decision, samplingResult.Decision);
            Assert.Empty(samplingResult.Attributes);
        }

        [Theory]
        [InlineData(false, SamplingDecision.Drop)]
        [InlineData(true, SamplingDecision.RecordAndSample)]
        public void VerifyCtor_Bool(bool isSampled, SamplingDecision expectedSamplingDecision)
        {
            var samplingResult = new SamplingResult(isSampled);
            Assert.Equal(expectedSamplingDecision, samplingResult.Decision);
            Assert.Empty(samplingResult.Attributes);
        }

        [Theory]
        [InlineData(SamplingDecision.Drop)]
        [InlineData(SamplingDecision.RecordAndSample)]
        [InlineData(SamplingDecision.RecordOnly)]
        public void VerifyCtor_SamplingDecisionAndAttributes(SamplingDecision decision)
        {
            var attributes = new Dictionary<string, object>
            {
                { "A", 1 },
                { "B", 2 },
                { "C", 3 },
            };

            var samplingResult = new SamplingResult(decision, attributes);
            Assert.Equal(decision, samplingResult.Decision);
            Assert.True(attributes.SequenceEqual(samplingResult.Attributes));
        }

        [Theory]
        [InlineData(SamplingDecision.Drop, true)]
        [InlineData(SamplingDecision.RecordAndSample, false)]
        [InlineData(SamplingDecision.RecordOnly, false)]
        public void VerifyOperator_Equals(SamplingDecision decision, bool expected)
        {
            var samplingResult1 = new SamplingResult(SamplingDecision.Drop);

            var samplingResult2 = new SamplingResult(decision);
            Assert.Equal(expected, samplingResult1 == samplingResult2);
        }

        [Theory]
        [InlineData(SamplingDecision.Drop, false)]
        [InlineData(SamplingDecision.RecordAndSample, true)]
        [InlineData(SamplingDecision.RecordOnly, true)]
        public void VerifyOperator_NotEquals(SamplingDecision decision, bool expected)
        {
            var samplingResult1 = new SamplingResult(SamplingDecision.Drop);

            var samplingResult2 = new SamplingResult(decision);
            Assert.Equal(expected, samplingResult1 != samplingResult2);
        }

        [Fact]
        public void Verify_Equals()
        {
            var samplingResult1 = new SamplingResult(SamplingDecision.Drop);
            Assert.True(samplingResult1.Equals(samplingResult1));
            Assert.True(samplingResult1.Equals((object)samplingResult1));

            var samplingResult2 = new SamplingResult(SamplingDecision.RecordAndSample);
            Assert.False(samplingResult1.Equals(samplingResult2));
            Assert.True(samplingResult2.Equals(samplingResult2));
            Assert.False(samplingResult1.Equals((object)samplingResult2));
            Assert.True(samplingResult2.Equals((object)samplingResult2));

            var samplingResult3 = new SamplingResult(
                SamplingDecision.RecordOnly,
                new Dictionary<string, object>
                {
                    { "A", 1 },
                    { "B", 2 },
                    { "C", 3 },
                });
            Assert.False(samplingResult1.Equals(samplingResult3));
            Assert.True(samplingResult3.Equals(samplingResult3));
            Assert.False(samplingResult1.Equals((object)samplingResult3));
            Assert.True(samplingResult3.Equals((object)samplingResult3));

            Assert.False(samplingResult1.Equals(Guid.Empty));
        }

        [Theory]
        [InlineData(SamplingDecision.Drop)]
        [InlineData(SamplingDecision.RecordAndSample)]
        [InlineData(SamplingDecision.RecordOnly)]
        public void Verify_GetHashCode(SamplingDecision decision)
        {
            var samplingResult1 = new SamplingResult(decision);
            var samplingResult2 = new SamplingResult(decision, new Dictionary<string, object>
                {
                    { "A", 1 },
                    { "B", 2 },
                    { "C", 3 },
                });

            Assert.NotEqual(samplingResult1.GetHashCode(), samplingResult2.GetHashCode());
        }
    }
}