A long-awaited train service between WA's two largest cities is set to get back on track in June after a near three-year wait.