The Justice Department will investigate the sudden departure of the US lawyer in Atlanta after the criticism of Trump