505
505
MillisecondsValueArgumentMixIn):
506
506
property = "ApprovalDuration"
508
def has_actions(options):
509
return any((options.enable,
511
options.bump_timeout,
512
options.start_checker,
513
options.stop_checker,
516
options.checker is not None,
517
options.timeout is not None,
518
options.extended_timeout is not None,
519
options.interval is not None,
520
options.approved_by_default is not None,
521
options.approval_delay is not None,
522
options.approval_duration is not None,
523
options.host is not None,
524
options.secret is not None,
508
528
def add_command_line_options(parser):
509
529
parser.add_argument("--version", action="version",
510
530
version="%(prog)s {}".format(version),
536
556
help="Remove client")
537
557
parser.add_argument("-c", "--checker",
538
558
help="Set checker command for client")
539
parser.add_argument("-t", "--timeout", type=string_to_delta,
559
parser.add_argument("-t", "--timeout",
540
560
help="Set timeout for client")
541
parser.add_argument("--extended-timeout", type=string_to_delta,
561
parser.add_argument("--extended-timeout",
542
562
help="Set extended timeout for client")
543
parser.add_argument("-i", "--interval", type=string_to_delta,
563
parser.add_argument("-i", "--interval",
544
564
help="Set checker interval for client")
545
565
approve_deny_default = parser.add_mutually_exclusive_group()
546
566
approve_deny_default.add_argument(
551
571
"--deny-by-default", action="store_false",
552
572
dest="approved_by_default",
553
573
help="Set client to be denied by default")
554
parser.add_argument("--approval-delay", type=string_to_delta,
574
parser.add_argument("--approval-delay",
555
575
help="Set delay before client approve/deny")
556
parser.add_argument("--approval-duration", type=string_to_delta,
576
parser.add_argument("--approval-duration",
557
577
help="Set duration of one client approval")
558
578
parser.add_argument("-H", "--host", help="Set host for client")
559
579
parser.add_argument("-s", "--secret",
647
def check_option_syntax(parser, options):
649
def has_actions(options):
650
return any((options.enable,
652
options.bump_timeout,
653
options.start_checker,
654
options.stop_checker,
657
options.checker is not None,
658
options.timeout is not None,
659
options.extended_timeout is not None,
660
options.interval is not None,
661
options.approved_by_default is not None,
662
options.approval_delay is not None,
663
options.approval_duration is not None,
664
options.host is not None,
665
options.secret is not None,
668
parser = argparse.ArgumentParser()
670
add_command_line_options(parser)
672
options = parser.parse_args()
669
674
if has_actions(options) and not (options.client or options.all):
670
675
parser.error("Options require clients names or --all.")
1105
1100
class TestSetTimeoutCmd(TestValueArgumentPropertyCmd):
1106
1101
command = SetTimeoutCmd
1107
1102
property = "Timeout"
1108
values_to_set = [datetime.timedelta(),
1109
datetime.timedelta(minutes=5),
1110
datetime.timedelta(seconds=1),
1111
datetime.timedelta(weeks=1),
1112
datetime.timedelta(weeks=52)]
1113
values_to_get = [0, 300000, 1000, 604800000, 31449600000]
1103
values_to_set = ["P0D", "PT5M", "PT1S", "PT120S", "P1Y"]
1104
values_to_get = [0, 300000, 1000, 120000, 31449600000]
1115
1106
class TestSetExtendedTimeoutCmd(TestValueArgumentPropertyCmd):
1116
1107
command = SetExtendedTimeoutCmd
1117
1108
property = "ExtendedTimeout"
1118
values_to_set = [datetime.timedelta(),
1119
datetime.timedelta(minutes=5),
1120
datetime.timedelta(seconds=1),
1121
datetime.timedelta(weeks=1),
1122
datetime.timedelta(weeks=52)]
1123
values_to_get = [0, 300000, 1000, 604800000, 31449600000]
1109
values_to_set = ["P0D", "PT5M", "PT1S", "PT120S", "P1Y"]
1110
values_to_get = [0, 300000, 1000, 120000, 31449600000]
1125
1112
class TestSetIntervalCmd(TestValueArgumentPropertyCmd):
1126
1113
command = SetIntervalCmd
1127
1114
property = "Interval"
1128
values_to_set = [datetime.timedelta(),
1129
datetime.timedelta(minutes=5),
1130
datetime.timedelta(seconds=1),
1131
datetime.timedelta(weeks=1),
1132
datetime.timedelta(weeks=52)]
1133
values_to_get = [0, 300000, 1000, 604800000, 31449600000]
1115
values_to_set = ["P0D", "PT5M", "PT1S", "PT120S", "P1Y"]
1116
values_to_get = [0, 300000, 1000, 120000, 31449600000]
1135
1118
class TestSetApprovalDelayCmd(TestValueArgumentPropertyCmd):
1136
1119
command = SetApprovalDelayCmd
1137
1120
property = "ApprovalDelay"
1138
values_to_set = [datetime.timedelta(),
1139
datetime.timedelta(minutes=5),
1140
datetime.timedelta(seconds=1),
1141
datetime.timedelta(weeks=1),
1142
datetime.timedelta(weeks=52)]
1143
values_to_get = [0, 300000, 1000, 604800000, 31449600000]
1121
values_to_set = ["P0D", "PT5M", "PT1S", "PT120S", "P1Y"]
1122
values_to_get = [0, 300000, 1000, 120000, 31449600000]
1145
1124
class TestSetApprovalDurationCmd(TestValueArgumentPropertyCmd):
1146
1125
command = SetApprovalDurationCmd
1147
1126
property = "ApprovalDuration"
1148
values_to_set = [datetime.timedelta(),
1149
datetime.timedelta(minutes=5),
1150
datetime.timedelta(seconds=1),
1151
datetime.timedelta(weeks=1),
1152
datetime.timedelta(weeks=52)]
1153
values_to_get = [0, 300000, 1000, 604800000, 31449600000]
1127
values_to_set = ["P0D", "PT5M", "PT1S", "PT120S", "P1Y"]
1128
values_to_get = [0, 300000, 1000, 120000, 31449600000]
1155
1130
class Test_command_from_options(unittest.TestCase):
1156
1131
def setUp(self):
1175
1149
self.assert_command_from_args(["--verbose"], PrintTableCmd,
1178
def test_print_table_verbose_short(self):
1179
self.assert_command_from_args(["-v"], PrintTableCmd,
1182
1152
def test_enable(self):
1183
1153
self.assert_command_from_args(["--enable", "foo"], EnableCmd)
1185
def test_enable_short(self):
1186
self.assert_command_from_args(["-e", "foo"], EnableCmd)
1188
1155
def test_disable(self):
1189
1156
self.assert_command_from_args(["--disable", "foo"],
1192
def test_disable_short(self):
1193
self.assert_command_from_args(["-d", "foo"], DisableCmd)
1195
1159
def test_bump_timeout(self):
1196
1160
self.assert_command_from_args(["--bump-timeout", "foo"],
1197
1161
BumpTimeoutCmd)
1199
def test_bump_timeout_short(self):
1200
self.assert_command_from_args(["-b", "foo"], BumpTimeoutCmd)
1202
1163
def test_start_checker(self):
1203
1164
self.assert_command_from_args(["--start-checker", "foo"],
1204
1165
StartCheckerCmd)
1211
1172
self.assert_command_from_args(["--remove", "foo"],
1214
def test_remove_short(self):
1215
self.assert_command_from_args(["-r", "foo"], RemoveCmd)
1217
1175
def test_checker(self):
1218
1176
self.assert_command_from_args(["--checker", ":", "foo"],
1219
1177
SetCheckerCmd, value_to_set=":")
1221
def test_checker_empty(self):
1222
self.assert_command_from_args(["--checker", "", "foo"],
1223
SetCheckerCmd, value_to_set="")
1225
def test_checker_short(self):
1226
self.assert_command_from_args(["-c", ":", "foo"],
1227
SetCheckerCmd, value_to_set=":")
1229
1179
def test_timeout(self):
1230
1180
self.assert_command_from_args(["--timeout", "PT5M", "foo"],
1232
1182
value_to_set=300000)
1234
def test_timeout_short(self):
1235
self.assert_command_from_args(["-t", "PT5M", "foo"],
1237
value_to_set=300000)
1239
1184
def test_extended_timeout(self):
1240
1185
self.assert_command_from_args(["--extended-timeout", "PT15M",
1294
1229
"foo"], SetSecretCmd,
1295
1230
value_to_set=value)
1297
def test_secret_devnull_short(self):
1298
self.assert_command_from_args(["-s", os.path.devnull, "foo"],
1299
SetSecretCmd, value_to_set=b"")
1301
def test_secret_tempfile_short(self):
1302
with tempfile.NamedTemporaryFile(mode="r+b") as f:
1303
value = b"secret\0xyzzy\nbar"
1306
self.assert_command_from_args(["-s", f.name, "foo"],
1310
1232
def test_approve(self):
1311
1233
self.assert_command_from_args(["--approve", "foo"],
1314
def test_approve_short(self):
1315
self.assert_command_from_args(["-A", "foo"], ApproveCmd)
1317
1236
def test_deny(self):
1318
1237
self.assert_command_from_args(["--deny", "foo"], DenyCmd)
1320
def test_deny_short(self):
1321
self.assert_command_from_args(["-D", "foo"], DenyCmd)
1323
1239
def test_dump_json(self):
1324
1240
self.assert_command_from_args(["--dump-json"], DumpJSONCmd)