Fix test code and delete deprecated assert method.

Fix the failing test.

`junit.framework` package is deprecated.
`assertEquals`, `assertTrue`...
Use `assertThat` instead.