Add connection unit tests