feat: add export and restore

Author: AriPerkkio

src/fixtures/file-system.ts

@@ -67,4 +67,12 @@ export class FileSystem {
   async rm(path: string) {
     return this._instance.fs.rm(path);
   }
+
+  async export(path = "./", options = { excludes: [], includes: [] }) {
+    return await this._instance.export(path, { ...options, format: "binary" });
+  }
+
+  async restore(snapshot: Uint8Array) {
+    return await this._instance.mount(snapshot);
+  }
 }

test/file-system.test.ts

@@ -40,3 +40,19 @@ test("user can remove files from webcontainer", async ({ webcontainer }) => {
     "third",
   ]);
 });
+
+test("user can export and restore snapshots", async ({ webcontainer }) => {
+  await webcontainer.writeFile("/first", "1");
+  await webcontainer.writeFile("/second", "2");
+
+  const snapshot = await webcontainer.export();
+
+  await webcontainer.rm("/second");
+  await expect(webcontainer.readdir("/")).resolves.toStrictEqual(["first"]);
+
+  await webcontainer.restore(snapshot);
+  await expect(webcontainer.readdir("/")).resolves.toStrictEqual([
+    "first",
+    "second",
+  ]);
+});