2020 IaC_installation_generator ,
2121 )
2222from app .template_generators .terraform .docker import (IaC_template_generator_docker )
23-
23+ import os
2424
2525@app .post ("/IaC-basic/" )
2626async def IaC_basic_generation (request :IaCBasicInput ) -> Output :
27-
27+ if os .environ .get ("TEST" ):
28+ return Output (output = 'Terraform developed by hashicorp and it is very usefull' )
2829 generated_prompt = IaC_basics_generator (request )
2930 output = gpt_service (generated_prompt )
3031 return Output (output = output )
3132
3233@app .post ("/IaC-bugfix/" )
3334async def IaC_bugfix_generation (request :IaCBugfixInput ) -> Output :
34-
35+ if os .environ .get ("TEST" ):
36+ return Output (output = 'fix this bug by adding x to the y' )
3537 generated_prompt = IaC_bugfix_generator (request )
3638 output = gpt_service (generated_prompt )
3739 return Output (output = output )
3840
3941
4042@app .post ("/IaC-install/" )
4143async def IaC_install_generation (request :IaCInstallationInput ) -> Output :
42-
44+ if os .environ .get ("TEST" ):
45+ return Output (output = 'apt-get install xyz \n apt-get update (covert them to shell file output)' )
4346 generated_prompt = IaC_installation_generator (request )
4447 output = gpt_service (generated_prompt )
4548 return Output (output = output )
4649
4750@app .post ("/IaC-template/docker" )
4851async def IaC_template_generation_docker (request :IaCTemplateGenerationDocker ) -> Output :
49-
52+ if os .environ .get ("TEST" ):
53+ return Output (output = 'output (nothing special)' )
5054 generated_prompt = IaC_template_generator_docker (request )
5155 output = gpt_service (generated_prompt )
5256 edit_directory_generator ("terraform_generator" ,output )
@@ -55,7 +59,8 @@ async def IaC_template_generation_docker(request:IaCTemplateGenerationDocker) ->
5559
5660@app .post ("/IaC-template/aws/ec2" )
5761async def IaC_template_generation_aws_ec2 (request :IaCTemplateGenerationEC2 ) -> Output :
58-
62+ if os .environ .get ("TEST" ):
63+ return Output (output = 'output (nothing special)' )
5964 generated_prompt = IaC_template_generator_docker (request )
6065 output = gpt_service (generated_prompt )
6166 edit_directory_generator ("terraform_generator" ,output )
@@ -64,7 +69,8 @@ async def IaC_template_generation_aws_ec2(request:IaCTemplateGenerationEC2) -> O
6469
6570@app .post ("/IaC-template/aws/s3" )
6671async def IaC_template_generation_aws_s3 (request :IaCTemplateGenerationS3 ) -> Output :
67-
72+ if os .environ .get ("TEST" ):
73+ return Output (output = 'output (nothing special)' )
6874 generated_prompt = IaC_template_generator_docker (request )
6975 output = gpt_service (generated_prompt )
7076 edit_directory_generator ("terraform_generator" ,output )
@@ -73,7 +79,8 @@ async def IaC_template_generation_aws_s3(request:IaCTemplateGenerationS3) -> Out
7379
7480@app .post ("/IaC-template/aws/iam" )
7581async def IaC_template_generation_aws_iam (request :IaCTemplateGenerationIAM ) -> Output :
76-
82+ if os .environ .get ("TEST" ):
83+ return Output (output = 'output (nothing special)' )
7784 generated_prompt = IaC_template_generator_docker (request )
7885 output = gpt_service (generated_prompt )
7986 edit_directory_generator ("terraform_generator" ,output )
0 commit comments