@@ -436,34 +436,34 @@ def print_failed_test_results(e2e_target: str, failed_test_results: List[TestRes
436436 )
437437
438438
439- @click .command ()
440- @click .option (
441- "--target" ,
442- "-t" ,
443- type = str ,
444- default = "postgres" ,
445- help = "The dbt target to run the tests against." ,
446- )
447- @click .option (
448- "--e2e-type" ,
449- "-e" ,
450- type = str ,
451- default = "default" ,
452- help = "The type of e2e tests to run." ,
453- )
454- @click .option (
455- "--generate-data" ,
456- "-g" ,
457- type = bool ,
458- default = False ,
459- help = "Set to true if you want to re-generate fake data." ,
460- )
461- @click .option (
462- "--clear-tests" ,
463- type = bool ,
464- default = True ,
465- help = "Set to true if you want to clear the tests." ,
466- )
439+ # @click.command()
440+ # @click.option(
441+ # "--target",
442+ # "-t",
443+ # type=str,
444+ # default="postgres",
445+ # help="The dbt target to run the tests against.",
446+ # )
447+ # @click.option(
448+ # "--e2e-type",
449+ # "-e",
450+ # type=str,
451+ # default="default",
452+ # help="The type of e2e tests to run.",
453+ # )
454+ # @click.option(
455+ # "--generate-data",
456+ # "-g",
457+ # type=bool,
458+ # default=False,
459+ # help="Set to true if you want to re-generate fake data.",
460+ # )
461+ # @click.option(
462+ # "--clear-tests",
463+ # type=bool,
464+ # default=True,
465+ # help="Set to true if you want to clear the tests.",
466+ # )
467467def main (target , e2e_type , generate_data , clear_tests ):
468468 if generate_data :
469469 generate_fake_data ()
@@ -472,7 +472,20 @@ def main(target, e2e_type, generate_data, clear_tests):
472472
473473 if e2e_type == "default" :
474474 e2e_types = [
475+ "seasonal_volume" ,
476+ "table" ,
477+ "column" ,
478+ "directional_anomalies" ,
479+ "backfill_days" ,
475480 "schema" ,
481+ "regular" ,
482+ "artifacts" ,
483+ "error_test" ,
484+ "error_model" ,
485+ "error_snapshot" ,
486+ "dimension" ,
487+ "create_table" ,
488+ "non_dbt_models" ,
476489 ]
477490 else :
478491 e2e_types = [e2e_type ]
@@ -501,4 +514,4 @@ def main(target, e2e_type, generate_data, clear_tests):
501514
502515
503516if __name__ == "__main__" :
504- main ()
517+ main ("postgres" , "default" , False , True )
0 commit comments