Search code examples
c#unit-testingasp.net-coremoqasp.net-core-identity

Asp.net Core Identity unit test controller actions


I'm having a problem working out how and what to test.

I have a controller that injects UserManager and calls the CreateAsync method to create a new user.

I don't want to test the Identity user manager as this has clearly been thoroughly tested already. What I would like to do is test that the controller runs through the correct paths (in my case, there are 3 paths, sending responses back with either model state errors, identity response errors or a simple string)

Should I be trying to create a mock of the user manager in order to create my test (I'm not sure how to set up user manager as a mock dependency) Second, how can I set conditions to verify that the controller has taken a given path.

I am using xUnit and Moq.

[Route("api/[controller]")]
public class MembershipController : BaseApiController
{
    private UserManager<ApplicationUser> _userManager;

    public MembershipController(UserManager<ApplicationUser> userManager)
    {
        _userManager = userManager;
    }

    [HttpGet("RegisterNewUser")]
    public HttpResponseMessage RegisterNewUser([FromBody] NewUserRegistration user)
    {
        if (ModelState.IsValid)
        {
            ApplicationUser newUser = new ApplicationUser();
            newUser.UserName = user.username;
            newUser.Email = user.password;
            IdentityResult result = _userManager.CreateAsync(newUser, user.password).Result;

            if (result.Errors.Count() > 0)
            {
                var errors = new IdentityResultErrorResponse().returnResponseErrors(result.Errors);
                return this.WebApiResponse(errors, HttpStatusCode.BadRequest);
            }
        }
        else
        {
            var errors = new ViewModelResultErrorResponse().returnResponseErrors(ModelState);
            return this.WebApiResponse(errors, HttpStatusCode.BadRequest);
        }

        return this.WebApiResponse(
                    "We have sent a valifation email to you, please click on the verify email account link.",
                    HttpStatusCode.OK);
    }
}

In My unit test I have the following to test a happy path scenario

    [Fact]
    public void RegisterNewUser_ReturnsHttpStatusOK_WhenValidModelPosted()
    {
        var mockStore = new Mock<IUserStore<ApplicationUser>>();
        var mockUserManager = new Mock<UserManager<ApplicationUser>>(mockStore.Object, null, null, null, null, null, null, null, null);

        ApplicationUser testUser = new ApplicationUser { UserName = "[email protected]" };

        mockStore.Setup(x => x.CreateAsync(testUser, It.IsAny<CancellationToken>()))
           .Returns(Task.FromResult(IdentityResult.Success));

        mockStore.Setup(x => x.FindByNameAsync(testUser.UserName, It.IsAny<CancellationToken>()))
                    .Returns(Task.FromResult(testUser));


        mockUserManager.Setup(x => x.CreateAsync(testUser).Result).Returns(new IdentityResult());

        MembershipController sut = new MembershipController(mockUserManager.Object);
        var input = new NewUserInputBuilder().Build();
        sut.RegisterNewUser(input);

    }

Where "input" in sut.RegisterNewUser(input); refers to a helper class which constructs the viewmodel which the controller action requires:

public class NewUserInputBuilder
{
    private string username { get; set; }
    private string password { get; set; }
    private string passwordConfirmation { get; set; }
    private string firstname { get; set; }
    private string lastname { get; set; }

    internal NewUserInputBuilder()
    {
        this.username = "[email protected]";
        this.password = "password";
        this.passwordConfirmation = "password";
        this.firstname = "user";
        this.lastname = "name";
    }

    internal NewUserInputBuilder WithNoUsername()
    {
        this.username = "";
        return this;
    }

    internal NewUserInputBuilder WithMisMatchedPasswordConfirmation()
    {
        this.passwordConfirmation = "MismatchedPassword";
        return this;
    }

    internal NewUserRegistration Build()
    {
        return new NewUserRegistration
        { username = this.username, password = this.password,
            passwordConfirmation = this.passwordConfirmation,
            firstname = this.firstname, lastname = this.lastname
        };
    }
} 

My aim here is to force 3 conditions via tests:

  1. Create a valid viewmodel and return a success message
  2. Create a valid viewmodel but returns a IdentityResponse error (eg. user exists) which gets converted to
  3. Create an invalid viewmodel and returns Modelstate errors

The errors are handled using a abstract class which returns a json object The base class for the controller simply constructs a HttpResponseMessage for return.

Basically I want to check that the correct error response class is called by forcing the test down the modelstate error path, the identityresult.errors path and that the happy path can be achieved.

Then my plan is to test the error response classes in isolation.

Hopefully that is enough detail.


Solution

  • Mehod under test should be made async and not use blocking calls ie .Result

    [HttpGet("RegisterNewUser")]
    public async Task<HttpResponseMessage> RegisterNewUser([FromBody] NewUserRegistration user) {
        if (ModelState.IsValid) {
            var newUser = new ApplicationUser() {
                UserName = user.username,
                Email = user.password
            };
            var result = await _userManager.CreateAsync(newUser, user.password);
            if (result.Errors.Count() > 0) {
                var errors = new IdentityResultErrorResponse().returnResponseErrors(result.Errors);
                return this.WebApiResponse(errors, HttpStatusCode.BadRequest);
            }
        } else {
            var errors = new ViewModelResultErrorResponse().returnResponseErrors(ModelState);
            return this.WebApiResponse(errors, HttpStatusCode.BadRequest);
        }
    
        return this.WebApiResponse(
                    "We have sent a valifation email to you, please click on the verify email account link.",
                    HttpStatusCode.OK);
    }
    

    Review of Happy path scenario and method under test shows that there is no need to setup the UserStore as test will be overriding the user manager virtual members directly.

    Note the test has also been made async as well.

    1. Create a valid viewmodel and return a success message
    [Fact]
    public async Task RegisterNewUser_ReturnsHttpStatusOK_WhenValidModelPosted() {
        //Arrange
        var mockStore = Mock.Of<IUserStore<ApplicationUser>>();
        var mockUserManager = new Mock<UserManager<ApplicationUser>>(mockStore, null, null, null, null, null, null, null, null);
    
        mockUserManager
            .Setup(x => x.CreateAsync(It.IsAny<ApplicationUser>(), It.IsAny<string>()))
            .ReturnsAsync(IdentityResult.Success);
    
        var sut = new MembershipController(mockUserManager.Object);
        var input = new NewUserInputBuilder().Build();
    
        //Act
        var actual = await sut.RegisterNewUser(input);
    
        //Assert
        actual
            .Should().NotBeNull()
            .And.Match<HttpResponseMessage>(_ => _.IsSuccessStatusCode == true);        
    }
    
    1. Create a valid viewmodel but returns a IdentityResponse error (eg. user exists) which gets converted

    For this you just need to setup the mock to return a result with errors.

    [Fact]
    public async Task RegisterNewUser_ReturnsHttpStatusBadRequest_WhenViewModelPosted() {
        //Arrange
    
        //...code removed for brevity
    
        mockUserManager
            .Setup(x => x.CreateAsync(It.IsAny<ApplicationUser>(), It.IsAny<string>()))
            .ReturnsAsync(IdentityResult.Failed(new IdentityError { Description = "test"}));
    
        //...code removed for brevity
    
        //Assert
        actual
            .Should().NotBeNull()
            .And.Match<HttpResponseMessage>(_ => _.StatusCode == HttpStatusCode.BadRequest);
    }
    

    And for

    1. Create an invalid viewmodel and returns Modelstate errors

    You just need to set the model state of the controller so that it is invalid.

    [Fact]
    public async Task RegisterNewUser_ReturnsHttpStatusBadRequest_WhenInvalidModelState() {
        //Arrange
        var mockStore = Mock.Of<IUserStore<ApplicationUser>>();
        var mockUserManager = new Mock<UserManager<ApplicationUser>>(mockStore, null, null, null, null, null, null, null, null);
    
        var sut = new MembershipController(mockUserManager.Object);
        sut.ModelState.AddModelError("", "invalid data");
        var input = new NewUserInputBuilder().Build();
    
        //Act
        var actual = await sut.RegisterNewUser(input);
    
        //Assert
        actual
            .Should().NotBeNull()
            .And.Match<HttpResponseMessage>(_ => _.StatusCode == HttpStatusCode.BadRequest);    
    }
    

    FluentAssertions were used to do all the assertions. You could just as easily used Assert.* API.

    This should be enough to get you on your way with the above question.