Add socket tests