For ENG-6482, modify the JUnit test to adapt code change.