Use a class for assert test instead of `PDO`