build/android/pylib/output/remote_output_manager_test.py

#! /usr/bin/env vpython3
# Copyright 2017 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

# pylint: disable=protected-access
"""Unit tests for remote_output_manager.py.

Example usage:
  vpython3 remote_output_manager_test.py
"""

from pathlib import Path
import sys
import unittest

build_android_path = Path(__file__).parents[2]
sys.path.append(str(build_android_path))

lib_path_root = Path(__file__).parents[3] / 'lib'
sys.path.append(str(lib_path_root))

from pylib.base import output_manager
from pylib.base import output_manager_test_case
from pylib.output import remote_output_manager

import mock  # pylint: disable=import-error


@mock.patch('lib.common.google_storage_helper')
class RemoteOutputManagerTest(output_manager_test_case.OutputManagerTestCase):

  def setUp(self):
    self._output_manager = remote_output_manager.RemoteOutputManager(
        'this-is-a-fake-bucket')

  def testUsableTempFile(self, google_storage_helper_mock):
    del google_storage_helper_mock
    self.assertUsableTempFile(
        self._output_manager._CreateArchivedFile('test_file', 'test_subdir',
                                                 output_manager.Datatype.TEXT,
                                                 None))


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