mirror of
https://github.com/ckaczor/sqltoolsservice.git
synced 2026-01-14 09:59:48 -05:00
f288bee294
* Make nullable warnings a per file opt-in * Remove unneeded compiler directives * Remove compiler directive for User Data
36 lines
1.0 KiB
C#
36 lines
1.0 KiB
C#
//
|
|
// Copyright (c) Microsoft. All rights reserved.
|
|
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
|
|
//
|
|
|
|
#nullable disable
|
|
|
|
using System;
|
|
using System.Threading.Tasks;
|
|
using Microsoft.SqlTools.Hosting.Protocol;
|
|
using Microsoft.SqlTools.Hosting.Protocol.Contracts;
|
|
using Moq;
|
|
|
|
namespace Microsoft.SqlTools.ServiceLayer.UnitTests.Utility
|
|
{
|
|
public static class ProtocolEndpointMocks
|
|
{
|
|
public static Mock<IProtocolEndpoint> AddEventHandling<TParams>(
|
|
this Mock<IProtocolEndpoint> mock,
|
|
EventType<TParams> expectedEvent,
|
|
Action<EventType<TParams>, TParams> eventCallback)
|
|
{
|
|
var flow = mock.Setup(h => h.SendEvent(
|
|
It.Is<EventType<TParams>>(m => m == expectedEvent),
|
|
It.IsAny<TParams>()))
|
|
.Returns(Task.FromResult(0));
|
|
if (eventCallback != null)
|
|
{
|
|
flow.Callback(eventCallback);
|
|
}
|
|
|
|
return mock;
|
|
}
|
|
}
|
|
}
|