scripts/fuzzing/test/cipd_test.py

#!/usr/bin/env python
# Copyright 2019 The Fuchsia Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import os
import shutil
import unittest
import tempfile

import test_env
from lib.args import Args
from lib.cipd import Cipd
from lib.fuzzer import Fuzzer

from cipd_mock import MockCipd
from device_mock import MockDevice


class TestCipd(unittest.TestCase):

  def test_from_args(self):
    fuzzer = Fuzzer(MockDevice(), u'mock-package1', u'mock-target3')
    parser = Args.make_parser('description')

    args = parser.parse_args(['1/3'])
    cipd = Cipd.from_args(fuzzer, args)
    self.assertTrue(os.path.exists(cipd.root))

    tmp_dir = tempfile.mkdtemp()
    try:
      args = parser.parse_args(['1/3', '--staging', tmp_dir])
      cipd = Cipd.from_args(fuzzer, args)
      self.assertEqual(tmp_dir, cipd.root)
    finally:
      shutil.rmtree(tmp_dir)

  def test_list(self):
    mock_cipd = MockCipd()
    mock_cipd.list()
    self.assertEqual(
        mock_cipd.last, mock_cipd._bin +
        ' instances fuchsia/test_data/fuzzing/' + str(mock_cipd.fuzzer))

  def test_install(self):
    mock_cipd = MockCipd()
    mock_cipd.install()
    self.assertEqual(
        mock_cipd.last, mock_cipd._bin + ' install fuchsia/test_data/fuzzing/' +
        str(mock_cipd.fuzzer) + ' --root ' + mock_cipd.root)

  def test_create(self):
    mock_cipd = MockCipd()
    mock_cipd.create()
    self.assertEqual(
        mock_cipd.last, mock_cipd._bin + ' create --pkg-def ' +
        os.path.join(mock_cipd.root, 'cipd.yaml') + ' --ref latest')


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