class Args(object):
def __init__(self, schema: str, command: str):
self.command = Command(command)
self.schema = Schema(schema)
def get_value(self, name):
return self.schema.get_value(name, self.command.get_value(name))
def test_normal(self):
a = Command("-l true -p 8080 -d /usr/logs")
self.assertEqual(a.get_value('l'), "true")
self.assertEqual(a.get_value('p'), "8080")
self.assertEqual(a.get_value('d'), "/usr/logs")
def test_negative_value(self):
a = Command("-l -p -9 -d /usr/logs")
self.assertEqual(a.get_value('l'), None)
self.assertEqual(a.get_value('p'), "-9")
def test_value_missing(self):
a = Command("-l -p 8080 -d /usr/logs")
self.assertEqual(a.get_value('l'), None)
