Ensure connection before accessing

This fixes an AttributeError in some situations.

I accidentally included an extra test from my working tree, but it should be in master anyway!