Use testify library.
TODO: Mocks