@typespec/emitter-framework 0.6.0-dev.4 → 0.6.0-dev.6

package/dist/src/core/context/name-policy-context.js

@@ -9,5 +9,4 @@ export const TransformNamePolicyContext = createNamedContext("TransfromNamePolic
 });
 export function useTransformNamePolicy() {
   return useContext(TransformNamePolicyContext);
-}
-//# sourceMappingURL=name-policy-context.js.map
+}

package/dist/src/core/index.js

@@ -1,4 +1,3 @@
 export * from "./context/index.js";
 export * from "./transport-name-policy.js";
-export * from "./write-output.js";
-//# sourceMappingURL=index.js.map
+export * from "./write-output.js";

package/dist/src/core/transport-name-policy.js

@@ -38,5 +38,4 @@ export function createTransformNamePolicy(namer) {
       return namer.applicationNamer(type);
     }
   };
-}
-//# sourceMappingURL=transport-name-policy.js.map
+}

package/dist/src/core/write-output.js

@@ -16,5 +16,4 @@ async function writeOutputDirectory(dir, emitterOutputDir) {
       });
     }
   }
-}
-//# sourceMappingURL=write-output.js.map
+}

package/dist/src/lib.js

@@ -14,5 +14,4 @@ export const $lib = createTypeSpecLibrary({
 export const {
   reportDiagnostic,
   createDiagnostic
-} = $lib;
-//# sourceMappingURL=lib.js.map
+} = $lib;

package/dist/src/testing/index.js

@@ -5,5 +5,4 @@ export const EmitterFrameworkTestLibrary = createTestLibrary({
   name: "@typespec/emitter-framework",
   packageRoot: resolvePath(fileURLToPath(import.meta.url), "../../../")
 });
-export * from "./scenario-test/index.js";
-//# sourceMappingURL=index.js.map
+export * from "./scenario-test/index.js";

package/dist/src/testing/scenario-test/harness.js

@@ -274,5 +274,4 @@ function splitByH1(content) {
     };
   });
   return sections;
-}
-//# sourceMappingURL=harness.js.map
+}

package/dist/src/testing/scenario-test/index.js

@@ -1,4 +1,3 @@
 export * from "./harness.js";
 export * from "./snippet-extractor.js";
-export * from "./test-host.js";
-//# sourceMappingURL=index.js.map
+export * from "./test-host.js";

package/dist/src/testing/scenario-test/snippet-extractor.js

@@ -149,5 +149,4 @@ class SnippetExtractorImpl {
     };
     return traverse(rootNode); // Start traversal from the root node
   }
-}
-//# sourceMappingURL=snippet-extractor.js.map
+}

package/dist/src/testing/scenario-test/test-host.js

@@ -46,5 +46,4 @@ async function readFilesRecursively(currentDir, emitterOutputDir, runner) {
     }
   }
   return result;
-}
-//# sourceMappingURL=test-host.js.map
+}

package/dist/src/typescript/components/array-expression.js

@@ -7,5 +7,4 @@ export function ArrayExpression({
   return ay.code`Array<${_$createComponent(TypeExpression, {
     type: elementType
   })}>`;
-}
-//# sourceMappingURL=array-expression.js.map
+}

package/dist/src/typescript/components/arrow-function.js

@@ -1,6 +1,4 @@
-import { createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
-import { mergeProps as _$mergeProps } from "@alloy-js/core/jsx-runtime";
-import { splitProps } from "@alloy-js/core/jsx-runtime";
+import { splitProps, mergeProps as _$mergeProps, createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
 import * as ts from "@alloy-js/typescript";
 import { buildParameterDescriptors, getReturnType } from "../utils/operation.js";
 import { TypeExpression } from "./type-expression.js";
@@ -27,5 +25,4 @@ export function ArrowFunction(props) {
     returnType: returnType,
     parameters: allParameters
   }));
-}
-//# sourceMappingURL=arrow-function.js.map
+}

package/dist/src/typescript/components/class-method.js

@@ -1,5 +1,4 @@
-import { memo as _$memo } from "@alloy-js/core/jsx-runtime";
-import { createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
+import { createComponent as _$createComponent, memo as _$memo } from "@alloy-js/core/jsx-runtime";
 import { refkey as getRefkey } from "@alloy-js/core";
 import * as ts from "@alloy-js/typescript";
 import { buildParameterDescriptors, getReturnType } from "../utils/operation.js";
@@ -35,5 +34,4 @@ export function ClassMethod(props) {
 }
 function isTypedMethodDeclarationProps(props) {
   return props.type !== undefined;
-}
-//# sourceMappingURL=class-method.js.map
+}

package/dist/src/typescript/components/enum-declaration.js

@@ -1,5 +1,4 @@
-import { memo as _$memo } from "@alloy-js/core/jsx-runtime";
-import { createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
+import { createComponent as _$createComponent, memo as _$memo } from "@alloy-js/core/jsx-runtime";
 import * as ay from "@alloy-js/core";
 import * as ts from "@alloy-js/typescript";
 import { $ } from "@typespec/compiler/experimental/typekit";
@@ -61,5 +60,4 @@ export function EnumMember(props) {
       return ay.refkey(props.refkey ?? props.type);
     }
   });
-}
-//# sourceMappingURL=enum-declaration.js.map
+}

package/dist/src/typescript/components/function-declaration.js

@@ -1,5 +1,4 @@
-import { memo as _$memo } from "@alloy-js/core/jsx-runtime";
-import { createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
+import { createComponent as _$createComponent, memo as _$memo } from "@alloy-js/core/jsx-runtime";
 import { refkey as getRefkey } from "@alloy-js/core";
 import * as ts from "@alloy-js/typescript";
 import { buildParameterDescriptors, getReturnType } from "../utils/operation.js";
@@ -71,5 +70,4 @@ function isTypedFunctionDeclarationProps(props) {
 function isTypedFunctionParametersProps(props) {
   return "type" in props;
 }
-const reservedFunctionKeywords = new Set(["break", "case", "catch", "class", "const", "continue", "debugger", "default", "delete", "do", "else", "enum", "export", "extends", "finally", "for", "function", "if", "import", "in", "instanceof", "new", "return", "super", "switch", "this", "throw", "try", "typeof", "var", "void", "while", "with", "yield", "let", "static", "implements", "interface", "package", "private", "protected", "public", "await"]);
-//# sourceMappingURL=function-declaration.js.map
+const reservedFunctionKeywords = new Set(["break", "case", "catch", "class", "const", "continue", "debugger", "default", "delete", "do", "else", "enum", "export", "extends", "finally", "for", "function", "if", "import", "in", "instanceof", "new", "return", "super", "switch", "this", "throw", "try", "typeof", "var", "void", "while", "with", "yield", "let", "static", "implements", "interface", "package", "private", "protected", "public", "await"]);

package/dist/src/typescript/components/function-expression.js

@@ -1,6 +1,4 @@
-import { createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
-import { mergeProps as _$mergeProps } from "@alloy-js/core/jsx-runtime";
-import { splitProps } from "@alloy-js/core/jsx-runtime";
+import { splitProps, mergeProps as _$mergeProps, createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
 import * as ts from "@alloy-js/typescript";
 import { buildParameterDescriptors, getReturnType } from "../utils/operation.js";
 import { TypeExpression } from "./type-expression.js";
@@ -27,5 +25,4 @@ export function FunctionExpression(props) {
     returnType: returnType,
     parameters: allParameters
   }));
-}
-//# sourceMappingURL=function-expression.js.map
+}

package/dist/src/typescript/components/function-type.js

@@ -1,6 +1,4 @@
-import { createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
-import { mergeProps as _$mergeProps } from "@alloy-js/core/jsx-runtime";
-import { splitProps } from "@alloy-js/core/jsx-runtime";
+import { splitProps, mergeProps as _$mergeProps, createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
 import * as ts from "@alloy-js/typescript";
 import { buildParameterDescriptors, getReturnType } from "../utils/operation.js";
 import { TypeExpression } from "./type-expression.js";
@@ -27,5 +25,4 @@ export function FunctionType(props) {
     returnType: returnType,
     parameters: allParameters
   }));
-}
-//# sourceMappingURL=function-type.js.map
+}

package/dist/src/typescript/components/index.js

@@ -13,5 +13,4 @@ export * from "./type-expression.js";
 export * from "./type-transform.js";
 export * from "./union-declaration.js";
 export * from "./union-expression.js";
-export * from "./value-expression.js";
-//# sourceMappingURL=index.js.map
+export * from "./value-expression.js";

package/dist/src/typescript/components/interface-declaration.js

@@ -1,6 +1,4 @@
-import { memo as _$memo } from "@alloy-js/core/jsx-runtime";
-import { mergeProps as _$mergeProps } from "@alloy-js/core/jsx-runtime";
-import { createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
+import { createComponent as _$createComponent, mergeProps as _$mergeProps, memo as _$memo } from "@alloy-js/core/jsx-runtime";
 import * as ay from "@alloy-js/core";
 import { refkey as getRefkey, mapJoin } from "@alloy-js/core";
 import * as ts from "@alloy-js/typescript";
@@ -133,5 +131,4 @@ function InterfaceBody(props) {
       });
     }
   })), _$memo(() => props.children)];
-}
-//# sourceMappingURL=interface-declaration.js.map
+}

package/dist/src/typescript/components/interface-member.js

@@ -1,5 +1,4 @@
-import { createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
-import { memo as _$memo } from "@alloy-js/core/jsx-runtime";
+import { memo as _$memo, createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
 import * as ts from "@alloy-js/typescript";
 import { isNeverType } from "@typespec/compiler";
 import { $ } from "@typespec/compiler/experimental/typekit";
@@ -37,5 +36,4 @@ export function InterfaceMember(props) {
       }
     });
   }
-}
-//# sourceMappingURL=interface-member.js.map
+}

package/dist/src/typescript/components/interface-method.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"interface-method.d.ts","sourceRoot":"","sources":["../../../../src/typescript/components/interface-method.tsx"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,MAAM,sBAAsB,CAAC;AAC3C,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAI/C,MAAM,WAAW,4BAA6B,SAAQ,IAAI,CAAC,EAAE,CAAC,oBAAoB,EAAE,MAAM,CAAC;IACzF,IAAI,EAAE,SAAS,CAAC;IAChB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,cAAc,CAAC,EAAE,SAAS,GAAG,QAAQ,GAAG,SAAS,CAAC;CACnD;AAED,MAAM,MAAM,oBAAoB,GAAG,4BAA4B,GAAG,EAAE,CAAC,oBAAoB,CAAC;AAE1F;;;;GAIG;AACH,wBAAgB,eAAe,CAAC,KAAK,EAAE,QAAQ,CAAC,oBAAoB,CAAC,iDA8BpE"}
+{"version":3,"file":"interface-method.d.ts","sourceRoot":"","sources":["../../../../src/typescript/components/interface-method.tsx"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,MAAM,sBAAsB,CAAC;AAC3C,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAI/C,MAAM,WAAW,4BAA6B,SAAQ,IAAI,CAAC,EAAE,CAAC,oBAAoB,EAAE,MAAM,CAAC;IACzF,IAAI,EAAE,SAAS,CAAC;IAChB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,cAAc,CAAC,EAAE,SAAS,GAAG,QAAQ,GAAG,SAAS,CAAC;CACnD;AAED,MAAM,MAAM,oBAAoB,GAAG,4BAA4B,GAAG,EAAE,CAAC,oBAAoB,CAAC;AAE1F;;;;GAIG;AACH,wBAAgB,eAAe,CAAC,KAAK,EAAE,QAAQ,CAAC,oBAAoB,CAAC,iDA4BpE"}

package/dist/src/typescript/components/interface-method.js

@@ -1,6 +1,4 @@
-import { mergeProps as _$mergeProps } from "@alloy-js/core/jsx-runtime";
-import { createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
-import { splitProps } from "@alloy-js/core/jsx-runtime";
+import { splitProps, createComponent as _$createComponent, mergeProps as _$mergeProps } from "@alloy-js/core/jsx-runtime";
 import * as ts from "@alloy-js/typescript";
 import { buildParameterDescriptors, getReturnType } from "../utils/operation.js";
 import { TypeExpression } from "./type-expression.js";
@@ -15,7 +13,7 @@ export function InterfaceMethod(props) {
     return _$createComponent(ts.InterfaceMethod, props);
   }
   const [efProps, updateProps, forwardProps] = splitProps(props, ["type"], ["returnType", "parameters"]);
-  const name = props.name ? props.name : ts.useTSNamePolicy().getName(efProps.type.name, "function");
+  const name = props.name ?? ts.useTSNamePolicy().getName(efProps.type.name, "function");
   const returnType = props.returnType ?? _$createComponent(TypeExpression, {
     get type() {
       return getReturnType(efProps.type);
@@ -25,10 +23,9 @@ export function InterfaceMethod(props) {
     params: props.parameters,
     mode: props.parametersMode
   });
-  return _$createComponent(ts.InterfaceMethod, _$mergeProps(forwardProps, {
+  return _$createComponent(ts.InterfaceMethod, _$mergeProps(forwardProps, updateProps, {
     name: name,
     returnType: returnType,
     parameters: allParameters
-  }, updateProps));
-}
-//# sourceMappingURL=interface-method.js.map
+  }));
+}

package/dist/src/typescript/components/record-expression.js

@@ -9,5 +9,4 @@ export function RecordExpression({
     type: elementType
   })}>
   `;
-}
-//# sourceMappingURL=record-expression.js.map
+}

package/dist/src/typescript/components/static-serializers.js

@@ -186,5 +186,4 @@ export function ArraySerializer() {
       `];
     }
   });
-}
-//# sourceMappingURL=static-serializers.js.map
+}

package/dist/src/typescript/components/type-alias-declaration.js

@@ -1,6 +1,4 @@
-import { memo as _$memo } from "@alloy-js/core/jsx-runtime";
-import { createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
-import { mergeProps as _$mergeProps } from "@alloy-js/core/jsx-runtime";
+import { mergeProps as _$mergeProps, createComponent as _$createComponent, memo as _$memo } from "@alloy-js/core/jsx-runtime";
 import { refkey as getRefkey } from "@alloy-js/core";
 import * as ts from "@alloy-js/typescript";
 import { $ } from "@typespec/compiler/experimental/typekit";
@@ -43,5 +41,4 @@ export function TypeAliasDeclaration(props) {
 }
 function isTypedAliasDeclarationProps(props) {
   return "type" in props;
-}
-//# sourceMappingURL=type-alias-declaration.js.map
+}

package/dist/src/typescript/components/type-declaration.js

@@ -1,5 +1,4 @@
-import { mergeProps as _$mergeProps } from "@alloy-js/core/jsx-runtime";
-import { createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
+import { createComponent as _$createComponent, mergeProps as _$mergeProps } from "@alloy-js/core/jsx-runtime";
 import * as ts from "@alloy-js/typescript";
 import { EnumDeclaration } from "./enum-declaration.js";
 import { InterfaceDeclaration } from "./interface-declaration.js";
@@ -35,5 +34,4 @@ export function TypeDeclaration(props) {
         type: type
       }, restProps));
   }
-}
-//# sourceMappingURL=type-declaration.js.map
+}

package/dist/src/typescript/components/type-expression.js

@@ -1,5 +1,4 @@
-import { memo as _$memo } from "@alloy-js/core/jsx-runtime";
-import { createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
+import { createComponent as _$createComponent, memo as _$memo } from "@alloy-js/core/jsx-runtime";
 import { For, refkey } from "@alloy-js/core";
 import { Reference, ValueExpression } from "@alloy-js/typescript";
 import { $ } from "@typespec/compiler/experimental/typekit";
@@ -213,5 +212,4 @@ function isDeclaration(type) {
     default:
       return false;
   }
-}
-//# sourceMappingURL=type-expression.js.map
+}

package/dist/src/typescript/components/type-transform.js

@@ -1,5 +1,4 @@
-import { memo as _$memo } from "@alloy-js/core/jsx-runtime";
-import { createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
+import { createComponent as _$createComponent, memo as _$memo } from "@alloy-js/core/jsx-runtime";
 import { code, For, mapJoin, refkey } from "@alloy-js/core";
 import * as ts from "@alloy-js/typescript";
 import { $ } from "@typespec/compiler/experimental/typekit";
@@ -389,5 +388,4 @@ function getCollapsedProperty(model, collapse) {
     return Array.from(modelProperties.values())[0];
   }
   return undefined;
-}
-//# sourceMappingURL=type-transform.js.map
+}

package/dist/src/typescript/components/union-declaration.js

@@ -1,5 +1,4 @@
-import { createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
-import { mergeProps as _$mergeProps } from "@alloy-js/core/jsx-runtime";
+import { mergeProps as _$mergeProps, createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
 import { refkey as getRefkey } from "@alloy-js/core";
 import * as ts from "@alloy-js/typescript";
 import { $ } from "@typespec/compiler/experimental/typekit";
@@ -41,5 +40,4 @@ export function UnionDeclaration(props) {
 }
 function isTypedUnionDeclarationProps(props) {
   return "type" in props;
-}
-//# sourceMappingURL=union-declaration.js.map
+}

package/dist/src/typescript/components/union-expression.js

@@ -1,5 +1,4 @@
-import { memo as _$memo } from "@alloy-js/core/jsx-runtime";
-import { createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
+import { createComponent as _$createComponent, memo as _$memo } from "@alloy-js/core/jsx-runtime";
 import * as ay from "@alloy-js/core";
 import * as ts from "@alloy-js/typescript";
 import { $ } from "@typespec/compiler/experimental/typekit";
@@ -32,5 +31,4 @@ export function UnionExpression({
     return [variants, " ", ` | ${children}`];
   }
   return variants;
-}
-//# sourceMappingURL=union-expression.js.map
+}

package/dist/src/typescript/components/value-expression.js

@@ -1,5 +1,4 @@
-import { memo as _$memo } from "@alloy-js/core/jsx-runtime";
-import { createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
+import { createComponent as _$createComponent, memo as _$memo } from "@alloy-js/core/jsx-runtime";
 import * as ts from "@alloy-js/typescript";
 import { compilerAssert } from "@typespec/compiler";
 
@@ -68,5 +67,4 @@ export function ValueExpression(props) {
         }
       });
   }
-}
-//# sourceMappingURL=value-expression.js.map
+}

package/dist/src/typescript/index.js

@@ -1,3 +1,2 @@
 export * from "./components/index.js";
-export * from "./utils/index.js";
-//# sourceMappingURL=index.js.map
+export * from "./utils/index.js";

package/dist/src/typescript/lib.js

@@ -61,5 +61,4 @@ export const $typescriptLib = createTypeSpecLibrary({
 export const {
   reportDiagnostic: reportTypescriptDiagnostic,
   createDiagnostic: CreateTypescriptDiagnostic
-} = $typescriptLib;
-//# sourceMappingURL=lib.js.map
+} = $typescriptLib;

package/dist/src/typescript/utils/operation.js

@@ -51,5 +51,4 @@ function normalizeParameters(params) {
     }
     return param;
   });
-}
-//# sourceMappingURL=operation.js.map
+}

package/dist/test/testing/snippet-extractor-csharp.test.js

@@ -0,0 +1,96 @@
+import { d } from "@alloy-js/core/testing";
+import { beforeEach, describe, expect, it } from "vitest";
+import { createCSharpExtractorConfig, createSnipperExtractor } from "../../src/testing/index.js";
+describe("C# Snippet Extractor", () => {
+  let extractor;
+  beforeEach(() => {
+    extractor = createSnipperExtractor(createCSharpExtractorConfig());
+  });
+  it("should extract a class", () => {
+    const content = d`
+    public class Foo {
+      public Foo() {
+        Console.WriteLine("Hello");
+      }
+    }
+    `;
+    const snippet = extractor.getClass(content, "Foo");
+    expect(snippet).toBe(d`
+      public class Foo {
+        public Foo() {
+          Console.WriteLine("Hello");
+        }
+      }
+    `);
+  });
+  it("should extract an interface", () => {
+    const content = d`
+    public interface IMyInterface {
+      string Foo();
+    }
+    `;
+    const snippet = extractor.getInterface(content, "IMyInterface");
+    expect(snippet).toBe(d`
+      public interface IMyInterface {
+        string Foo();
+      }
+    `);
+  });
+  it("should extract a function", () => {
+    const content = d`
+    public string Greet(string name) {
+      return "Hello " + name;
+    }
+    `;
+    const snippet = extractor.getFunction(content, "Greet");
+    expect(snippet).toBe(d`
+      public string Greet(string name) {
+        return "Hello " + name;
+      }
+    `);
+  });
+});
+describe("C# Snippet Extractor - Enums", () => {
+  let extractor;
+  beforeEach(() => {
+    extractor = createSnipperExtractor(createCSharpExtractorConfig());
+  });
+  it("should extract a basic enum", async () => {
+    const content = d`
+      public enum Direction
+      {
+          Up,
+          Down,
+          Left,
+          Right
+      }
+    `;
+    const snippet = extractor.getEnum(content, "Direction");
+    expect(snippet).toBe(d`
+      public enum Direction
+      {
+          Up,
+          Down,
+          Left,
+          Right
+      }
+    `);
+  });
+  it("should extract an enum with values", async () => {
+    const content = d`
+      public enum Status
+      {
+          Active = 1,
+          Inactive = 2
+      }
+    `;
+    const snippet = extractor.getEnum(content, "Status");
+    expect(snippet).toBe(d`
+      public enum Status
+      {
+          Active = 1,
+          Inactive = 2
+      }
+    `);
+  });
+});

package/dist/test/testing/snippet-extractor-java.test.js

@@ -0,0 +1,104 @@
+import { d } from "@alloy-js/core/testing";
+import { beforeEach, describe, expect, it } from "vitest";
+import { createJavaExtractorConfig, createSnipperExtractor } from "../../src/testing/index.js";
+describe("Java Snippet Extractor", () => {
+  let extractor;
+  beforeEach(() => {
+    extractor = createSnipperExtractor(createJavaExtractorConfig());
+  });
+  it("should extract a class", () => {
+    const content = d`
+    public class Foo {
+      public Foo() {
+        System.out.println("Hello");
+      }
+    }
+    `;
+    const snippet = extractor.getClass(content, "Foo");
+    expect(snippet).toBe(d`
+      public class Foo {
+        public Foo() {
+          System.out.println("Hello");
+        }
+      }
+    `);
+  });
+  it("should extract an interface", () => {
+    const content = d`
+    public interface MyInterface {
+      String foo();
+    }
+    `;
+    const snippet = extractor.getInterface(content, "MyInterface");
+    expect(snippet).toBe(d`
+      public interface MyInterface {
+        String foo();
+      }
+    `);
+  });
+  it("should extract a function", () => {
+    const content = d`
+    public String greet(String name) {
+      return "Hello " + name;
+    }
+    `;
+    const snippet = extractor.getFunction(content, "greet");
+    expect(snippet).toBe(d`
+      public String greet(String name) {
+        return "Hello " + name;
+      }
+    `);
+  });
+});
+describe("Java Snippet Extractor - Enums", () => {
+  let extractor;
+  beforeEach(() => {
+    extractor = createSnipperExtractor(createJavaExtractorConfig());
+  });
+  it("should extract a basic enum", async () => {
+    const content = d`
+      public enum Direction {
+          UP,
+          DOWN,
+          LEFT,
+          RIGHT
+      }
+    `;
+    const snippet = extractor.getEnum(content, "Direction");
+    expect(snippet).toBe(d`
+      public enum Direction {
+          UP,
+          DOWN,
+          LEFT,
+          RIGHT
+      }
+    `);
+  });
+  it("should extract an enum with constructor values", async () => {
+    const content = d`
+      public enum Status {
+          ACTIVE(1),
+          INACTIVE(2);
+
+          private final int value;
+
+          Status(int value) {
+              this.value = value;
+          }
+      }
+    `;
+    const snippet = extractor.getEnum(content, "Status");
+    expect(snippet).toBe(d`
+      public enum Status {
+          ACTIVE(1),
+          INACTIVE(2);
+
+          private final int value;
+
+          Status(int value) {
+              this.value = value;
+          }
+      }
+    `);
+  });
+});

package/dist/test/testing/snippet-extractor-python.test.js

@@ -0,0 +1,33 @@
+import { d } from "@alloy-js/core/testing";
+import { beforeEach, describe, expect, it } from "vitest";
+import { createPythonExtractorConfig, createSnipperExtractor } from "../../src/testing/index.js";
+describe("Python Snippet Extractor", () => {
+  let extractor;
+  beforeEach(() => {
+    extractor = createSnipperExtractor(createPythonExtractorConfig());
+  });
+  it("should extract a class", () => {
+    const content = d`
+    class Foo:
+        def __init__(self):
+            print("Hello")
+    `;
+    const snippet = extractor.getClass(content, "Foo");
+    expect(snippet).toBe(d`
+      class Foo:
+          def __init__(self):
+              print("Hello")
+    `);
+  });
+  it("should extract a function", () => {
+    const content = d`
+    def greet(name: str) -> str:
+        return f"Hello {name}"
+    `;
+    const snippet = extractor.getFunction(content, "greet");
+    expect(snippet).toBe(d`
+      def greet(name: str) -> str:
+          return f"Hello {name}"
+    `);
+  });
+});