As noted below, I think this test is difficult to understand because the common test is doing so much - it might be easier to read if it was explicit about the expected return value, and just used common code for the Diameter mocking.