data-api-builder/src/Service.Tests/UnitTests/DynamicLogLevelProviderTests.cs at 0e64f6254a3b86b5c93bed7ab7c133fea931d381 · Azure/data-api-builder · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.

#nullable enable

using Azure.DataApiBuilder.Service.Telemetry;
using Microsoft.Extensions.Logging;
using Microsoft.VisualStudio.TestTools.UnitTesting;

namespace Azure.DataApiBuilder.Service.Tests.UnitTests
{
    /// <summary>
    /// Unit tests for the DynamicLogLevelProvider class.
    /// Tests the MCP logging/setLevel support.
    /// </summary>
    [TestClass]
    public class DynamicLogLevelProviderTests
    {
        [TestMethod]
        public void UpdateFromMcp_ValidLevel_ChangesLogLevel()
        {
            // Arrange
            DynamicLogLevelProvider provider = new();
            provider.SetInitialLogLevel(LogLevel.Error, isCliOverridden: false);

            // Act
            bool result = provider.UpdateFromMcp("debug");

            // Assert
            Assert.IsTrue(result);
            Assert.AreEqual(LogLevel.Debug, provider.CurrentLogLevel);
        }

        [TestMethod]
        public void UpdateFromMcp_CliOverridden_DoesNotChangeLogLevel()
        {
            // Arrange
            DynamicLogLevelProvider provider = new();
            provider.SetInitialLogLevel(LogLevel.Error, isCliOverridden: true);

            // Act
            bool result = provider.UpdateFromMcp("debug");

            // Assert
            Assert.IsFalse(result);
            Assert.AreEqual(LogLevel.Error, provider.CurrentLogLevel);
        }

        [TestMethod]
        public void UpdateFromMcp_ConfigOverridden_DoesNotChangeLogLevel()
        {
            // Arrange
            DynamicLogLevelProvider provider = new();
            provider.SetInitialLogLevel(LogLevel.Warning, isCliOverridden: false, isConfigOverridden: true);

            // Act
            bool result = provider.UpdateFromMcp("debug");

            // Assert
            Assert.IsFalse(result);
            Assert.AreEqual(LogLevel.Warning, provider.CurrentLogLevel);
        }

        [TestMethod]
        public void UpdateFromMcp_InvalidLevel_ReturnsFalse()
        {
            // Arrange
            DynamicLogLevelProvider provider = new();
            provider.SetInitialLogLevel(LogLevel.Error, isCliOverridden: false);

            // Act
            bool result = provider.UpdateFromMcp("invalid");

            // Assert
            Assert.IsFalse(result);
            Assert.AreEqual(LogLevel.Error, provider.CurrentLogLevel);
        }

        [TestMethod]
        public void ShouldLog_ReturnsCorrectResult()
        {
            // Arrange
            DynamicLogLevelProvider provider = new();
            provider.SetInitialLogLevel(LogLevel.Warning, isCliOverridden: false);

            // Assert - logs at or above Warning should pass
            Assert.IsTrue(provider.ShouldLog(LogLevel.Warning));
            Assert.IsTrue(provider.ShouldLog(LogLevel.Error));
            Assert.IsFalse(provider.ShouldLog(LogLevel.Debug));
        }
    }
}