Update tests for connection module.