1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
# Copyright (C) 2016 the V8 project authors. All rights reserved.
# This code is governed by the BSD license found in the LICENSE file.
import re, os
from case import Case
from template import Template
caseFilenamePattern = re.compile(r'^[^\.].*\.case$')
templateFilenamePattern = re.compile(r'^[^\.].*\.template$')
class Expander:
def __init__(self, case_dir):
self.templates = dict()
self.case_dir = case_dir
def _load_templates(self, template_class):
directory = os.path.join(self.case_dir, template_class)
file_names = map(
lambda x: os.path.join(directory, x),
filter(self.is_template_file, os.listdir(directory))
)
self.templates[template_class] = [Template(x) for x in file_names]
def _get_templates(self, template_class):
if not template_class in self.templates:
self._load_templates(template_class)
return self.templates[template_class]
def is_template_file(self, filename):
return re.match(templateFilenamePattern, filename)
def list_cases(self):
for name in os.listdir(self.case_dir):
full = os.path.join(self.case_dir, name)
if os.path.isfile(full) and caseFilenamePattern.match(name):
yield full
def expand(self, encoding, case_file = None):
if case_file:
case_files = [case_file]
else:
case_files = self.list_cases()
for case_file in case_files:
for test in self.expand_case(case_file, encoding):
yield test
def expand_case(self, file_name, encoding):
case = Case(file_name)
template_class = case.attribs['meta']['template']
templates = self.templates.get(template_class)
for template in self._get_templates(template_class):
yield template.expand(file_name, os.path.basename(file_name[:-5]), case.attribs, encoding)
|
