Add restore command tests