summaryrefslogtreecommitdiffstats
path: root/scripts/automation/trex_control_plane/unit_tests/control_plane_unit_test.py
blob: 37130ee4831d2fa84e7fb5a5935dddbf3bab18cf (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
#!/router/bin/python

__copyright__ = "Copyright 2014"



import os
import sys
import nose_outer_packages
import nose
from nose.plugins import Plugin
from rednose import RedNose
import termstyle
import control_plane_general_test

class TRexCPConfiguringPlugin(Plugin):
    def options(self, parser, env = os.environ):
        super(TRexCPConfiguringPlugin, self).options(parser, env)
        parser.add_option('-t', '--trex-server', action='store',
                            dest='trex_server', default='trex-dan',
                            help='Specify T-Rex server hostname. This server will be used to test control-plane functionality.')

    def configure(self, options, conf):
        if options.trex_server:
        	self.trex_server = options.trex_server

    def begin (self):
        # initialize CTRexCP global testing class, to be used by and accessible all tests
        print "assigned trex_server name"
        control_plane_general_test.CTRexCP.trex_server = self.trex_server
    
    def finalize(self, result):
        pass



if __name__ == "__main__":
    
    # setting defaults. By default we run all the test suite
    specific_tests    = False
    disableLogCapture = False
    long_test         = False
    report_dir        = "reports"

    nose_argv= sys.argv + ['-s', '-v', '--exe', '--rednose', '--detailed-errors']

    try:
        result = nose.run(argv = nose_argv, addplugins = [RedNose(), TRexCPConfiguringPlugin()])
        if (result == True):
            print termstyle.green("""
                     ..::''''::..
                   .;''        ``;.
                  ::    ::  ::    ::
                 ::     ::  ::     ::
                 ::     ::  ::     ::
                 :: .:' ::  :: `:. ::
                 ::  :          :  ::
                  :: `:.      .:' ::
                   `;..``::::''..;'
                     ``::,,,,::''

                   ___  ___   __________
                  / _ \/ _ | / __/ __/ /
                 / ___/ __ |_\ \_\ \/_/ 
                /_/  /_/ |_/___/___(_)  

            """)
            sys.exit(0)
        else:
            sys.exit(-1)
    
    finally:
        pass