def test_without_transformer(self):
# ARRANGE #
src_file = fs.File('source-file.txt', 'contents of source file')
src_rel_opt_conf = conf_rel_any(RelOptionType.REL_TMP)
checker = integration_check.checker(parse_check.rel_opts_conf_of_single(src_rel_opt_conf.relativity))
file_contents_abs_stx = string_source_abs_stx.StringSourceOfFileAbsStx(
src_rel_opt_conf.path_abs_stx_of_name(src_file.name)
)
# ACT & ASSERT #
checker.check__abs_stx__layouts__std_source_variants__wo_input(
self,
OptionallyOnNewLine(file_contents_abs_stx),
arrangement_w_tcds(
tcds_contents=src_rel_opt_conf.populator_for_relativity_option_root(
DirContents([src_file]))
),
MultiSourceExpectation.of_prim__const(
asrt_string_source.pre_post_freeze__matches_str__const(
src_file.contents,
may_depend_on_external_resources=True,
),
symbol_references=asrt.is_empty_sequence,
)
)
def runTest(self):
# ARRANGE #
to_upper_transformer = StringTransformerSymbolContext.of_primitive(
'TRANSFORMER_SYMBOL',
string_transformers.to_uppercase(),
)
src_file = fs.File('src-file.txt', 'contents of source file')
src_file_rel_conf = conf_rel_hds(RelHdsOptionType.REL_HDS_CASE)
expected_contents = src_file.contents.upper()
path_relativity_variants = PathRelativityVariants({src_file_rel_conf.relativity}, True)
checker = integration_check.checker(RelOptionsConfiguration(path_relativity_variants,
src_file_rel_conf.relativity))
src_file_symbol = ConstantSuffixPathDdvSymbolContext(
'SRC_FILE_SYMBOL',
src_file_rel_conf.relativity,
src_file.name,
path_relativity_variants,
)
string_source_syntax = string_source_abs_stx.TransformedStringSourceAbsStx(
string_source_abs_stx.StringSourceOfFileAbsStx(src_file_symbol.abstract_syntax),
to_upper_transformer.abstract_syntax,
)
symbol_table = SymbolContext.symbol_table_of_contexts([
src_file_symbol,
to_upper_transformer,
])
# ACT & ASSERT #
checker.check__abs_stx__layouts__std_source_variants__wo_input(
self,
OptionallyOnNewLine(string_source_syntax),
arrangement_w_tcds(
symbols=symbol_table,
hds_contents=src_file_rel_conf.populator_for_relativity_option_root__hds(
DirContents([src_file]))
),
MultiSourceExpectation.of_prim__const(
asrt_string_source.pre_post_freeze__matches_str__const(
expected_contents,
may_depend_on_external_resources=True,
),
symbol_references=asrt.matches_sequence([
src_file_symbol.reference_assertion__path_or_string,
to_upper_transformer.reference_assertion,
]),
),
)
def test_here_doc(self):
# ARRANGE #
string_value = rich_str_abs_stx.HereDocAbsStx(
'single line in here doc\n')
string_source_syntax = string_source_abs_stx.StringSourceOfStringAbsStx(
string_value)
CHECKER.check__abs_stx__layouts__source_variants__wo_input(
self, equivalent_source_variants__for_full_line_expr_parse__s__nsc,
OptionallyOnNewLine(
WithinParensAbsStx(string_source_syntax,
end_paren_on_separate_line=True)),
arrangement_w_tcds(),
MultiSourceExpectation.of_prim__const(
asrt_string_source.pre_post_freeze__matches_str__const(
string_value.value,
may_depend_on_external_resources=False,
)))
def _expect_definition_of(
defined_name: str,
symbols_for_evaluation: SymbolTable,
references: Assertion[Sequence[SymbolReference]],
expected_contents: str,
may_depend_on_external_resources: bool,
) -> Expectation[None]:
sdv_expectation = type_sdv_assertions.matches_sdv(
asrt.is_instance(StringSourceSdv),
references=references,
symbols=symbols_for_evaluation,
resolved_value=asrt.anything_goes(),
custom=asrt.is_instance_with(
StringSourceSdv,
str_src_check.execution_assertion(
intgr_arr_exp.Arrangement(
symbols=symbols_for_evaluation,
tcds=TcdsArrangement(),
),
primitive=str_src_check.primitive__const(
asrt_string_source.pre_post_freeze__matches_str__const(
expected_contents,
may_depend_on_external_resources=may_depend_on_external_resources
)
)
)
)
)
expected_container = matches_container(
asrt.equals(ValueType.STRING_SOURCE),
sdv_expectation,
)
return Expectation.phase_agnostic(
symbol_usages=asrt.matches_sequence([
asrt_sym_usage.matches_definition(asrt.equals(defined_name),
expected_container),
]),
symbols_after_main=assert_symbol_table_is_singleton(
defined_name,
expected_container,
)
)
def test__with_transformer(self):
# ARRANGE #
src_file = fs.File('source-file.txt', 'contents of source file')
expected_contents = src_file.contents.upper()
to_upper_transformer = StringTransformerSymbolContext.of_primitive(
'TRANSFORMER_SYMBOL',
string_transformers.to_uppercase(),
)
src_rel_opt_conf = conf_rel_any(RelOptionType.REL_TMP)
transformed_file_contents_abs_stx = string_source_abs_stx.TransformedStringSourceAbsStx(
string_source_abs_stx.StringSourceOfFileAbsStx(
src_rel_opt_conf.path_abs_stx_of_name(src_file.name)
),
to_upper_transformer.abstract_syntax,
)
symbols = to_upper_transformer.symbol_table
checker = integration_check.checker(parse_check.rel_opts_conf_of_single(src_rel_opt_conf.relativity))
# ACT & ASSERT #
checker.check__abs_stx__layouts__std_source_variants__wo_input(
self,
OptionallyOnNewLine(transformed_file_contents_abs_stx),
arrangement_w_tcds(
tcds_contents=src_rel_opt_conf.populator_for_relativity_option_root(
DirContents([src_file])),
symbols=symbols,
),
MultiSourceExpectation.of_prim__const(
asrt_string_source.pre_post_freeze__matches_str__const(
expected_contents,
may_depend_on_external_resources=True,
),
symbol_references=to_upper_transformer.references_assertion,
)
)
def get_assertion_on_primitive(
env: AssertionResolvingEnvironment) -> Assertion[StringSource]:
return asrt_string_source.pre_post_freeze__matches_str__const(
contents_transformation(string_source_contents),
may_depend_on_external_resources=False)
def get_assertion_on_primitive(
env: AssertionResolvingEnvironment) -> Assertion[StringSource]:
path_as_str = str(env.tcds.sds.user_tmp_dir / path_suffix)
return asrt_string_source.pre_post_freeze__matches_str__const(
contents_transformation(path_as_str),
may_depend_on_external_resources=False)
def test_string_transformer_should_be_parsed_as_simple_expression(self):
# ARRANGE #
the_layout = LayoutSpec.of_default()
src_rel_opt_conf = conf_rel_any(RelOptionType.REL_TMP)
src_file = fs.File('source-file.txt', 'contents of source file')
expected_contents = src_file.contents.upper()
to_upper_transformer = StringTransformerSymbolContext.of_primitive(
'TRANSFORMER_SYMBOL',
string_transformers.to_uppercase(),
)
str_trans__unused = StringTransformerSymbolReferenceAbsStx('UNUSED_TRANSFORMER')
transformation_w_infix_op = str_trans_abs_stx.StringTransformerCompositionAbsStx(
[
to_upper_transformer.abstract_syntax,
str_trans__unused,
],
within_parens=False,
allow_elements_on_separate_lines=False,
)
expected_remaining_tokens = TokenSequence.concat([
TokenSequence.singleton(str_trans_abs_stx.names.SEQUENCE_OPERATOR_NAME),
str_trans__unused.tokenization(),
])
expected_remaining_source = expected_remaining_tokens.layout(the_layout)
file_contents_syntax = string_source_abs_stx.TransformedStringSourceAbsStx(
string_source_abs_stx.StringSourceOfFileAbsStx(
src_rel_opt_conf.path_abs_stx_of_name(src_file.name)
),
transformation_w_infix_op
)
checker = integration_check.checker(parse_check.rel_opts_conf_of_single(src_rel_opt_conf.relativity))
# ACT & ASSERT #
checker.check__abs_stx(
self,
file_contents_syntax,
None,
arrangement_w_tcds(
tcds_contents=src_rel_opt_conf.populator_for_relativity_option_root(
fs.DirContents([src_file])
),
symbols=to_upper_transformer.symbol_table
),
Expectation.of_prim__const(
parse=ParseExpectation(
source=asrt_source.source_is_not_at_end(
remaining_source=asrt.equals(expected_remaining_source)
),
symbol_references=to_upper_transformer.references_assertion,
),
primitive=asrt_string_source.pre_post_freeze__matches_str__const(
expected_contents,
may_depend_on_external_resources=True,
)
),
the_layout
)
