blob: 46074d2241a52c195521bda71dc0121235ff9be1 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
diff --git b/test/util.py a/test/util.py
new file mode 100644
index 0000000..0551d06
--- /dev/null
+++ a/test/util.py
@@ -0,0 +1,18 @@
+from click.testing import CliRunner
+from ykman.cli.__main__ import cli
+import os
+
+
+PKG_DIR = os.path.dirname(os.path.abspath(__file__))
+
+
+def open_file(*relative_path):
+ return open(os.path.join(PKG_DIR, 'files', *relative_path), 'rb')
+
+
+def ykman_cli(*argv, **kwargs):
+ runner = CliRunner()
+ result = runner.invoke(cli, list(argv), obj={}, **kwargs)
+ if result.exit_code != 0:
+ raise result.exception
+ return result.output
|