def test_extract_env_vars_from_xml(self):
    xml_template = '<appengine-web-app xmlns="http://appengine.google.com/ns/1.0">\
                      <application>{}</application>\
                      {}\
                    </appengine-web-app>'

    env_var_section = '<env-variables>\
                         <env-var name="custom-var-1" value="foo"/>\
                         <env-var name="custom-var-2" value="bar"/>\
                       </env-variables>'

    xml = xml_template.format('app-id', env_var_section)
    flexmock(ElementTree).should_receive('parse').\
      and_return(flexmock(getroot=lambda: ElementTree.fromstring(xml)))
    assert len(app_manager_server.extract_env_vars_from_xml('/file.xml')) == 2

    xml = xml_template.format('app-id', '')
    flexmock(ElementTree).should_receive('parse').\
      and_return(flexmock(getroot=lambda: ElementTree.fromstring(xml)))
    assert app_manager_server.extract_env_vars_from_xml('/file.xml') == {}
Example #2
0
    def test_extract_env_vars_from_xml(self):
        xml_template = '<appengine-web-app xmlns="http://appengine.google.com/ns/1.0">\
                      <application>{}</application>\
                      {}\
                    </appengine-web-app>'

        env_var_section = '<env-variables>\
                         <env-var name="custom-var-1" value="foo"/>\
                         <env-var name="custom-var-2" value="bar"/>\
                       </env-variables>'

        xml = xml_template.format('app-id', env_var_section)
        flexmock(ElementTree).should_receive('parse').\
          and_return(flexmock(getroot=lambda: ElementTree.fromstring(xml)))
        assert len(
            app_manager_server.extract_env_vars_from_xml('/file.xml')) == 2

        xml = xml_template.format('app-id', '')
        flexmock(ElementTree).should_receive('parse').\
          and_return(flexmock(getroot=lambda: ElementTree.fromstring(xml)))
        assert app_manager_server.extract_env_vars_from_xml('/file.xml') == {}