[CHORE] switch tests to call wrapper connection class