Mock Internal Types Via Proxy

With JustMock you can mock internal types with InternalsVisibleToAttribute the same way you mock public types. Without InternalsVisibleToAttribute you are forced to mock as you mock non-public members.

In the further examples, we will use the following sample classes to test:

• C#
• VB

internal class FooInternal 
{ 
    internal FooInternal() 
    { 
        builder = new StringBuilder(); 
    } 
 
    public StringBuilder Builder 
    { 
        get 
        { 
            return builder; 
        } 
    } 
 
    private StringBuilder builder; 
} 
 
public class FooInternalCtor 
{ 
    internal FooInternalCtor(string name) 
    { 
        this.name = name; 
    } 
 
    public string Name { get { return name; } } 
 
    private string name; 
} 

Friend Class FooInternal 
    Friend Sub New() 
        m_builder = New StringBuilder() 
    End Sub 
 
    Public ReadOnly Property Builder() As StringBuilder 
        Get 
            Return m_builder 
        End Get 
    End Property 
 
    Private m_builder As StringBuilder 
End Class 
 
Public Class FooInternalCtor 
    Friend Sub New(name As String) 
        Me.m_name = name 
    End Sub 
 
    Public ReadOnly Property Name() As String 
        Get 
            Return m_name 
        End Get 
    End Property 
 
    Private m_name As String 
End Class 

Mock Internal Type Via Proxy

Provided that the InternalsVisibleTo attribute is included in the AssemblyInfo.cs, let's consider the following example:

• C#
• VB

[TestMethod] 
public void ShouldMockInternalTypeViaProxy() 
{ 
    // Arrange 
    var foo = Mock.Create<FooInternal>(Behavior.CallOriginal); 
 
    // Assert 
    Assert.IsNotNull(foo.Builder); 
} 

<TestMethod()> 
Public Sub ShouldMockInternalTypeViaProxy() 
    ' Arrange 
    Dim foo = Mock.Create(Of FooInternal)(Behavior.CallOriginal) 
 
    ' Assert 
    Assert.IsNotNull(foo.Builder) 
End Sub 

Here we verify that the mock is created and its constructor is called. Thus, foo.Builder will be assigned.

The key to be used in the InternalsVisibleTo attribute is as follows:

• For projects different from Silverlight: InternalMocking#Key
• For Silverlight projects: InternalMocking#KeySilverlight

Create Mock With Internal Constructor

You can mock class that exposes internal constructor in the same way you mock public types.

• C#
• VB

[TestMethod] 
public void ShouldCreateMockWithInternalCtor() 
{ 
    // Arrange 
    var expected = "hello"; 
 
    var foo = Mock.Create<FooInternalCtor>(x => 
    { 
        x.CallConstructor(() => new FooInternalCtor(expected)); 
        x.SetBehavior(Behavior.CallOriginal); 
    }); 
 
    // Assert 
    Assert.AreEqual(foo.Name, expected); 
} 

<TestMethod()> 
Public Sub ShouldCreateMockWithInternalCtor() 
    ' Arrange 
    Dim expected = "hello" 
 
    Dim foo = Mock.Create(Of FooInternalCtor)(Function(x) 
                                                    x.CallConstructor(Function() New FooInternalCtor(expected)) 
                                                    x.SetBehavior(Behavior.CallOriginal) 
                                                End Function) 
 
    ' Assert 
    Assert.AreEqual(foo.Name, expected) 
End Sub 

In the example above, we mock the internal constructor of FooInternalCtor class. When foo is created the internal constructor will be called with "hello" as argument and the Name property will be assigned "hello" respectively. We verify that by calling foo.Name in the assertion phase of the AAA pattern.

Mock An Interface Member, Privately Implemented In Base Class

Here is an interface with a single member in it and a class which is inheriting that interface.

• C#
• VB

public interface IManager 
{ 
    object Provider { get; } 
} 
 
public class FooBase : IManager 
{ 
    object IManager.Provider 
    { 
        get { throw new NotImplementedException(); } 
    } 
} 
 
public class Bar : FooBase 
{ 
} 

Public Interface IManager 
    ReadOnly Property Provider() As Object 
End Interface 
 
Public Class FooBase 
    Implements IManager 
    ReadOnly Property Provider() As Object Implements IManager.Provider 
        Get 
            Throw New NotImplementedException() 
        End Get 
    End Property 
End Class 
 
Public Class Bar 
    Inherits FooBase 
End Class 

The derived class Bar will be our object to mock, in the example below.

• C#
• VB

[TestMethod] 
public void ShouldMockAnInterfaceMemberPrivatelyImplementedInBaseClass() 
{ 
    // Arrange 
    var bar = Mock.Create<Bar>(); 
 
    string expected = "dummy"; 
 
    Mock.Arrange(() => ((IManager)bar).Provider).Returns("dummy"); 
 
    // Act 
    var actual = ((IManager)bar).Provider; 
 
    // Assert 
    Assert.AreEqual(expected, actual); 
} 

<TestMethod()> 
Public Sub ShouldMockAnInterfaceMemberPrivatelyImplementedInBaseClass() 
    ' Arrange 
    Dim bar = Mock.Create(Of Bar)() 
 
    Dim expected As String = "dummy" 
 
    Mock.Arrange(Function() DirectCast(bar, IManager).Provider).Returns("dummy") 
 
    ' Act 
    Dim actual = DirectCast(bar, IManager).Provider 
 
    ' Assert 
    Assert.AreEqual(expected, actual) 
End Sub 

We have mocked the Bar class. Then we make arranges about the interface member, Provider and we act. Finally, we can assert that the test behavior is as expected.