feat: add ability to reference validators in tests closes #632
This commit is contained in:
@@ -88,6 +88,23 @@ impl<'a> CodeGenerator<'a> {
|
||||
self.defined_functions = IndexMap::new();
|
||||
}
|
||||
|
||||
pub fn insert_function(
|
||||
&mut self,
|
||||
module_name: String,
|
||||
function_name: String,
|
||||
variant_name: String,
|
||||
value: &'a TypedFunction,
|
||||
) -> Option<&'a TypedFunction> {
|
||||
self.functions.insert(
|
||||
FunctionAccessKey {
|
||||
module_name,
|
||||
function_name,
|
||||
variant_name,
|
||||
},
|
||||
value,
|
||||
)
|
||||
}
|
||||
|
||||
pub fn generate(
|
||||
&mut self,
|
||||
TypedValidator {
|
||||
|
||||
@@ -162,7 +162,7 @@ impl fmt::Display for Token {
|
||||
Token::Fn => "fn",
|
||||
Token::If => "if",
|
||||
Token::Else => "else",
|
||||
Token::Use => "import",
|
||||
Token::Use => "use",
|
||||
Token::Let => "let",
|
||||
Token::Opaque => "opaque",
|
||||
Token::Pub => "pub",
|
||||
|
||||
Reference in New Issue
Block a user