319 lines
11 KiB
Python
319 lines
11 KiB
Python
|
# Copyright (c) 2016 The Chromium Embedded Framework Authors. All rights
|
||
|
|
# reserved. Use of this source code is governed by a BSD-style license that
|
||
|
|
# can be found in the LICENSE file.
|
||
|
|
|
||
|
|
from cef_json_builder import cef_json_builder
|
||
|
|
import datetime
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
class TestCefJSONBuilder(unittest.TestCase):
|
||
|
|
|
||
|
|
# Write builder contents to string and then read in.
|
||
|
|
def _verify_write_read(self, builder):
|
||
|
|
output = str(builder)
|
||
|
|
builder2 = cef_json_builder()
|
||
|
|
builder2.load(output)
|
||
|
|
self.assertEqual(output, str(builder2))
|
||
|
|
|
||
|
|
# Add a file record for testing purposes.
|
||
|
|
def _add_test_file(self, builder, platform='linux32', version='3.2704.1414.g185cd6c',
|
||
|
|
type='standard', attrib_idx=0, shouldfail=False):
|
||
|
|
name = cef_json_builder.get_file_name(version, platform, type) + '.tar.gz'
|
||
|
|
|
||
|
|
# Some random attribute information. sha1 must be different to trigger replacement.
|
||
|
|
attribs = [
|
||
|
|
{
|
||
|
|
'date_str': '2016-05-18T22:42:15.487Z',
|
||
|
|
'date_val': datetime.datetime(2016, 5, 18, 22, 42, 15, 487000),
|
||
|
|
'sha1': '2d48ee05ea6385c8fe80879c98c5dd505ad4b100',
|
||
|
|
'size': 48395610
|
||
|
|
},
|
||
|
|
{
|
||
|
|
'date_str': '2016-05-14T22:42:15.487Z',
|
||
|
|
'date_val': datetime.datetime(2016, 5, 14, 22, 42, 15, 487000),
|
||
|
|
'sha1': '2d48ee05ea6385c8fe80879c98c5dd505ad4b200',
|
||
|
|
'size': 48395620
|
||
|
|
}
|
||
|
|
]
|
||
|
|
|
||
|
|
# Populate the Chromium version to avoid queries.
|
||
|
|
chromium_version = '49.0.2705.50'
|
||
|
|
self.assertEqual(chromium_version, builder.set_chromium_version(version, chromium_version))
|
||
|
|
self.assertEqual(0, builder.get_query_count())
|
||
|
|
|
||
|
|
result = builder.add_file(name,
|
||
|
|
attribs[attrib_idx]['size'],
|
||
|
|
attribs[attrib_idx]['date_str'],
|
||
|
|
attribs[attrib_idx]['sha1'])
|
||
|
|
# Failure should be expected when adding the same file multiple times with the same sha1.
|
||
|
|
self.assertEqual(not shouldfail, result)
|
||
|
|
|
||
|
|
# Return the result expected from get_files().
|
||
|
|
return {
|
||
|
|
'chromium_version': chromium_version,
|
||
|
|
'sha1': attribs[attrib_idx]['sha1'],
|
||
|
|
'name': name,
|
||
|
|
'platform': platform,
|
||
|
|
'last_modified': attribs[attrib_idx]['date_val'],
|
||
|
|
'cef_version': version,
|
||
|
|
'type': type,
|
||
|
|
'size': attribs[attrib_idx]['size']
|
||
|
|
}
|
||
|
|
|
||
|
|
# Test with no file contents.
|
||
|
|
def test_empty(self):
|
||
|
|
builder = cef_json_builder()
|
||
|
|
|
||
|
|
self._verify_write_read(builder)
|
||
|
|
|
||
|
|
files = builder.get_files()
|
||
|
|
self.assertEqual(len(files), 0)
|
||
|
|
|
||
|
|
# Test add/get of a single file with the specified type.
|
||
|
|
def _test_add_file(self, type):
|
||
|
|
builder = cef_json_builder()
|
||
|
|
|
||
|
|
expected = self._add_test_file(builder, type=type)
|
||
|
|
self._verify_write_read(builder)
|
||
|
|
|
||
|
|
files = builder.get_files()
|
||
|
|
self.assertEqual(len(files), 1)
|
||
|
|
self.assertEqual(expected, files[0])
|
||
|
|
|
||
|
|
# Test add/get of a standard type file.
|
||
|
|
def test_add_standard_file(self):
|
||
|
|
self._test_add_file('standard')
|
||
|
|
|
||
|
|
# Test add/get of a minimal type file.
|
||
|
|
def test_add_minimal_file(self):
|
||
|
|
self._test_add_file('minimal')
|
||
|
|
|
||
|
|
# Test add/get of a client type file.
|
||
|
|
def test_add_client_file(self):
|
||
|
|
self._test_add_file('client')
|
||
|
|
|
||
|
|
# Test add/get of a debug_symbols type file.
|
||
|
|
def test_add_debug_symbols_file(self):
|
||
|
|
self._test_add_file('debug_symbols')
|
||
|
|
|
||
|
|
# Test add/get of a release_symbols type file.
|
||
|
|
def test_add_release_symbols_file(self):
|
||
|
|
self._test_add_file('release_symbols')
|
||
|
|
|
||
|
|
# Test get_files() behavior with a single file.
|
||
|
|
def test_get_files_single(self):
|
||
|
|
builder = cef_json_builder()
|
||
|
|
|
||
|
|
# Specify all values just in case the defaults change.
|
||
|
|
expected = self._add_test_file(builder,
|
||
|
|
platform='linux32', version='3.2704.1414.g185cd6c', type='standard')
|
||
|
|
|
||
|
|
# No filter.
|
||
|
|
files = builder.get_files()
|
||
|
|
self.assertEqual(len(files), 1)
|
||
|
|
self.assertEqual(expected, files[0])
|
||
|
|
|
||
|
|
# Platform filter.
|
||
|
|
files = builder.get_files(platform='linux32')
|
||
|
|
self.assertEqual(len(files), 1)
|
||
|
|
self.assertEqual(expected, files[0])
|
||
|
|
files = builder.get_files(platform='linux64')
|
||
|
|
self.assertEqual(len(files), 0)
|
||
|
|
|
||
|
|
# Version filter exact.
|
||
|
|
files = builder.get_files(version='3.2704.1414.g185cd6c')
|
||
|
|
self.assertEqual(len(files), 1)
|
||
|
|
self.assertEqual(expected, files[0])
|
||
|
|
|
||
|
|
# Version filter partial.
|
||
|
|
files = builder.get_files(version='3.2704')
|
||
|
|
self.assertEqual(len(files), 1)
|
||
|
|
self.assertEqual(expected, files[0])
|
||
|
|
files = builder.get_files(version='3.2623')
|
||
|
|
self.assertEqual(len(files), 0)
|
||
|
|
|
||
|
|
# Type filter.
|
||
|
|
files = builder.get_files(type='standard')
|
||
|
|
self.assertEqual(len(files), 1)
|
||
|
|
self.assertEqual(expected, files[0])
|
||
|
|
files = builder.get_files(type='client')
|
||
|
|
self.assertEqual(len(files), 0)
|
||
|
|
|
||
|
|
# All filters.
|
||
|
|
files = builder.get_files(platform='linux32', version='3.2704', type='standard')
|
||
|
|
self.assertEqual(len(files), 1)
|
||
|
|
self.assertEqual(expected, files[0])
|
||
|
|
files = builder.get_files(platform='linux32', version='3.2704', type='minimal')
|
||
|
|
self.assertEqual(len(files), 0)
|
||
|
|
files = builder.get_files(platform='linux32', version='3.2623', type='standard')
|
||
|
|
self.assertEqual(len(files), 0)
|
||
|
|
files = builder.get_files(platform='linux64', version='3.2704', type='standard')
|
||
|
|
self.assertEqual(len(files), 0)
|
||
|
|
|
||
|
|
# Test add/get of multiple files.
|
||
|
|
def test_add_multiple_files(self):
|
||
|
|
builder = cef_json_builder()
|
||
|
|
|
||
|
|
expected = []
|
||
|
|
|
||
|
|
platforms = cef_json_builder.get_platforms()
|
||
|
|
versions = ['3.2704.1414.g185cd6c', '3.2704.1400.gde36543']
|
||
|
|
types = cef_json_builder.get_distrib_types()
|
||
|
|
for platform in platforms:
|
||
|
|
for version in versions:
|
||
|
|
for type in types:
|
||
|
|
expected.append(self._add_test_file(builder, platform=platform, type=type, version=version))
|
||
|
|
|
||
|
|
self._verify_write_read(builder)
|
||
|
|
|
||
|
|
# No filter.
|
||
|
|
files = builder.get_files()
|
||
|
|
self.assertEqual(len(files), len(platforms) * len(versions) * len(types))
|
||
|
|
|
||
|
|
# Version filter.
|
||
|
|
files = builder.get_files(version=version)
|
||
|
|
self.assertEqual(len(files), len(platforms) * len(types))
|
||
|
|
|
||
|
|
# Type filter.
|
||
|
|
files = builder.get_files(type='client')
|
||
|
|
self.assertEqual(len(files), len(platforms) * len(versions))
|
||
|
|
|
||
|
|
# Platform filter.
|
||
|
|
files = builder.get_files(platform='windows32')
|
||
|
|
self.assertEqual(len(files), len(types) * len(versions))
|
||
|
|
|
||
|
|
# All filters.
|
||
|
|
idx = 0
|
||
|
|
for platform in platforms:
|
||
|
|
for version in versions:
|
||
|
|
for type in types:
|
||
|
|
files = builder.get_files(platform=platform, type=type, version=version)
|
||
|
|
self.assertEqual(len(files), 1)
|
||
|
|
self.assertEqual(expected[idx], files[0])
|
||
|
|
idx = idx + 1
|
||
|
|
|
||
|
|
# Test add/get/replace of multiple files.
|
||
|
|
def test_replace_all_files(self):
|
||
|
|
builder = cef_json_builder()
|
||
|
|
|
||
|
|
version = '3.2704.1414.g185cd6c'
|
||
|
|
platforms = ['linux32', 'linux64']
|
||
|
|
types = ['standard', 'minimal']
|
||
|
|
|
||
|
|
# Initial file versions.
|
||
|
|
for platform in platforms:
|
||
|
|
for type in types:
|
||
|
|
self._add_test_file(builder, platform=platform, type=type, version=version)
|
||
|
|
|
||
|
|
# No filter.
|
||
|
|
files = builder.get_files()
|
||
|
|
self.assertEqual(len(files), len(platforms) * len(types))
|
||
|
|
|
||
|
|
expected = []
|
||
|
|
|
||
|
|
# Replace all file versions (due to new sha1).
|
||
|
|
for platform in platforms:
|
||
|
|
for type in types:
|
||
|
|
expected.append(self._add_test_file(builder,
|
||
|
|
platform=platform, type=type, version=version, attrib_idx=1))
|
||
|
|
|
||
|
|
# No filter.
|
||
|
|
files = builder.get_files()
|
||
|
|
self.assertEqual(len(files), len(platforms) * len(types))
|
||
|
|
|
||
|
|
# All filters.
|
||
|
|
idx = 0
|
||
|
|
for platform in platforms:
|
||
|
|
for type in types:
|
||
|
|
files = builder.get_files(platform=platform, type=type, version=version)
|
||
|
|
self.assertEqual(len(files), 1)
|
||
|
|
self.assertEqual(expected[idx], files[0])
|
||
|
|
idx = idx + 1
|
||
|
|
|
||
|
|
# Test add/get/no replace of multiple files.
|
||
|
|
def test_replace_no_files(self):
|
||
|
|
builder = cef_json_builder()
|
||
|
|
|
||
|
|
version = '3.2704.1414.g185cd6c'
|
||
|
|
platforms = ['linux32', 'linux64']
|
||
|
|
types = ['standard', 'minimal']
|
||
|
|
|
||
|
|
# Initial file versions.
|
||
|
|
for platform in platforms:
|
||
|
|
for type in types:
|
||
|
|
self._add_test_file(builder, platform=platform, type=type, version=version)
|
||
|
|
|
||
|
|
# No filter.
|
||
|
|
files = builder.get_files()
|
||
|
|
self.assertEqual(len(files), len(platforms) * len(types))
|
||
|
|
|
||
|
|
expected = []
|
||
|
|
|
||
|
|
# Replace no file versions (due to same sha1).
|
||
|
|
for platform in platforms:
|
||
|
|
for type in types:
|
||
|
|
expected.append(self._add_test_file(builder,
|
||
|
|
platform=platform, type=type, version=version, shouldfail=True))
|
||
|
|
|
||
|
|
# No filter.
|
||
|
|
files = builder.get_files()
|
||
|
|
self.assertEqual(len(files), len(platforms) * len(types))
|
||
|
|
|
||
|
|
# All filters.
|
||
|
|
idx = 0
|
||
|
|
for platform in platforms:
|
||
|
|
for type in types:
|
||
|
|
files = builder.get_files(platform=platform, type=type, version=version)
|
||
|
|
self.assertEqual(len(files), 1)
|
||
|
|
self.assertEqual(expected[idx], files[0])
|
||
|
|
idx = idx + 1
|
||
|
|
|
||
|
|
# Test Chromium version.
|
||
|
|
def test_chromium_version(self):
|
||
|
|
builder = cef_json_builder()
|
||
|
|
|
||
|
|
self.assertTrue(builder.is_valid_chromium_version('master'))
|
||
|
|
self.assertTrue(builder.is_valid_chromium_version('49.0.2704.0'))
|
||
|
|
self.assertTrue(builder.is_valid_chromium_version('49.0.2704.50'))
|
||
|
|
self.assertTrue(builder.is_valid_chromium_version('49.0.2704.100'))
|
||
|
|
|
||
|
|
self.assertFalse(builder.is_valid_chromium_version(None))
|
||
|
|
self.assertFalse(builder.is_valid_chromium_version('09.0.2704.50'))
|
||
|
|
self.assertFalse(builder.is_valid_chromium_version('00.0.0000.00'))
|
||
|
|
self.assertFalse(builder.is_valid_chromium_version('foobar'))
|
||
|
|
|
||
|
|
# The Git hashes must exist but the rest of the CEF version can be fake.
|
||
|
|
versions = (
|
||
|
|
('3.2704.1414.g185cd6c', '51.0.2704.47'),
|
||
|
|
('3.2623.9999.gb90a3be', '49.0.2623.110'),
|
||
|
|
('3.2623.9999.g2a6491b', '49.0.2623.87'),
|
||
|
|
('3.9999.9999.gab2636b', 'master'),
|
||
|
|
)
|
||
|
|
|
||
|
|
# Test with no query.
|
||
|
|
for (cef, chromium) in versions:
|
||
|
|
self.assertFalse(builder.has_chromium_version(cef))
|
||
|
|
# Valid version specified, so no query sent.
|
||
|
|
self.assertEqual(chromium, builder.set_chromium_version(cef, chromium))
|
||
|
|
self.assertEqual(chromium, builder.get_chromium_version(cef))
|
||
|
|
self.assertTrue(builder.has_chromium_version(cef))
|
||
|
|
# Ignore attempts to set invalid version after setting valid version.
|
||
|
|
self.assertEqual(chromium, builder.set_chromium_version(cef, None))
|
||
|
|
|
||
|
|
self.assertEqual(0, builder.get_query_count())
|
||
|
|
builder.clear()
|
||
|
|
|
||
|
|
# Test with query.
|
||
|
|
for (cef, chromium) in versions:
|
||
|
|
self.assertFalse(builder.has_chromium_version(cef))
|
||
|
|
# No version known, so query sent.
|
||
|
|
self.assertEqual(chromium, builder.get_chromium_version(cef))
|
||
|
|
self.assertTrue(builder.has_chromium_version(cef))
|
||
|
|
|
||
|
|
self.assertEqual(len(versions), builder.get_query_count())
|
||
|
|
|
||
|
|
|
||
|
|
# Program entry point.
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|