mirror of
https://github.com/flatcar/scripts.git
synced 2025-09-24 07:01:13 +02:00
b54301bb9a
Sets up dir paths as follows: test_root/test_name i.e. SimpleUpdate/<num>_stage For example: test_root/testNormalUpdateKeepStateful/2_verify/...test_results... corresponds to the second invocation of verify image. The test results for this stage are stored here. This is a predecessor for storing arbitrary logs e.g. update into this test results root. Change-Id: I7183420b5dcb0d6971aa508a338c048c3557e359 BUG=chromium-os:12211 TEST=With simple a full vm test suite w/ w/out explicit test root set. Review URL: http://codereview.chromium.org/6614029
46 lines
1.6 KiB
Python
46 lines
1.6 KiB
Python
# Copyright (c) 2011 The Chromium OS Authors. All rights reserved.
|
|
# Use of this source code is governed by a BSD-style license that can be
|
|
# found in the LICENSE file.
|
|
|
|
"""Module containing a fake au worker class."""
|
|
|
|
import unittest
|
|
|
|
import au_worker
|
|
|
|
class DummyAUWorker(au_worker.AUWorker):
|
|
"""AU worker that emulates work for an au_worker without actually doing work.
|
|
|
|
Collects different updates that would be generated that can be obtained
|
|
from the class object delta_list.
|
|
"""
|
|
|
|
# Class variable that stores the list of payloads that would be needed.
|
|
delta_list = {}
|
|
|
|
def __init__(self, options, test_results_root):
|
|
au_worker.AUWorker.__init__(self, options, test_results_root)
|
|
self.au_type = options.type
|
|
|
|
def PrepareBase(self, image_path):
|
|
"""Copy how the actual worker would prepare the base image."""
|
|
if self.au_type == 'vm':
|
|
self.PrepareVMBase(image_path)
|
|
else:
|
|
self.PrepareRealBase(image_path)
|
|
|
|
def UpdateImage(self, image_path, src_image_path='', stateful_change='old',
|
|
proxy_port=None, private_key_path=None):
|
|
"""Emulate Update and record the update payload in delta_list."""
|
|
if self.au_type == 'vm' and src_image_path and self._first_update:
|
|
src_image_path = self.vm_image_path
|
|
self._first_update = False
|
|
|
|
# Generate a value that combines delta with private key path.
|
|
val = src_image_path
|
|
if private_key_path: val = '%s+%s' % (val, private_key_path)
|
|
if not self.delta_list.has_key(image_path):
|
|
self.delta_list[image_path] = set([val])
|
|
else:
|
|
self.delta_list[image_path].add(val)
|