Add end-to-end simulation

1 files changed, 8 insertions, 8 deletions

diff --git a/controller/fw/tools/presig_gen.py b/controller/fw/tools/presig_gen.py
index ff9dffe..3f85522 100644
--- a/controller/fw/tools/presig_gen.py
+++ b/controller/fw/tools/presig_gen.py
@@ -123,17 +123,17 @@ if __name__ == '__main__':
     auth_parser.add_argument('height', type=int, help='Authentication string height, counting from 0 (root key)')
     auth_parser.set_defaults(func=auth_cmd)
     auth_parser.add_argument('-a', '--all', action='store_const', const='all', help='Vendor name for vendor domain')
-    auth_parser.add_argument('-v', '--vendor', type=str, nargs='?', const=test_vendor, help='Vendor name for vendor domain')
-    auth_parser.add_argument('-s', '--series', type=str, nargs='?', const=test_series, help='Series identifier for series domain')
-    auth_parser.add_argument('-r', '--region', type=str, nargs='?', const=test_region, help='Region name for region domain')
-    auth_parser.add_argument('-c', '--country', type=str, nargs='?', const=test_country, help='Country name for country domain')
+    auth_parser.add_argument('-v', '--vendor', type=str, nargs='?', const=TEST_VENDOR, help='Vendor name for vendor domain')
+    auth_parser.add_argument('-s', '--series', type=str, nargs='?', const=TEST_SERIES, help='Series identifier for series domain')
+    auth_parser.add_argument('-r', '--region', type=str, nargs='?', const=TEST_REGION, help='Region name for region domain')
+    auth_parser.add_argument('-c', '--country', type=str, nargs='?', const=TEST_COUNTRY, help='Country name for country domain')
 
     prekey_parser = subparsers.add_parser('prekey', help='Generate prekey data .C source code file')
     prekey_parser.add_argument('-m', '--max-height', type=int, default=8, help='Height of generated prekey')
-    prekey_parser.add_argument('-v', '--vendor', type=str, default=test_vendor, help='Vendor name for vendor domain')
-    prekey_parser.add_argument('-s', '--series', type=str, default=test_series, help='Series identifier for series domain')
-    prekey_parser.add_argument('-r', '--region', type=str, default=test_region, help='Region name for region domain')
-    prekey_parser.add_argument('-c', '--country', type=str, default=test_country, help='Country name for country domain')
+    prekey_parser.add_argument('-v', '--vendor', type=str, default=TEST_VENDOR, help='Vendor name for vendor domain')
+    prekey_parser.add_argument('-s', '--series', type=str, default=TEST_SERIES, help='Series identifier for series domain')
+    prekey_parser.add_argument('-r', '--region', type=str, default=TEST_REGION, help='Region name for region domain')
+    prekey_parser.add_argument('-c', '--country', type=str, default=TEST_COUNTRY, help='Country name for country domain')
     prekey_parser.set_defaults(func=prekey_cmd, all='all')
 
     args = parser.parse_args()