Separate DB-related tests into db_test.go