"""Simple test suite for the code server.  Running this requires that one start
up the code server as::

    $ sudo ./code_server.py

"""
from exam.xmlrpc_clients import code_server


def check_result(result, check='correct answer'):
   if check != 'correct answer':
       assert result[0] == False
   else:
       assert result[0] == True
   if "unable to connect" in result[1].lower():
        assert result[0], result[1]
   assert check in result[1].lower(), result[1]

def test_python():
    """Test if server runs Python code as expected."""
    src = 'while True: pass'
    result = code_server.run_code(src, '', '/tmp', language="python")
    check_result(result, 'more than ')
    src = 'x = 1'
    result = code_server.run_code(src, 'assert x == 1', '/tmp',
                language="python")
    check_result(result, 'correct answer')

    result = code_server.run_code(src, 'assert x == 0', '/tmp',
                language="python")
    check_result(result, 'assertionerror')

    src = 'abracadabra'
    result = code_server.run_code(src, 'assert x == 0', '/tmp',
            language="python")
    check_result(result, 'nameerror')


def test_c():
    """Test if server runs c code as expected."""
    src = """
                #include<stdiol.h>
                int ad(int a, int b)
                {return a+b;}
          """
    result = code_server.run_code(src, 'c_cpp_files/main.cpp', '/tmp', language="C")
    check_result(result, 'error')

    src = """
                int add(int a, int b)
                {return a+b}
          """
    result = code_server.run_code(src, 'c_cpp_files/main.cpp', '/tmp', language="C")
    check_result(result, 'compilation error')

    src = """
                int add(int a, int b)
                {while(1>0){}
                return a+b;}
          """
    result = code_server.run_code(src, 'c_cpp_files/main.cpp', '/tmp', language="C")
    check_result(result, 'more than')

    src = """
                int add(int a, int b)
                {return a+b;}
          """
    result = code_server.run_code(src, 'c_cpp_files/main.cpp', '/tmp', language="C")
    check_result(result, 'correct answer')

    src = """
                #include<stdio.h>
                int add(int a, int b)
                {printf("All Correct");}
          """
    result = code_server.run_code(src, 'c_cpp_files/main.cpp', '/tmp', language="C")
    check_result(result, 'incorrect')


def test_cpp():
    """Test if server runs c code as expected."""
    src = """
                int add(int a, int b)
                {
                     return a+b
                }
          """
    result = code_server.run_code(src, 'c_cpp_files/main.cpp', '/tmp', language="C++")
    check_result(result, 'compilation error')

    src = """
                int add(int a, int b)
                {
                     return a+b;
                }
          """
    result = code_server.run_code(src, 'c_cpp_files/main.cpp', '/tmp', language="C++")
    check_result(result, 'correct answer')

    src = """
                int dd(int a, int b)
                {
                      return a+b;
                }
          """
    result = code_server.run_code(src, 'c_cpp_files/main.cpp', '/tmp', language="C++")
    check_result(result, 'error')

    src = """
                int add(int a, int b)
                {
                      while(0==0)
                      {}
                      return a+b;
                }
          """
    result = code_server.run_code(src, 'c_cpp_files/main.cpp', '/tmp', language="C++")
    check_result(result, 'more than')

def test_bash():
    """Test if server runs Bash code as expected."""
    src = """
#!/bin/bash
    [[ $# -eq 2 ]] && echo $(( $1 + $2 )) && exit $(( $1 + $2 ))
    """
    result = code_server.run_code(src,
            'docs/sample.sh\ndocs/sample.args', '/tmp', language="bash")
    check_result(result)

    src = """
#!/bin/bash
    [[ $# -eq 2 ]] && echo $(( $1 - $2 )) && exit $(( $1 - $2 ))
    """
    result = code_server.run_code(src,
            'docs/sample.sh\ndocs/sample.args', '/tmp', language="bash")
    check_result(result, 'error')

    src = """\
#!/bin/bash
    while [ 1 ] ; do echo "" > /dev/null ; done
    """
    result = code_server.run_code(src,
            'docs/sample.sh\ndocs/sample.args', '/tmp', language="bash")
    check_result(result, 'more than ')

    src = '''
#!/bin/bash
    while [ 1 ] ; do echo "" > /dev/null
    '''
    result = code_server.run_code(src,
           'docs/sample.sh\ndocs/sample.args', '/tmp', language="bash")
    check_result(result, 'error')

    src = '''# Enter your code here.
#!/bin/bash
    while [ 1 ] ; do echo "" > /dev/null
    '''
    result = code_server.run_code(src,
            'docs/sample.sh\ndocs/sample.args', '/tmp', language="bash")
    check_result(result, 'oserror')

if __name__ == '__main__':
    test_python()
    test_bash()
    test_c()
    test_cpp()