Update Connection class test

Update Connection class test