* Fix restore VM with allocated root disk * Add e2e test for restore vm * Add more checks for e2e test