Adding another socket test and fixing a mock