File: cipd_test.py

package info (click to toggle)
chromium 138.0.7204.157-1
  • links: PTS, VCS
  • area: main
  • in suites: sid, trixie
  • size: 6,071,864 kB
  • sloc: cpp: 34,936,859; ansic: 7,176,967; javascript: 4,110,704; python: 1,419,953; asm: 946,768; xml: 739,967; pascal: 187,324; sh: 89,623; perl: 88,663; objc: 79,944; sql: 50,304; cs: 41,786; fortran: 24,137; makefile: 21,806; php: 13,980; tcl: 13,166; yacc: 8,925; ruby: 7,485; awk: 3,720; lisp: 3,096; lex: 1,327; ada: 727; jsp: 228; sed: 36
file content (44 lines) | stat: -rwxr-xr-x 1,351 bytes parent folder | download | duplicates (4)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
#!/usr/bin/env vpython3
# Copyright 2024 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Tests for cipd.py"""

import pathlib
import subprocess
import tempfile
import unittest
from unittest import mock

import cipd


class FetchBundleTest(unittest.TestCase):

  def setUp(self):
    self.tmp_dir = pathlib.Path(tempfile.mkdtemp())
    patch_cipd_dir = mock.patch('cipd._CIPD_ROOT_BASE_DIR', self.tmp_dir)
    patch_cipd_dir.start()
    self.addCleanup(patch_cipd_dir.stop)

    self.subp_mock = mock.MagicMock()
    patch_subp = mock.patch('subprocess.check_call', self.subp_mock)
    patch_subp.start()
    self.addCleanup(patch_subp.stop)

  def testBasic(self):
    bundle_dir = cipd.fetch_recipe_bundle('chromium', True)
    self.assertEqual(bundle_dir, self.tmp_dir.joinpath('chromium'))

    bundle_dir = cipd.fetch_recipe_bundle('chrome', True)
    self.assertEqual(bundle_dir, self.tmp_dir.joinpath('chrome'))

  def testUnknownProject(self):
    # Unknown project defaults to chrome for safety.
    bundle_dir = cipd.fetch_recipe_bundle('unknown-project', True)
    self.assertIn(cipd._CHROME_RECIPE_BUNDLE, self.subp_mock.call_args.args[0])
    self.assertEqual(bundle_dir, self.tmp_dir.joinpath('unknown-project'))


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