# Copyright (c) 2010 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.

import logging, time
from autotest_lib.client.bin import test, utils

ROOTFS_LSB_RELEASE = '/etc/lsb-release'
STATEFUL_LSB_RELEASE = '/mnt/stateful_partition/etc/lsb-release'

class autoupdate_SetUp(test.test):
    version = 1

    def _override_lsb_release(self, devserver):
        """Override the lsb-release with one in the stateful partition
        """
        auserver_key = 'CHROMEOS_AUSERVER'
        devserver_key = 'CHROMEOS_DEVSERVER'
        track_key = 'CHROMEOS_RELEASE_TRACK'

        if not devserver.startswith('http'):
            devserver = 'http://%s' % devserver

        new_auserver_value = '%s/update' % devserver
        new_devserver_value = devserver
        new_track_value = 'test-channel'

        # Create initial lsb-release file in stateful partition.
        comment = '# Override %s and %s on %s' % (auserver_key,
                                                  devserver_key,
                                                  time.asctime())
        utils.write_one_line(STATEFUL_LSB_RELEASE, comment)

        # Read and override auserver and devserver values.
        lsb_release = utils.read_keyval(ROOTFS_LSB_RELEASE)
        lsb_release[auserver_key] = new_auserver_value
        lsb_release[devserver_key] = new_devserver_value
        lsb_release[track_key] = new_track_value

        utils.write_keyval(STATEFUL_LSB_RELEASE, lsb_release)
        logging.info('Current release: % s' % lsb_release['GOOGLE_RELEASE'])

    def run_once(self, devserver):
        self._override_lsb_release(devserver)