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
93
94
95
96
97
98
99
|
// Copyright (c) Microsoft Corporation. All rights reserved. See License.txt in the project root for license information.
using System.Collections.Generic;
using System.Dynamic;
using System.IO;
using System.Web.WebPages.Instrumentation;
using Xunit;
namespace System.Web.WebPages.Test.Instrumentation
{
public class InstrumentationServiceTest
{
[Fact]
public void BeginContextDelegatesToRegisteredListeners()
{
// Arrange
dynamic listener1 = CreateListener();
dynamic listener2 = CreateListener();
InstrumentationService inst = CreateInstrumentationService(listener1, listener2);
TextWriter mockWriter = new StringWriter();
// Act
inst.BeginContext(null, "Foo.cshtml", mockWriter, 42, 24, isLiteral: false);
// Assert
Assert.Equal(1, listener1.BeginContextCalls.Count);
Assert.Equal(0, listener1.EndContextCalls.Count);
Assert.Equal(1, listener2.BeginContextCalls.Count);
Assert.Equal(0, listener2.EndContextCalls.Count);
AssertContext("Foo.cshtml", mockWriter, 42, 24, false, listener1.BeginContextCalls[0]);
AssertContext("Foo.cshtml", mockWriter, 42, 24, false, listener2.BeginContextCalls[0]);
}
[Fact]
public void EndContextDelegatesToRegisteredListeners()
{
// Arrange
dynamic listener1 = CreateListener();
dynamic listener2 = CreateListener();
InstrumentationService inst = CreateInstrumentationService(listener1, listener2);
TextWriter mockWriter = new StringWriter();
// Act
inst.EndContext(null, "Foo.cshtml", mockWriter, 42, 24, isLiteral: false);
// Assert
Assert.Equal(1, listener1.EndContextCalls.Count);
Assert.Equal(0, listener1.BeginContextCalls.Count);
Assert.Equal(1, listener2.EndContextCalls.Count);
Assert.Equal(0, listener2.BeginContextCalls.Count);
AssertContext("Foo.cshtml", mockWriter, 42, 24, false, listener1.EndContextCalls[0]);
AssertContext("Foo.cshtml", mockWriter, 42, 24, false, listener2.EndContextCalls[0]);
}
private void AssertContext(string virtualPath, TextWriter writer, int startPosition, int length, bool isLiteral, dynamic context)
{
PageExecutionContextAdapter ctx = new PageExecutionContextAdapter(context);
Assert.Equal(virtualPath, ctx.VirtualPath);
Assert.Same(writer, ctx.TextWriter);
Assert.Equal(startPosition, ctx.StartPosition);
Assert.Equal(length, ctx.Length);
Assert.Equal(isLiteral, ctx.IsLiteral);
}
private InstrumentationService CreateInstrumentationService(params dynamic[] listeners)
{
dynamic service = new ExpandoObject();
service.ExecutionListeners = new List<dynamic>(listeners);
InstrumentationService inst = new InstrumentationService();
inst.IsAvailable = true;
inst.ExtractInstrumentationService = _ => new PageInstrumentationServiceAdapter(service);
inst.CreateContext = CreateExpandoContext;
return inst;
}
private dynamic CreateListener()
{
dynamic listener = new ExpandoObject();
listener.BeginContextCalls = new List<dynamic>();
listener.EndContextCalls = new List<dynamic>();
listener.BeginContext = (Action<dynamic>)(d => { listener.BeginContextCalls.Add(d); });
listener.EndContext = (Action<dynamic>)(d => { listener.EndContextCalls.Add(d); });
return listener;
}
private PageExecutionContextAdapter CreateExpandoContext(string virtualPath, TextWriter writer, int startPosition, int length, bool isLiteral)
{
dynamic ctx = new ExpandoObject();
ctx.VirtualPath = virtualPath;
ctx.TextWriter = writer;
ctx.StartPosition = startPosition;
ctx.Length = length;
ctx.IsLiteral = isLiteral;
return new PageExecutionContextAdapter(ctx);
}
}
}
|
