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
|
from unittest import TestCase
from click.testing import CliRunner
from pyecsca.codegen.builder import build_impl, list_impl
from parameterized import parameterized
class BuilderTests(TestCase):
@parameterized.expand([
("basic", ["--platform", "HOST", "shortw", "projective", "add-1998-cmo", "dbl-1998-cmo", "z", "ltr(complete=True)", "."]),
("karatsuba", ["--platform", "HOST", "--mul", "KARATSUBA", "shortw", "projective", "add-1998-cmo", "dbl-1998-cmo", "z", "ltr(complete=True)", "."]),
("strip", ["--platform", "HOST", "--strip", "--no-remove", "shortw", "projective", "add-1998-cmo", "dbl-1998-cmo", "z", "ltr(complete=True)", "."]),
("montgom", ["--platform", "HOST", "--no-ecdsa", "montgom", "xz", "ladd-1987-m", "dbl-1987-m", "scale", "ldr()", "."]),
("jacobian", ["--platform", "HOST", "--no-ecdsa", "shortw", "jacobian", "add-2007-bl", "dbl-2007-bl", "rtl()", "."])
])
def test_cli_build(self, name, args):
runner = CliRunner()
with runner.isolated_filesystem():
result = runner.invoke(build_impl, args)
self.assertEqual(result.exit_code, 0)
def test_cli_build_fails(self):
runner = CliRunner()
with runner.isolated_filesystem():
# unknown model
result = runner.invoke(build_impl,
["--platform", "HOST", "missing", "projective", "add-1998-cmo",
"dbl-1998-cmo", "z", "ltr(complete=True)", "."])
self.assertEqual(result.exit_code, 2)
# unknown coordinates
result = runner.invoke(build_impl,
["--platform", "HOST", "shortw", "missing", "add-1998-cmo",
"dbl-1998-cmo", "z", "ltr(complete=True)", "."])
self.assertEqual(result.exit_code, 2)
# unknown formula
result = runner.invoke(build_impl,
["--platform", "HOST", "shortw", "projective", "missing",
"dbl-1998-cmo", "z", "ltr(complete=True)", "."])
self.assertEqual(result.exit_code, 2)
# bad formatted mult spec
result = runner.invoke(build_impl,
["--platform", "HOST", "shortw", "projective", "add-1998-cmo",
"dbl-1998-cmo", "z", "missing", "."])
self.assertEqual(result.exit_code, 2)
# unknown mult
result = runner.invoke(build_impl,
["--platform", "HOST", "shortw", "projective", "add-1998-cmo",
"dbl-1998-cmo", "z", "missing()", "."])
self.assertEqual(result.exit_code, 2)
# missing required formulas to mult
result = runner.invoke(build_impl,
["--platform", "HOST", "shortw", "projective", "add-1998-cmo",
"z", "ltr(complete=True)", "."])
self.assertEqual(result.exit_code, 2)
# duplicate formulas
result = runner.invoke(build_impl,
["--platform", "HOST", "shortw", "projective", "add-1998-cmo",
"add-1998-cmo", "z", "ltr(complete=True)", "."])
self.assertEqual(result.exit_code, 2)
def test_cli_list(self):
runner = CliRunner()
result = runner.invoke(list_impl, [])
self.assertEqual(result.exit_code, 0)
result = runner.invoke(list_impl, ["montgom"])
self.assertEqual(result.exit_code, 0)
result = runner.invoke(list_impl, ["montgom", "xz"])
self.assertEqual(result.exit_code, 0)
result = runner.invoke(list_impl, ["montgom", "xz", "ladd-1987-m"])
self.assertEqual(result.exit_code, 0)
|
