Migrate test to use javadsl.TestKit