def test_unfindable_args_config_file(self): """ Test unfindable config file """ args = argparse.Namespace() args.config_file_name = "\b" args.template_file_name = __file__ args.start_date = args.end_date = args.num_nodes = None with self.assertRaises(FileNotFoundError): yaml_gen.handle_args(args)
def test_handle_missing_end_date_arg(self): """ Test the argument handler start range only """ args = argparse.Namespace() args.start_date = "9999-12-31" args.template_file_name = __file__ args.config_file_name = args.end_date = args.num_nodes = None args.random = False with self.assertRaises(yaml_gen.DateRangeArgsError): yaml_gen.handle_args(args)
def test_handle_args_date_range(self): """ Test the argument handler start and end range """ args = argparse.Namespace() args.start_date = "9999-12-31" args.end_date = "9999-12-31" args.template_file_name = __file__ args.config_file_name = args.num_nodes = None args.provider = "ocp" args.random = False yaml_gen.handle_args(args) self.assertTrue(isinstance(args.start_date, date)) self.assertTrue(isinstance(args.end_date, date))
def test_zero_num_nodes_arg(self): """ Test zero num_nodes arg """ args = argparse.Namespace() args.template_file_name = __file__ args.start_date = args.end_date = args.config_file_name = None args.provider = "ocp" args.num_nodes = 0 args = yaml_gen.handle_args(args) self.assertTrue(args.num_nodes is None)
def test_positive_num_nodes_arg(self): """ Test positive num_nodes arg """ args = argparse.Namespace() args.template_file_name = __file__ args.num_nodes = 10 args.start_date = args.end_date = args.config_file_name = None args.provider = "ocp" args = yaml_gen.handle_args(args) self.assertEqual(args.num_nodes, 10)