def test_verify_args_general(self):
parser = yt_script.setup_arg_parser()
try:
yt_script.verify_arguments(
parser, [self.TESTING_CHANNEL_ID, self.API_KEY,
'-d', '1', '-f', self.TESTING_FILENAME + '.' + self.TESTING_EXT])
yt_script.verify_arguments(
parser, [self.TESTING_CHANNEL_ID, self.API_KEY,
'-d', '99999', '-f', '.' + self.TESTING_FILENAME + '.' + self.TESTING_EXT])
yt_script.verify_arguments(
parser, [self.TESTING_CHANNEL_ID, self.API_KEY,
'-d', '99999', '-f', self.TESTING_FILENAME + '.' + self.TESTING_EXT + '.'])
yt_script.verify_arguments(
parser, [self.TESTING_CHANNEL_ID, self.API_KEY,
'-d', '99999', '-f', self.TESTING_FILENAME])
except AssertionError as e:
self.fail(str(e.message))
self.assertRaises(AttributeError, yt_script.verify_arguments,
parser, [self.TESTING_CHANNEL_ID, self.API_KEY,
'-d', '0', '-f', self.TESTING_FILENAME])
self.assertRaises(AttributeError, yt_script.verify_arguments,
parser, [self.TESTING_CHANNEL_ID, self.API_KEY,
'-d', '-1', '-f', self.TESTING_FILENAME])
for char in self.BAD_CHARS:
self.assertRaises(Exception, yt_script.verify_arguments,
[self.TESTING_CHANNEL_ID, self.API_KEY,
'-f', self.TESTING_FILENAME + char])
def test_args_output(self):
testing_outputs = ['text', 'graphml', 'gml', 'gexf', 'yaml']
parser = yt_script.setup_arg_parser()
for option in testing_outputs:
response = parser.parse_args([self.TESTING_CHANNEL_ARG,
self.TESTING_API_KEY, '-o', option])
self.assertEqual(response.output, option)
def test_args_verbose(self):
testing_verbosity = [1, 2, 3]
parser = yt_script.setup_arg_parser()
for option in testing_verbosity:
response = parser.parse_args([self.TESTING_CHANNEL_ARG,
self.TESTING_API_KEY, '-v', str(option)])
self.assertEqual(int(response.verbose), option)
def test_args_degrees(self):
parser = yt_script.setup_arg_parser()
testing_degrees = ['1', '2', '3', '4', '5', '6', '7', '8', '9']
for test_degree in testing_degrees:
response = parser.parse_args([self.TESTING_CHANNEL_ARG,
self.TESTING_API_KEY, '-d', test_degree])
self.assertTrue(response.degree, int(test_degree))
def test_args_filename(self):
filename = 'mock_filename'
parser = yt_script.setup_arg_parser()
response = parser.parse_args([self.TESTING_CHANNEL_ARG,
self.TESTING_API_KEY, '-f', filename])
self.assertEqual(response.filename, filename)
def test_args_output(self):
testing_outputs = ['text', 'graphml', 'gml', 'gexf', 'yaml']
parser = yt_script.setup_arg_parser()
for option in testing_outputs:
response = parser.parse_args(
[self.TESTING_CHANNEL_ARG, self.TESTING_API_KEY, '-o', option])
self.assertEqual(response.output, option)
def test_args_filename(self):
filename = 'mock_filename'
parser = yt_script.setup_arg_parser()
response = parser.parse_args(
[self.TESTING_CHANNEL_ARG, self.TESTING_API_KEY, '-f', filename])
self.assertEqual(response.filename, filename)
def test_args_verbose(self):
testing_verbosity = [1, 2, 3]
parser = yt_script.setup_arg_parser()
for option in testing_verbosity:
response = parser.parse_args([
self.TESTING_CHANNEL_ARG, self.TESTING_API_KEY, '-v',
str(option)
])
self.assertEqual(int(response.verbose), option)
def test_args_degrees(self):
parser = yt_script.setup_arg_parser()
testing_degrees = ['1', '2', '3', '4', '5', '6', '7', '8', '9']
for test_degree in testing_degrees:
response = parser.parse_args([
self.TESTING_CHANNEL_ARG, self.TESTING_API_KEY, '-d',
test_degree
])
self.assertTrue(response.degree, int(test_degree))
def test_args_defaults(self):
expected_defaults = "Namespace(api_key=" + repr(self.TESTING_API_KEY) + \
", degree=1, filename=" + repr(yt_script.DEFAULT_OUTPUT_FILENAME) \
+ ", id=" + repr(self.TESTING_CHANNEL_ARG) + \
", output=None, show_graph=False, verbose=0)"
parser = yt_script.setup_arg_parser()
response = parser.parse_args([self.TESTING_CHANNEL_ARG, self.TESTING_API_KEY])
self.assertEqual(str(response), expected_defaults)
def test_args_defaults(self):
expected_defaults = "Namespace(api_key=" + repr(self.TESTING_API_KEY) + \
", degree=1, filename=" + repr(yt_script.DEFAULT_OUTPUT_FILENAME) \
+ ", id=" + repr(self.TESTING_CHANNEL_ARG) + \
", output=None, show_graph=False, verbose=0)"
parser = yt_script.setup_arg_parser()
response = parser.parse_args(
[self.TESTING_CHANNEL_ARG, self.TESTING_API_KEY])
self.assertEqual(str(response), expected_defaults)
def test_verify_args_general(self):
parser = yt_script.setup_arg_parser()
try:
yt_script.verify_arguments(parser, [
self.TESTING_CHANNEL_ID, self.API_KEY, '-d', '1', '-f',
self.TESTING_FILENAME + '.' + self.TESTING_EXT
])
yt_script.verify_arguments(parser, [
self.TESTING_CHANNEL_ID, self.API_KEY, '-d', '99999', '-f',
'.' + self.TESTING_FILENAME + '.' + self.TESTING_EXT
])
yt_script.verify_arguments(parser, [
self.TESTING_CHANNEL_ID, self.API_KEY, '-d', '99999', '-f',
self.TESTING_FILENAME + '.' + self.TESTING_EXT + '.'
])
yt_script.verify_arguments(parser, [
self.TESTING_CHANNEL_ID, self.API_KEY, '-d', '99999', '-f',
self.TESTING_FILENAME
])
except AssertionError as e:
self.fail(str(e.message))
self.assertRaises(AttributeError, yt_script.verify_arguments, parser, [
self.TESTING_CHANNEL_ID, self.API_KEY, '-d', '0', '-f',
self.TESTING_FILENAME
])
self.assertRaises(AttributeError, yt_script.verify_arguments, parser, [
self.TESTING_CHANNEL_ID, self.API_KEY, '-d', '-1', '-f',
self.TESTING_FILENAME
])
for char in self.BAD_CHARS:
self.assertRaises(Exception, yt_script.verify_arguments, [
self.TESTING_CHANNEL_ID, self.API_KEY, '-f',
self.TESTING_FILENAME + char
])