Complete Castle unit test