blob: 42f08fcc84f43ef2ffcc67482d389887ee934b32 [file] [log] [blame]
#! /usr/bin/env vpython
#
# Copyright 2018 The Chromium 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 tempfile
import unittest
import update_sdk
class ChangeVersionInGNITests(unittest.TestCase):
def setUp(self):
self._temp_dir = tempfile.mkdtemp()
self._gni_file_path = os.path.join(self._temp_dir, 'test_file.gni')
def tearDown(self):
shutil.rmtree(self._temp_dir)
def testBasic(self):
with open(self._gni_file_path, 'w') as gni_file:
gni_file.write('sample_gn_version_var = "1.2.3.4"')
package = 'sample_package'
arg_version = '2.3.4.5'
gn_args_dict = {
package: 'sample_gn_version_var'
}
update_sdk.ChangeVersionInGNI(
package, arg_version, gn_args_dict, self._gni_file_path, False)
with open(self._gni_file_path, 'r') as gni_file:
self.assertEquals(
'sample_gn_version_var = "2.3.4.5"',
gni_file.read().strip())
def testNoQuotes(self):
with open(self._gni_file_path, 'w') as gni_file:
gni_file.write('sample_gn_version_var = 1234')
package = 'sample_package'
arg_version = '2345'
gn_args_dict = {
package: 'sample_gn_version_var'
}
update_sdk.ChangeVersionInGNI(
package, arg_version, gn_args_dict, self._gni_file_path, False)
with open(self._gni_file_path, 'r') as gni_file:
self.assertEquals(
'sample_gn_version_var = 2345',
gni_file.read().strip())
if __name__ == '__main__':
unittest.main()