Add connection unit testing