Add socket test