def test_validate_arguments_valid_keyboard_input_args(s):
args = mock.Mock()
args.inputtype = 'kb'
args.key = 123
args.input = args.output = args.encrypt = args.decrypt = None
result = feistel.validate_arguments(args)
s.assertEqual(result, True)
def test_validate_arguments_no_key_for_file_encryption(s, print_func):
args = mock.Mock()
args.inputtype = 'f'
args.key = None
args.input = "in.txt"
args.output = "out"
args.encrypt = True
args.decrypt = None
result = feistel.validate_arguments(args)
s.assertEqual(result, False)
def test_validate_arguments_valid_file_decryption_args(s):
args = mock.Mock()
args.inputtype = 'f'
args.key = 123
args.input = "out"
args.output = "final.txt"
args.encrypt = False
args.decrypt = True
result = feistel.validate_arguments(args)
s.assertEqual(result, True)
def test_validate_arguments_empty_input_file(s, print_func):
args = mock.Mock()
args.inputtype = 'f'
args.key = 123
args.input = "in.txt"
args.output = "out"
args.encrypt = True
args.decrypt = None
file_stats = type('', (object, ), {"st_size": 0})()
patcher = patch('os.stat', lambda x: file_stats)
patcher.start()
result = feistel.validate_arguments(args)
patcher.stop()
s.assertEqual(result, False)
def test_validate_arguments_no_key_for_keyboard_input(s, print_func):
args = mock.Mock()
args.inputtype = 'kb'
args.key = args.input = args.output = args.encrypt = args.decrypt = None
result = feistel.validate_arguments(args)
s.assertEqual(result, False)