DI for socket class + tests