88 lines
2.8 KiB
C#
88 lines
2.8 KiB
C#
// Copyright (c) Microsoft Open Technologies, Inc. All rights reserved. See License.txt in the project root for license information.
|
|
|
|
namespace System.Data.Entity
|
|
{
|
|
using System.Data.Common;
|
|
using Xunit;
|
|
|
|
public class ConnectionEventsTracker
|
|
{
|
|
private int countOpenClose;
|
|
private int countCloseOpen;
|
|
private int countOtherConnectionStates;
|
|
|
|
public ConnectionEventsTracker(DbConnection connection)
|
|
{
|
|
if (connection == null)
|
|
{
|
|
throw new ArgumentException("Cannot track events for a null connection!");
|
|
}
|
|
|
|
connection.StateChange += OnStateChange;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Verifies the no connection events were fired.
|
|
/// </summary>
|
|
public void VerifyNoConnectionEventsWereFired()
|
|
{
|
|
Assert.True(countOpenClose == 0);
|
|
Assert.True(countCloseOpen == 0);
|
|
Assert.True(countOtherConnectionStates == 0);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Verifies the connection open and close events were fired.
|
|
/// </summary>
|
|
public void VerifyConnectionOpenCloseEventsWereFired()
|
|
{
|
|
Assert.True(countCloseOpen == 1);
|
|
Assert.True(countOpenClose == 1);
|
|
Assert.True(countOtherConnectionStates == 0);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Verifies the connection opened event was fired.
|
|
/// </summary>
|
|
public void VerifyConnectionOpenedEventWasFired()
|
|
{
|
|
Assert.True(countCloseOpen == 1);
|
|
Assert.True(countOpenClose == 0);
|
|
Assert.True(countOtherConnectionStates == 0);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Verifies the connection closed event was fired.
|
|
/// </summary>
|
|
public void VerifyConnectionClosedEventWasFired()
|
|
{
|
|
Assert.True(countOpenClose == 1);
|
|
Assert.True(countCloseOpen == 0);
|
|
Assert.True(countOtherConnectionStates == 0);
|
|
}
|
|
|
|
public void ResetEventCounters()
|
|
{
|
|
countOpenClose = countCloseOpen = countOtherConnectionStates = 0;
|
|
}
|
|
|
|
private void OnStateChange(object sender, StateChangeEventArgs args)
|
|
{
|
|
if (args.OriginalState == ConnectionState.Closed
|
|
&& args.CurrentState == ConnectionState.Open)
|
|
{
|
|
countCloseOpen++;
|
|
}
|
|
else if (args.OriginalState == ConnectionState.Open
|
|
&& args.CurrentState == ConnectionState.Closed)
|
|
{
|
|
countOpenClose++;
|
|
}
|
|
else
|
|
{
|
|
countOtherConnectionStates++;
|
|
}
|
|
}
|
|
}
|
|
}
|