536
536
help="Remove client")
537
537
parser.add_argument("-c", "--checker",
538
538
help="Set checker command for client")
539
parser.add_argument("-t", "--timeout", type=string_to_delta,
539
parser.add_argument("-t", "--timeout",
540
540
help="Set timeout for client")
541
parser.add_argument("--extended-timeout", type=string_to_delta,
541
parser.add_argument("--extended-timeout",
542
542
help="Set extended timeout for client")
543
parser.add_argument("-i", "--interval", type=string_to_delta,
543
parser.add_argument("-i", "--interval",
544
544
help="Set checker interval for client")
545
545
approve_deny_default = parser.add_mutually_exclusive_group()
546
546
approve_deny_default.add_argument(
551
551
"--deny-by-default", action="store_false",
552
552
dest="approved_by_default",
553
553
help="Set client to be denied by default")
554
parser.add_argument("--approval-delay", type=string_to_delta,
554
parser.add_argument("--approval-delay",
555
555
help="Set delay before client approve/deny")
556
parser.add_argument("--approval-duration", type=string_to_delta,
556
parser.add_argument("--approval-duration",
557
557
help="Set duration of one client approval")
558
558
parser.add_argument("-H", "--host", help="Set host for client")
559
559
parser.add_argument("-s", "--secret",
1100
1100
class TestSetTimeoutCmd(TestValueArgumentPropertyCmd):
1101
1101
command = SetTimeoutCmd
1102
1102
property = "Timeout"
1103
values_to_set = [datetime.timedelta(),
1104
datetime.timedelta(minutes=5),
1105
datetime.timedelta(seconds=1),
1106
datetime.timedelta(weeks=1),
1107
datetime.timedelta(weeks=52)]
1108
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]
1110
1106
class TestSetExtendedTimeoutCmd(TestValueArgumentPropertyCmd):
1111
1107
command = SetExtendedTimeoutCmd
1112
1108
property = "ExtendedTimeout"
1113
values_to_set = [datetime.timedelta(),
1114
datetime.timedelta(minutes=5),
1115
datetime.timedelta(seconds=1),
1116
datetime.timedelta(weeks=1),
1117
datetime.timedelta(weeks=52)]
1118
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]
1120
1112
class TestSetIntervalCmd(TestValueArgumentPropertyCmd):
1121
1113
command = SetIntervalCmd
1122
1114
property = "Interval"
1123
values_to_set = [datetime.timedelta(),
1124
datetime.timedelta(minutes=5),
1125
datetime.timedelta(seconds=1),
1126
datetime.timedelta(weeks=1),
1127
datetime.timedelta(weeks=52)]
1128
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]
1130
1118
class TestSetApprovalDelayCmd(TestValueArgumentPropertyCmd):
1131
1119
command = SetApprovalDelayCmd
1132
1120
property = "ApprovalDelay"
1133
values_to_set = [datetime.timedelta(),
1134
datetime.timedelta(minutes=5),
1135
datetime.timedelta(seconds=1),
1136
datetime.timedelta(weeks=1),
1137
datetime.timedelta(weeks=52)]
1138
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]
1140
1124
class TestSetApprovalDurationCmd(TestValueArgumentPropertyCmd):
1141
1125
command = SetApprovalDurationCmd
1142
1126
property = "ApprovalDuration"
1143
values_to_set = [datetime.timedelta(),
1144
datetime.timedelta(minutes=5),
1145
datetime.timedelta(seconds=1),
1146
datetime.timedelta(weeks=1),
1147
datetime.timedelta(weeks=52)]
1148
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]
1150
1130
class Test_command_from_options(unittest.TestCase):
1151
1131
def setUp(self):
1169
1149
self.assert_command_from_args(["--verbose"], PrintTableCmd,
1172
def test_print_table_verbose_short(self):
1173
self.assert_command_from_args(["-v"], PrintTableCmd,
1176
1152
def test_enable(self):
1177
1153
self.assert_command_from_args(["--enable", "foo"], EnableCmd)
1179
def test_enable_short(self):
1180
self.assert_command_from_args(["-e", "foo"], EnableCmd)
1182
1155
def test_disable(self):
1183
1156
self.assert_command_from_args(["--disable", "foo"],
1186
def test_disable_short(self):
1187
self.assert_command_from_args(["-d", "foo"], DisableCmd)
1189
1159
def test_bump_timeout(self):
1190
1160
self.assert_command_from_args(["--bump-timeout", "foo"],
1191
1161
BumpTimeoutCmd)
1193
def test_bump_timeout_short(self):
1194
self.assert_command_from_args(["-b", "foo"], BumpTimeoutCmd)
1196
1163
def test_start_checker(self):
1197
1164
self.assert_command_from_args(["--start-checker", "foo"],
1198
1165
StartCheckerCmd)
1216
1180
self.assert_command_from_args(["--checker", "", "foo"],
1217
1181
SetCheckerCmd, value_to_set="")
1219
def test_checker_short(self):
1220
self.assert_command_from_args(["-c", ":", "foo"],
1221
SetCheckerCmd, value_to_set=":")
1223
1183
def test_timeout(self):
1224
1184
self.assert_command_from_args(["--timeout", "PT5M", "foo"],
1226
1186
value_to_set=300000)
1228
def test_timeout_short(self):
1229
self.assert_command_from_args(["-t", "PT5M", "foo"],
1231
value_to_set=300000)
1233
1188
def test_extended_timeout(self):
1234
1189
self.assert_command_from_args(["--extended-timeout", "PT15M",
1288
1233
"foo"], SetSecretCmd,
1289
1234
value_to_set=value)
1291
def test_secret_devnull_short(self):
1292
self.assert_command_from_args(["-s", os.path.devnull, "foo"],
1293
SetSecretCmd, value_to_set=b"")
1295
def test_secret_tempfile_short(self):
1296
with tempfile.NamedTemporaryFile(mode="r+b") as f:
1297
value = b"secret\0xyzzy\nbar"
1300
self.assert_command_from_args(["-s", f.name, "foo"],
1304
1236
def test_approve(self):
1305
1237
self.assert_command_from_args(["--approve", "foo"],
1308
def test_approve_short(self):
1309
self.assert_command_from_args(["-A", "foo"], ApproveCmd)
1311
1240
def test_deny(self):
1312
1241
self.assert_command_from_args(["--deny", "foo"], DenyCmd)
1314
def test_deny_short(self):
1315
self.assert_command_from_args(["-D", "foo"], DenyCmd)
1317
1243
def test_dump_json(self):
1318
1244
self.assert_command_from_args(["--dump-json"], DumpJSONCmd)