xdZddlmZddlmZddlmZddlZddlZddlZddlZddl Z ddl m Z Gdde Z Gd d eZGd d e j Zy) z8Unit test support library for GAE Externalized Runtimes.)absolute_import)division)print_functionN) ext_runtimeceZdZdZy)InvalidRuntimez?Raised when the runtime directory is doesn't match the runtime.N)__name__ __module__ __qualname____doc__+lib/third_party/gae_ext_runtime/testutil.pyrrsIrrceZdZdZdZy) AppInfoFakez'Serves as a fake for an AppInfo object.c|SNr selfs rToDictzAppInfoFake.ToDict%s rN)r r r r rr rrrr"s 1rrc`eZdZdZdZdZdZddZddZddZ dd Z d Z d Z d Z d ZdZy)TestBasezUnit testing base class. Derived classes must define a setUp() method that sets a runtime_def_root attribute containing the path to the root directory of the runtime. ctj|_tj|_|j t|ddy)Nruntime_def_rootzoYour test suite must define a setUp() method that sets a runtime_def_root attribute to the root of the runtime.)rDefaultExecutionEnvironmentexec_envtempfilemkdtemp temp_path assertTruehasattrrs rsetUpzTestBase.setUp0s=#??A !))+ &89# $rcBtj|jyr)shutilrmtreerrs rtearDownzTestBase.tearDown8s dnn%rc||_y)a9Set the execution environment used by generate_configs. If this is not set, an instance of ext_runtime.DefaultExecutionEnvironment is used. Args: exec_env: (ext_runtime.ExecutionEnvironment) The execution environment to be used for config generation. N)r)rrs rset_execution_environmentz"TestBase.set_execution_environment;s ! rNc tjj|j|j}|xstj di|}t |j|j|j|}|S)aLoad the runtime definition. Args: params: (ext_runtime.Params) Runtime parameters. DEPRECATED. Use the keyword args, instead. **kwargs: ({str: object, ...}) If specified, these are the arguments to the ext_runtime.Params() constructor (valid args are at this time are: appinfo, custom and deploy, check ext_runtime.Params() for full details) Returns: configurator or None if configurator didn't match r ) rExternalizedRuntimeLoadrrParamsprintrDetectrrparamskwargsrt configurators rmaybe_get_configuratorzTestBase.maybe_get_configuratorGsj , , 1 1$2G2G26--A7;--77 fmmoyy8 rc n|j|fi|}|sy|j|jS)adLoad the runtime definition and generate configs from it. Args: params: (ext_runtime.Params) Runtime parameters. DEPRECATED. Use the keyword args, instead. **kwargs: ({str: object, ...}) If specified, these are the arguments to the ext_runtime.Params() constructor (valid args are at this time are: appinfo, custom and deploy, check ext_runtime.Params() for full details) Returns: (bool) Returns True if files are generated, False if not, None if configurator didn't match N)r4PrebuildGenerateConfigsrr0r1r3s rgenerate_configszTestBase.generate_configs\s<3t226DVD ++--rc |j|fi|}|s/tdj|j|j|j |j S)aLoad the runtime definition and generate configs from it. Args: params: (ext_runtime.Params) Runtime parameters. DEPRECATED. Use the keyword args, instead. **kwargs: ({str: object, ...}) If specified, these are the arguments to the ext_runtime.Params() constructor (valid args are at this time are: appinfo, custom and deploy, check ext_runtime.Params() for full details) Returns: ([ext_runtime.GeneratedFile, ...]) Returns list of generated files. Raises: InvalidRuntime: Couldn't detect a matching runtime. z/Runtime defined in {} did not detect code in {})r4rformatrrr6GenerateConfigDatar8s rgenerate_config_datazTestBase.generate_config_datassi"3t226DVD  "..4fT5J5J59^^/EF F ..00rc tjj|j|j}|xstj di|}|j |j|}|S)aYLoad the runtime definition and generate configs from it. Args: params: (ext_runtime.Params) Runtime parameters. DEPRECATED. Use the keyword args, instead. **kwargs: ({str: object, ...}) If specified, these are the arguments to the ext_runtime.Params() constructor (valid args are at this time are: appinfo, custom and deploy, check ext_runtime.Params() for full details) Returns: (ext_runtime.Configurator or None) the identified runtime if found, None if not. r )rr*r+rrr,r.rr/s rdetectzTestBase.detects\ , , 1 1$2G2G26--A7;--77yy8 rcVtjj|jg|S)zReturns the fully qualified path for a relative filename. e.g. self.full_path('foo', 'bar', 'baz') -> '/temp/path/foo/bar/baz' Args: *path_components: ([str, ...]) Path components. Returns: (str) )ospathjoinr)rpath_componentss r full_pathzTestBase.full_paths ww||DNN=_==rcttjj|j|d5}|j |dddy#1swYyxYw)Nw)openrArBrCrwrite)rfilenamecontentsfps r write_filezTestBase.write_files9 "'',,t~~x8# >" HHX ? > >s AActtjj|jg|5}|j cdddS#1swYyxYw)aRead the entire contents of the file. Returns the entire contents of the file identified by a set of arguments forming a path relative to the root of the runtime definition. Args: *args: A set of path components (see full_path()). Note that these are relative to the runtime definition root, not the temporary directory. N)rHrArBrCrread)rargsrLs rread_runtime_def_filezTestBase.read_runtime_def_files;"'',,t44 = =s AAc|j|}|jtjj |t |5}|j }ddd|j|y#1swYxYw)zAssert that the specified file exists with the given contents. Args: filename: (str) New file name. contents: (str) File contents. N)rEr rArBexistsrHrO assertEqual)rrJrK full_namerLactual_contentss r assert_file_exists_with_contentsz)TestBase.assert_file_exists_with_contentss\NN8,  y12 )_ ggiO ?3_s A77Bc|D]/}|j|k(s|j|j|y|jdj ||y)Nz+filename {} not found in generated files {})rJrTrKfailr;)r gen_filesrJrKgen_files r#assert_genfile_exists_with_contentsz,TestBase.assert_genfile_exists_with_contentssT!H   ( *x00(;" ))AHH # $rr)r r r r r"r&r(r4r9r=r?rErMrQrWr\r rrrr)sC $& !*..16, >  4$rr)r __future__rrrloggingrAr$runittestgae_ext_runtimer ExceptionrdictrTestCaserr rrrdsS?&% 'JYJ$p$x  p$r