Skip to content

Commit c929b7b

Browse files
committed
use sampled_from
1 parent 894590a commit c929b7b

File tree

3 files changed

+10
-10
lines changed

3 files changed

+10
-10
lines changed

.github/workflows/ci.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ on:
88

99
jobs:
1010
format:
11-
name: Format
11+
name: format
1212
runs-on: ubuntu-latest
1313
steps:
1414
- uses: actions/checkout@v4
@@ -18,7 +18,7 @@ jobs:
1818
- run: cargo fmt --check
1919

2020
test:
21-
name: Tests
21+
name: tests
2222
runs-on: ubuntu-latest
2323
steps:
2424
- uses: actions/checkout@v4

hegel-derive/src/lib.rs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -526,11 +526,11 @@ fn derive_enum_generate(input: &DeriveInput, data: &syn::DataEnum) -> TokenStrea
526526

527527
// Generate the schema() implementation
528528
let schema_impl = if data_variants.is_empty() {
529-
// All unit variants - pure enum schema
529+
// All unit variants - pure sampled_from schema
530530
quote! {
531531
fn schema(&self) -> Option<serde_json::Value> {
532532
let variants: Vec<&str> = vec![#(#all_variant_names),*];
533-
Some(serde_json::json!({ "enum": variants }))
533+
Some(serde_json::json!({ "sampled_from": variants }))
534534
}
535535
}
536536
} else {
@@ -553,7 +553,7 @@ fn derive_enum_generate(input: &DeriveInput, data: &syn::DataEnum) -> TokenStrea
553553
fn generate(&self) -> #enum_name {
554554
let variants: Vec<&str> = vec![#(#all_variant_names),*];
555555
let selected: String = hegel::gen::generate_from_schema(
556-
&serde_json::json!({ "enum": variants })
556+
&serde_json::json!({ "sampled_from": variants })
557557
);
558558

559559
match selected.as_str() {
@@ -575,7 +575,7 @@ fn derive_enum_generate(input: &DeriveInput, data: &syn::DataEnum) -> TokenStrea
575575
hegel::gen::group(hegel::gen::labels::ENUM_VARIANT, || {
576576
let variants: Vec<&str> = vec![#(#all_variant_names),*];
577577
let selected: String = hegel::gen::generate_from_schema(
578-
&serde_json::json!({ "enum": variants })
578+
&serde_json::json!({ "sampled_from": variants })
579579
);
580580

581581
match selected.as_str() {

src/gen/combinators.rs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -178,7 +178,7 @@ impl<T: Clone + Send + Sync + serde::Serialize> Generate<T> for SampledFromGener
178178
fn generate(&self) -> T {
179179
crate::assume(!self.elements.is_empty());
180180

181-
// Check if elements are primitive enough for enum schema
181+
// Check if elements are primitive enough for sampled_from schema
182182
if let Some(schema) = self.schema() {
183183
let value: Value = generate_from_schema(&schema);
184184
// Find matching element
@@ -202,7 +202,7 @@ impl<T: Clone + Send + Sync + serde::Serialize> Generate<T> for SampledFromGener
202202
}
203203

204204
fn schema(&self) -> Option<Value> {
205-
// Only use enum schema for JSON-primitive types
205+
// Only use sampled_from schema for JSON-primitive types
206206
let json_values: Vec<Value> = self.elements.iter().map(|e| json!(e)).collect();
207207

208208
// Check if all values are primitives (not objects/arrays)
@@ -214,7 +214,7 @@ impl<T: Clone + Send + Sync + serde::Serialize> Generate<T> for SampledFromGener
214214
});
215215

216216
if all_primitive {
217-
Some(json!({"enum": json_values}))
217+
Some(json!({"sampled_from": json_values}))
218218
} else {
219219
None
220220
}
@@ -253,7 +253,7 @@ impl<'a, T: Clone + Send + Sync + serde::Serialize + serde::de::DeserializeOwned
253253

254254
fn schema(&self) -> Option<Value> {
255255
Some(json!({
256-
"enum": self.elements
256+
"sampled_from": self.elements
257257
}))
258258
}
259259
}

0 commit comments

Comments
 (0)