Refactor unit test to use AssertJ for assertions.
Simply tests by using @Test(expected) attribute.