test(tvix/eval): impl Arbitrary for Value
Impl Arbitrary for Value (and NixAttrs and NixList) in the same way we did for NixString. Value currently only generates non-"internal" values (no thunks, AttrNotFound, etc.) and can't generate functions (builtins or closures), because those'd require full generation of tvix bytecode, which is a bit more work than I'd like to do now - there's a `todo!` left in the code for a place where we could allow opting-in to internal values and functions later. Change-Id: I07a59e2b1d89cfaa912d4ecebd642caf4ddb040a Reviewed-on: https://cl.tvl.fyi/c/depot/+/6627 Autosubmit: grfn <grfn@gws.fyi> Tested-by: BuildkiteCI Reviewed-by: tazjin <tazjin@tvl.su>
This commit is contained in:
parent
221d3b9485
commit
f8b3806720
4 changed files with 128 additions and 0 deletions
|
|
@ -149,6 +149,35 @@ impl PartialEq for NixAttrs {
|
|||
}
|
||||
}
|
||||
|
||||
#[cfg(feature = "arbitrary")]
|
||||
mod arbitrary {
|
||||
use super::*;
|
||||
|
||||
use proptest::prelude::*;
|
||||
use proptest::prop_oneof;
|
||||
use proptest::strategy::{BoxedStrategy, Just, Strategy};
|
||||
|
||||
impl Arbitrary for NixAttrs {
|
||||
type Parameters = <BTreeMap<NixString, Value> as Arbitrary>::Parameters;
|
||||
|
||||
type Strategy = BoxedStrategy<Self>;
|
||||
|
||||
fn arbitrary_with(args: Self::Parameters) -> Self::Strategy {
|
||||
prop_oneof![
|
||||
Just(Self(AttrsRep::Empty)),
|
||||
(
|
||||
any_with::<Value>(args.2.clone()),
|
||||
any_with::<Value>(args.2.clone())
|
||||
)
|
||||
.prop_map(|(name, value)| Self(AttrsRep::KV { name, value })),
|
||||
any_with::<BTreeMap<NixString, Value>>(args)
|
||||
.prop_map(|map| Self(AttrsRep::Map(map)))
|
||||
]
|
||||
.boxed()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl NixAttrs {
|
||||
/// Return an attribute set containing the merge of the two
|
||||
/// provided sets. Keys from the `other` set have precedence.
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue