add test socket code