B
    u9ai                 @   s   d Z ddlZddlZddlZddlZddlmZ G dd dejZG dd dej	Z
ddd	Zd
d Zedkrdejkred ndejkre   ne  dS )z@
Test script for the 'cmd' module
Original by Michael Schneider
    N)supportc               @   sP   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d Zdd Z	dd Z
dd ZdS )samplecmdclassa-  
    Instance the sampleclass:
    >>> mycmd = samplecmdclass()

    Test for the function parseline():
    >>> mycmd.parseline("")
    (None, None, '')
    >>> mycmd.parseline("?")
    ('help', '', 'help ')
    >>> mycmd.parseline("?help")
    ('help', 'help', 'help help')
    >>> mycmd.parseline("!")
    ('shell', '', 'shell ')
    >>> mycmd.parseline("!command")
    ('shell', 'command', 'shell command')
    >>> mycmd.parseline("func")
    ('func', '', 'func')
    >>> mycmd.parseline("func arg1")
    ('func', 'arg1', 'func arg1')


    Test for the function onecmd():
    >>> mycmd.onecmd("")
    >>> mycmd.onecmd("add 4 5")
    9
    >>> mycmd.onecmd("")
    9
    >>> mycmd.onecmd("test")
    *** Unknown syntax: test

    Test for the function emptyline():
    >>> mycmd.emptyline()
    *** Unknown syntax: test

    Test for the function default():
    >>> mycmd.default("default")
    *** Unknown syntax: default

    Test for the function completedefault():
    >>> mycmd.completedefault()
    This is the completedefault method
    >>> mycmd.completenames("a")
    ['add']

    Test for the function completenames():
    >>> mycmd.completenames("12")
    []
    >>> mycmd.completenames("help")
    ['help']

    Test for the function complete_help():
    >>> mycmd.complete_help("a")
    ['add']
    >>> mycmd.complete_help("he")
    ['help']
    >>> mycmd.complete_help("12")
    []
    >>> sorted(mycmd.complete_help(""))
    ['add', 'exit', 'help', 'shell']

    Test for the function do_help():
    >>> mycmd.do_help("testet")
    *** No help on testet
    >>> mycmd.do_help("add")
    help text for add
    >>> mycmd.onecmd("help add")
    help text for add
    >>> mycmd.do_help("")
    <BLANKLINE>
    Documented commands (type help <topic>):
    ========================================
    add  help
    <BLANKLINE>
    Undocumented commands:
    ======================
    exit  shell
    <BLANKLINE>

    Test for the function print_topics():
    >>> mycmd.print_topics("header", ["command1", "command2"], 2 ,10)
    header
    ======
    command1
    command2
    <BLANKLINE>

    Test for the function columnize():
    >>> mycmd.columnize([str(i) for i in range(20)])
    0  1  2  3  4  5  6  7  8  9  10  11  12  13  14  15  16  17  18  19
    >>> mycmd.columnize([str(i) for i in range(20)], 10)
    0  7   14
    1  8   15
    2  9   16
    3  10  17
    4  11  18
    5  12  19
    6  13

    This is an interactive test, put some commands in the cmdqueue attribute
    and let it execute
    This test includes the preloop(), postloop(), default(), emptyline(),
    parseline(), do_help() functions
    >>> mycmd.use_rawinput=0
    >>> mycmd.cmdqueue=["", "add", "add 4 5", "help", "help add","exit"]
    >>> mycmd.cmdloop()
    Hello from preloop
    help text for add
    *** invalid number of arguments
    9
    <BLANKLINE>
    Documented commands (type help <topic>):
    ========================================
    add  help
    <BLANKLINE>
    Undocumented commands:
    ======================
    exit  shell
    <BLANKLINE>
    help text for add
    Hello from postloop
    c             C   s   t d d S )NzHello from preloop)print)self r   /usr/lib/python3.7/test_cmd.pypreloop   s    zsamplecmdclass.preloopc             C   s   t d d S )NzHello from postloop)r   )r   r   r   r   postloop   s    zsamplecmdclass.postloopc             G   s   t d d S )Nz"This is the completedefault method)r   )r   Zignoredr   r   r   completedefault   s    zsamplecmdclass.completedefaultc             C   s   t d d S )Nzcomplete command)r   )r   r   r   r   complete_command   s    zsamplecmdclass.complete_commandc             C   s   d S )Nr   )r   sr   r   r   do_shell   s    zsamplecmdclass.do_shellc             C   sh   |  }t|dkr td d S ydd |D }W n tk
rN   td d S X t|d |d   d S )N   z*** invalid number of argumentsc             S   s   g | ]}t |qS r   )int).0ir   r   r   
<listcomp>   s    z)samplecmdclass.do_add.<locals>.<listcomp>z*** arguments should be numbersr      )splitlenr   
ValueError)r   r   lr   r   r   do_add   s    zsamplecmdclass.do_addc             C   s   t d d S )Nzhelp text for add)r   )r   r   r   r   help_add   s    zsamplecmdclass.help_addc             C   s   dS )NTr   )r   argr   r   r   do_exit   s    zsamplecmdclass.do_exitN)__name__
__module____qualname____doc__r   r	   r
   r   r   r   r   r   r   r   r   r   r      s   yr   c               @   s>   e Zd ZG dd dejZG dd deZdd Zdd Zd	S )
TestAlternateInputc               @   s   e Zd Zdd Zdd ZdS )zTestAlternateInput.simplecmdc             C   s   t || jd d S )N)file)r   stdout)r   argsr   r   r   do_print   s    z%TestAlternateInput.simplecmd.do_printc             C   s   dS )NTr   )r   r#   r   r   r   do_EOF   s    z#TestAlternateInput.simplecmd.do_EOFN)r   r   r   r$   r%   r   r   r   r   	simplecmd   s   r&   c               @   s   e Zd Zdd ZdS )zTestAlternateInput.simplecmd2c             C   s   t d| jd dS )Nz*** Unknown syntax: EOF)r!   T)r   r"   )r   r#   r   r   r   r%      s    z$TestAlternateInput.simplecmd2.do_EOFN)r   r   r   r%   r   r   r   r   
simplecmd2   s   r'   c             C   sB   t d}t  }| j||d}d|_|  | | d d S )Nzprint test
print test2)stdinr"   Fz(Cmd) test
(Cmd) test2
(Cmd) )ioStringIOr&   use_rawinputcmdloopassertMultiLineEqualgetvalue)r   inputoutputcmdr   r   r   test_file_with_missing_final_nl   s    

z2TestAlternateInput.test_file_with_missing_final_nlc             C   sx   t d}t  }| j||d}d|_|  | | d t d}t  }||_||_|  | | d d S )Nzprint test
print test2)r(   r"   Fz5(Cmd) test
(Cmd) test2
(Cmd) *** Unknown syntax: EOF
zprint 

z,(Cmd) 
(Cmd) 
(Cmd) *** Unknown syntax: EOF
)	r)   r*   r'   r+   r,   r-   r.   r(   r"   )r   r/   r0   r1   r   r   r   test_input_reset_at_EOF   s    



z*TestAlternateInput.test_input_reset_at_EOFN)	r   r   r   r1   Cmdr&   r'   r2   r3   r   r   r   r   r       s   	r    c             C   s&   ddl m} t||  tt d S )Nr   )test_cmd)testr5   r   Zrun_doctestZrun_unittestr    )verboser5   r   r   r   	test_main   s    r8   c             C   sP   t d}|jtjtjgddd}|d | }td |j	dd| d d S )	Ntracer   r   )Z
ignoredirsr9   countz4import importlib; importlib.reload(cmd); test_main()zWriting coverage results...T)Zshow_missingZsummarycoverdir)
r   import_moduleZTracesysbase_prefixbase_exec_prefixrunresultsr   Zwrite_results)r;   r9   Ztracerrr   r   r   test_coverage   s    


rC   __main__z-cz/tmp/cmd.coverz-i)N)r   r1   r=   Zunittestr)   r6   r   r4   r   ZTestCaser    r8   rC   r   argvr,   r   r   r   r   <module>   s     3
	


