Add case_name as constructor arg
[functest.git] / functest / opnfv_tests / vnf / aaa / aaa.py
1 #!/usr/bin/env python
2
3 # Copyright (c) 2016 Orange and others.
4 #
5 # All rights reserved. This program and the accompanying materials
6 # are made available under the terms of the Apache License, Version 2.0
7 # which accompanies this distribution, and is available at
8 # http://www.apache.org/licenses/LICENSE-2.0
9
10 import sys
11
12 import argparse
13
14 import functest.core.testcase as testcase
15 import functest.core.vnf_base as vnf_base
16 import functest.utils.functest_logger as ft_logger
17
18
19 class AaaVnf(vnf_base.VnfOnBoardingBase):
20
21     logger = ft_logger.Logger("VNF AAA").getLogger()
22
23     def __init__(self):
24         super(AaaVnf, self).__init__(case_name="aaa")
25
26     def deploy_orchestrator(self):
27         self.logger.info("No VNFM needed to deploy a free radius here")
28         return None
29
30 # TODO see how to use build in exception form releng module
31     def deploy_vnf(self):
32         self.logger.info("Freeradius VNF deployment")
33         # TODO apt-get update + config tuning
34         deploy_vnf = {}
35         deploy_vnf['status'] = "PASS"
36         deploy_vnf['result'] = {}
37         return deploy_vnf
38
39     def test_vnf(self):
40         self.logger.info("Run test towards freeradius")
41         # TODO:  once the freeradius is deployed..make some tests
42         test_vnf = {}
43         test_vnf['status'] = "PASS"
44         test_vnf['result'] = {}
45         return test_vnf
46
47     def main(self, **kwargs):
48         self.logger.info("AAA VNF onboarding")
49         self.execute()
50         if self.criteria is "PASS":
51             return self.EX_OK
52         else:
53             return self.EX_RUN_ERROR
54
55     def run(self):
56         kwargs = {}
57         return self.main(**kwargs)
58
59
60 if __name__ == '__main__':
61     parser = argparse.ArgumentParser()
62     args = vars(parser.parse_args())
63     aaa_vnf = AaaVnf()
64     try:
65         result = aaa_vnf.main(**args)
66         if result != testcase.TestCase.EX_OK:
67             sys.exit(result)
68         if args['pushtodb']:
69             sys.exit(aaa_vnf.push_to_db())
70     except Exception:
71         sys.exit(testcase.TestCase.EX_RUN_ERROR)