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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import sys
import argparse
import json
import logging
def main():
logging.basicConfig(format="assert_intercepted: %(message)s", level=logging.INFO)
parser = argparse.ArgumentParser()
parser.add_argument(dest='input', type=argparse.FileType(mode='r'))
subparsers = parser.add_subparsers(dest='command')
count_parser = subparsers.add_parser('count').add_mutually_exclusive_group()
count_parser.add_argument('-eq', dest='eq', type=int)
count_parser.add_argument('-gt', dest='gt', type=int)
count_parser.add_argument('-ge', dest='ge', type=int)
count_parser.add_argument('-lt', dest='lt', type=int)
count_parser.add_argument('-le', dest='le', type=int)
entry_parser = subparsers.add_parser('contains')
entry_parser.add_argument('-program')
entry_parser.add_argument('-working_dir')
entry_parser.add_argument('-arguments', nargs=argparse.REMAINDER)
args = parser.parse_args()
logging.debug(args)
commands = list(load_commands(args.input))
if args.command == 'count':
count = len(commands)
logging.debug("%s has %d command entries", args.input.name, count)
if args.eq is not None and args.eq != count:
logging.info('failed: expected %d, but got %d', args.eq, count)
return 1
elif args.gt is not None and args.gt >= count:
logging.info('failed: expected greater than %d, but got %d', args.gt, count)
return 1
elif args.ge is not None and args.ge > count:
logging.info('failed: expected greater or equal %d, but got %d', args.ge, count)
return 1
elif args.lt is not None and args.lt <= count:
logging.info('failed: expected less than %d, but got %d', args.lt, count)
return 1
elif args.le is not None and args.le < count:
logging.info('failed: expected less or equal %d, but got %d', args.le, count)
return 1
elif args.command == 'contains':
logging.debug("%s has command entries: %s", args.input.name, commands)
check = filter_from(args)
count = len([cmd for cmd in commands if check(cmd)])
if count == 0:
request = to_string(args)
logging.info('failed: expected at least one entry, but found none: %s', request)
return 1
return 0
def load_commands(input):
for line in input:
event = json.loads(line)
if 'started' in event:
yield {
'program': event['started']['execution']['executable'],
'working_dir': event['started']['execution']['working_dir'],
'arguments': event['started']['execution']['arguments']
}
def filter_from(args):
def test(command):
p1 = True if args.program is None else command['program'] == args.program
p2 = True if args.working_dir is None else command['working_dir'] == args.working_dir
p3 = True if args.arguments is None else command['arguments'] == args.arguments
return p1 and p2 and p3
return test
def to_string(args):
p1 = '' if args.program is None else '-program {}'.format(args.program)
p2 = '' if args.working_dir is None else '-working_dir {}'.format(args.working_dir)
p3 = '' if args.arguments is None else ' '.join(['-arguments'] + args.arguments)
return ' '.join([p1, p2, p3])
if __name__ == "__main__":
sys.exit(main())
|
