"""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, '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_java():
    """Test if server runs java code as expected."""
    src = """
                class Test
                {
                int square_num(int a)
                {
                     return a*a;
                }
                }
          """
    result = code_server.run_code(src, 'java_files/main_square.java',
                                  '/tmp', language="java")
    check_result(result, 'correct answer')

    src = """
                class Test
                {
                int square_num(int a)
                {
                     return b*b;
                }
                }
          """
    result = code_server.run_code(src, 'java_files/main_square.java',
                                  '/tmp', language="java")
    check_result(result, 'error')

    src = """
                class Test
                {
                int square_nu(int a)
                {
                      return a*a;
                }
                }
          """
    result = code_server.run_code(src, 'java_files/main_square.java',
                                  '/tmp', language="java")
    check_result(result, 'error')

    src = """
                class Test
                {
                int square_num(int a)
                {
                      while(0==0)
                      {}
                }
                }
          """
    result = code_server.run_code(src, 'java_files/main_square.java',
                                  '/tmp', language="java")
    check_result(result, 'more than')

    src = """
                class Test
                {
                int square_num(int a)
                {
                      return a+b
                }
                }
          """
    result = code_server.run_code(src, 'java_files/main_square.java',
                                  '/tmp', language="java")
    check_result(result, 'error')

    src = """
                class Test
                {
                int square_num(int a)
                {
                      return a+b
          """
    result = code_server.run_code(src, 'java_files/main_square.java',
                                  '/tmp', language="java")
    check_result(result, 'error')

def test_scilab():
    """Test if server runs scilab code as expected."""
    src = """
                funcprot(0)
function[c]=add(a,b)
    c=a+b;
endfunction
          """
    result = code_server.run_code(src, 'scilab_files/test_add.sce',
                                  '/tmp', language="scilab")
    check_result(result, 'correct answer')

    src = """
                funcprot(0)
function[c]=add(a,b)
    c=a-b;
endfunction
          """
    result = code_server.run_code(src, 'scilab_files/test_add.sce',
                                  '/tmp', language="scilab")
    check_result(result, 'correct answer')

    src = """
                funcprot(0)
function[c]=add(a,b)
    c=a+b;
dis(
endfunction
          """
    result = code_server.run_code(src, 'scilab_files/test_add.sce',
                                  '/tmp', language="scilab")
    check_result(result, 'error')

    src = """
               funcprot(0)
function[c]=add(a,b)
    c=a
    while(1==1)
    end
endfunction
          """
    result = code_server.run_code(src, 'scilab_files/test_add.sce',
                                  '/tmp', language="scilab")
    check_result(result, 'error')

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()
    test_java()
    test_scilab()