Eu definitivamente não sou um especialista em testes, mas acho que você está no caminho certo, o que você precisa são mais testes: você deve fazer um teste para cada tipo de coisa que o banco de dados pode retornar (eventualmente; você não precisa aparecer com todos eles imediatamente). Zombar de um tipo diferente de valor de retorno que você poderia esperar do banco de dados (retornou uma linha de resultado, retornou vários, nenhum resultado, por exemplo) para cada teste e certifique-se de que seu método retorna os valores corretos.
Dessa forma, sua primeira implementação pode retornar apenas os resultados esperados do seu teste, mas à medida que você adiciona mais testes, você continua revisando a implementação até que ela seja concluída, conforme necessário.