feat: add arc support

src/jest.ts

@@ -1,6 +1,7 @@
 import { cleanup } from "@marko/testing-library";
 import { findProjectFixtures, defaultNormalizer } from "./";
 import { toMatchFile } from "jest-file-snapshot";
+import { getArcFlag } from "./utils";
 
 expect.extend({ toMatchFile });
 
@@ -15,8 +16,11 @@ export default function snapshotComponentFixtures(
       Object.keys(component.fixtures).forEach(name => {
         it(name, async () => {
           const fixture = component.fixtures[name];
+          const fileExtension = getArcFlag(component.path); // this returns "[mobile]"
+          const replaceValue = fileExtension.length === 0 ? ".html" : `${fileExtension}.html`;
+
           expect(await fixture.toString(normalize)).toMatchFile(
-            fixture.path.replace(fixture.ext, ".html")
+            fixture.path.replace(fixture.ext, replaceValue)
           );
         });
       });

src/mocha.ts

@@ -3,6 +3,7 @@ import path from "path";
 import assert from "assert";
 import { cleanup } from "@marko/testing-library";
 import { findProjectFixtures, defaultNormalizer } from "./";
+import { getArcFlag } from "./utils";
 
 declare const before: (fn: () => any) => any;
 
@@ -17,8 +18,11 @@ export default function snapshotComponentFixtures(
       Object.keys(component.fixtures).forEach(name => {
         it(name, async () => {
           const fixture = component.fixtures[name];
+          const fileExtension = getArcFlag(component.path); // this returns "[mobile]"
+          const replaceValue = fileExtension.length === 0 ? ".html" : `${fileExtension}.html`;
+
           snapshot(
-            fixture.path.replace(fixture.ext, ".html"),
+            fixture.path.replace(fixture.ext, replaceValue),
             await fixture.toString(normalize)
           );
         });

src/utils.ts

@@ -0,0 +1,21 @@
+import path from "path";
+
+export function getArcFlag(filename: string) {
+  const start = filename.lastIndexOf(path.sep) + 1;
+  const leftDot = filename.indexOf(".", start);
+
+  const closeBracket = leftDot - 1;
+  if (filename[closeBracket] === "]") {
+    const openBracket = filename.lastIndexOf("[", closeBracket);
+    if (openBracket > start) {
+      // If we match a "]" before the extension and find a "[" before that,
+      // then we have an arc flag. Strip it off.
+      return (
+        filename.slice(openBracket, closeBracket + 1)
+      );
+    }
+  }
+
+  return '';
+}
+

test/components/hello/fixtures/data[mobile].html

@@ -0,0 +1,3 @@
+<div>
+  Hello Arc File Data!
+</div>

test/components/hello/fixtures/template[mobile].html

@@ -0,0 +1,3 @@
+<div>
+  Hello Template!
+</div>

test/components/hello/index[mobile].marko

@@ -0,0 +1,2 @@
+class {}
+<div>Hello Arc File ${input.name}!</div>

test/index.test.ts

@@ -134,6 +134,32 @@ describe("findProjectFixtures", () => {
           "name": "hello",
           "path": "(cwd)/test/components/hello/index.marko",
         },
+        Object {
+          "component": [Component "(cwd)/test/components/hello/index[mobile].marko"],
+          "fixtures": Object {
+            "data": Object {
+              "ext": ".json",
+              "fixture": Object {
+                "name": "Data",
+              },
+              "name": "data",
+              "path": "(cwd)/test/components/hello/fixtures/data.json",
+              "render": [Function],
+              "toString": [Function],
+            },
+            "template": Object {
+              "ext": ".marko",
+              "fixture": [Component "(cwd)/test/components/hello/fixtures/template.marko"],
+              "name": "template",
+              "path": "(cwd)/test/components/hello/fixtures/template.marko",
+              "render": [Function],
+              "toString": [Function],
+            },
+          },
+          "fixturesPath": "(cwd)/test/components/hello/fixtures",
+          "name": "hello",
+          "path": "(cwd)/test/components/hello/index[mobile].marko",
+        },
       ]
     `);
   });