perms.Grant('creator', perms.Role.MAP_CREATOR, 'xyz.com')

    with test_utils.Login('creator'):
      response = self.DoPost('/xyz.com/.create', 'xsrf_token=XSRF', 302)
      location = response.headers['Location']
      # initial_domain_role is unset so domain_role should be None.
      map_object = model.Map.Get(location.split('/')[-1])
      self.assertTrue(map_object is not None)
      # With no initial_domain_role set, domain_role should be None.
      self.assertEquals(['xyz.com'], map_object.domains)
      self.assertEquals(None, map_object.domain_role)

    # Now set the initial_domain_role for xyz.com.
    with test_utils.RootLogin():
      domain.initial_domain_role = perms.Role.MAP_EDITOR
      domain.Put()

    # Create another map.
    with test_utils.Login('creator'):
      response = self.DoPost('/.create?domain=xyz.com', 'xsrf_token=XSRF', 302)
      location = response.headers['Location']
      # Check the map; initial_domain_role is set so domain_role should be set.
      map_object = model.Map.Get(location.split('/')[-1])
      self.assertTrue(map_object is not None)
      self.assertEquals(['xyz.com'], map_object.domains)
      self.assertEquals(perms.Role.MAP_EDITOR, map_object.domain_role)


if __name__ == '__main__':
  test_utils.main()

import os
import pytest
import yaml

from test_utils import TEST_DIR, main
from ksgen.settings import load_configuration


def test_invalid_yml():
    path = os.path.join(TEST_DIR, 'data', 'yml-syntax')

    yaml_path = os.path.join(path, 'dashes.yml')
    with pytest.raises(yaml.scanner.ScannerError):
        load_configuration(yaml_path)

    yaml_path = os.path.join(path, 'spaces.yml')
    with pytest.raises(yaml.parser.ParserError):
        load_configuration(yaml_path)

    yaml_path = os.path.join(path, 'tab.yml')
    with pytest.raises(yaml.scanner.ScannerError):
        load_configuration(yaml_path)

    yaml_path = os.path.join(path, 'map.yml')
    with pytest.raises(TypeError):
        load_configuration(yaml_path)


if __name__ == '__main__':
    main(locals())