Use testify testing framework