Python import_projects示例

示例#1

  def test_import_projects_exception(self):
    self.mock(gitiles_import, 'import_project', mock.Mock())
    gitiles_import.import_project.side_effect = Exception
    self.mock(projects, 'get_refs', mock.Mock(return_value=[]))

    self.mock(projects, 'get_projects', mock.Mock())
    projects.get_projects.return_value = [
      service_config_pb2.Project(
          id='chromium',
          config_location=service_config_pb2.ConfigSetLocation(
            url='https://localhost/chromium/src/',
            storage_type=service_config_pb2.ConfigSetLocation.GITILES,
          ),
      ),
      service_config_pb2.Project(
          id='will-fail',
          config_location=service_config_pb2.ConfigSetLocation(
            url='https://localhost/chromium/src/',
            storage_type=service_config_pb2.ConfigSetLocation.GITILES,
          ),
      )
    ]

    gitiles_import.import_projects()
    self.assertEqual(gitiles_import.import_project.call_count, 2)

示例#2

    def test_import_projects_and_branches(self):
        self.mock(gitiles_import, 'import_config_set', mock.Mock())
        self.mock(projects, 'get_projects', mock.Mock())
        self.mock(projects, 'get_branches', mock.Mock())
        projects.get_projects.return_value = [
            service_config_pb2.Project(
                id='chromium',
                config_location='https://localhost/chromium/src/',
                config_storage_type=service_config_pb2.Project.GITILES,
            ),
            service_config_pb2.Project(
                id='bad_location',
                config_location='https://localhost/',
                config_storage_type=service_config_pb2.Project.GITILES,
            ),
            service_config_pb2.Project(id='non-gitiles', ),
        ]
        BranchType = project_config_pb2.BranchesCfg.Branch
        projects.get_branches.return_value = [
            BranchType(name='master'),
            BranchType(name='release42', config_path='/my-configs'),
        ]

        gitiles_import.import_projects()

        self.assertEqual(gitiles_import.import_config_set.call_count, 3)
        gitiles_import.import_config_set.assert_any_call(
            'projects/chromium', 'https://localhost/chromium/src/+/luci')
        gitiles_import.import_config_set.assert_any_call(
            'projects/chromium/branches/master',
            'https://localhost/chromium/src/+/master/luci')
        gitiles_import.import_config_set.assert_any_call(
            'projects/chromium/branches/release42',
            'https://localhost/chromium/src/+/release42/my-configs')

示例#3

文件： gitiles_import_test.py 项目： rmistry/luci-py

  def test_import_projects_and_refs(self):
    self.mock(gitiles_import, 'import_config_set', mock.Mock())
    self.mock(projects, 'get_projects', mock.Mock())
    self.mock(projects, 'get_refs', mock.Mock())
    projects.get_projects.return_value = [
      service_config_pb2.Project(
          id='chromium',
          config_location=service_config_pb2.ConfigSetLocation(
            url='https://localhost/chromium/src/',
            storage_type=service_config_pb2.ConfigSetLocation.GITILES,
          )
      ),
      service_config_pb2.Project(
          id='bad_location',
          config_location=service_config_pb2.ConfigSetLocation(
            url='https://localhost/',
            storage_type=service_config_pb2.ConfigSetLocation.GITILES,
          ),
      ),
      service_config_pb2.Project(
          id='non-gitiles',
      ),
    ]
    RefType = project_config_pb2.RefsCfg.Ref
    projects.get_refs.return_value = [
      RefType(name='refs/heads/master'),
      RefType(name='refs/heads/release42', config_path='/my-configs'),
    ]

    gitiles_import.import_projects()

    self.assertEqual(gitiles_import.import_config_set.call_count, 3)
    gitiles_import.import_config_set.assert_any_call(
        'projects/chromium', 'https://localhost/chromium/src/+/refs/heads/luci')
    gitiles_import.import_config_set.assert_any_call(
        'projects/chromium/refs/heads/master',
        'https://localhost/chromium/src/+/refs/heads/master/luci')
    gitiles_import.import_config_set.assert_any_call(
        'projects/chromium/refs/heads/release42',
        'https://localhost/chromium/src/+/refs/heads/release42/my-configs')

示例#4

文件： gitiles_import_test.py 项目： pombreda/luci-py

  def test_import_projects_and_branches(self):
    self.mock(gitiles_import, 'import_config_set', mock.Mock())
    self.mock(projects, 'get_projects', mock.Mock())
    self.mock(projects, 'get_branches', mock.Mock())
    projects.get_projects.return_value = [
      service_config_pb2.Project(
          id='chromium',
          config_location='https://localhost/chromium/src/',
          config_storage_type=service_config_pb2.Project.GITILES,
      ),
      service_config_pb2.Project(
          id='bad_location',
          config_location='https://localhost/',
          config_storage_type=service_config_pb2.Project.GITILES,
      ),
      service_config_pb2.Project(
          id='non-gitiles',
      ),
    ]
    BranchType = project_config_pb2.BranchesCfg.Branch
    projects.get_branches.return_value = [
      BranchType(name='master'),
      BranchType(name='release42', config_path='/my-configs'),
    ]

    gitiles_import.import_projects()

    self.assertEqual(gitiles_import.import_config_set.call_count, 3)
    gitiles_import.import_config_set.assert_any_call(
        'projects/chromium', 'https://localhost/chromium/src/+/luci')
    gitiles_import.import_config_set.assert_any_call(
        'projects/chromium/branches/master',
        'https://localhost/chromium/src/+/master/luci')
    gitiles_import.import_config_set.assert_any_call(
        'projects/chromium/branches/release42',
        'https://localhost/chromium/src/+/release42/my-configs')