# Copyright 2010-2011 Free Software Foundation, Inc.

# 

# This file is part of GNU Radio

# 

# GNU Radio is free software; you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation; either version 3, or (at your option)

# any later version.

# 

# GNU Radio is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU General Public License for more details.

# 

# You should have received a copy of the GNU General Public License

# along with GNU Radio; see the file COPYING.  If not, write to

# the Free Software Foundation, Inc., 51 Franklin Street,

# Boston, MA 02110-1301, USA.

if(DEFINED __INCLUDED_GR_TEST_CMAKE)

    return()

endif()

set(__INCLUDED_GR_TEST_CMAKE TRUE)

########################################################################

# Add a unit test and setup the environment for a unit test.

# Takes the same arguments as the ADD_TEST function.

#

# Before calling set the following variables:

# GR_TEST_TARGET_DEPS  - built targets for the library path

# GR_TEST_LIBRARY_DIRS - directories for the library path

# GR_TEST_PYTHON_DIRS  - directories for the python path

########################################################################

function(GR_ADD_TEST test_name)

    if(WIN32)

        #Ensure that the build exe also appears in the PATH.

        list(APPEND GR_TEST_TARGET_DEPS ${ARGN})

        #In the land of windows, all libraries must be in the PATH.

        #Since the dependent libraries are not yet installed,

        #we must manually set them in the PATH to run tests.

        #The following appends the path of a target dependency.

        foreach(target ${GR_TEST_TARGET_DEPS})

            get_target_property(location ${target} LOCATION)

            if(location)

                get_filename_component(path ${location} PATH)

                string(REGEX REPLACE "\\$\\(.*\\)" ${CMAKE_BUILD_TYPE} path ${path})

                list(APPEND GR_TEST_LIBRARY_DIRS ${path})

            endif(location)

        endforeach(target)

        #SWIG generates the python library files into a subdirectory.

        #Therefore, we must append this subdirectory into PYTHONPATH.

        #Only do this for the python directories matching the following:

        foreach(pydir ${GR_TEST_PYTHON_DIRS})

            get_filename_component(name ${pydir} NAME)

            if(name MATCHES "^(swig|lib|src)$")

                list(APPEND GR_TEST_PYTHON_DIRS ${pydir}/${CMAKE_BUILD_TYPE})

            endif()

        endforeach(pydir)

    endif(WIN32)

    file(TO_NATIVE_PATH ${CMAKE_CURRENT_SOURCE_DIR} srcdir)

    file(TO_NATIVE_PATH "${GR_TEST_LIBRARY_DIRS}" libpath) #ok to use on dir list?

    file(TO_NATIVE_PATH "${GR_TEST_PYTHON_DIRS}" pypath) #ok to use on dir list?

    set(environs "GR_DONT_LOAD_PREFS=1" "srcdir=${srcdir}")

    #http://www.cmake.org/pipermail/cmake/2009-May/029464.html

    #Replaced this add test + set environs code with the shell script generation.

    #Its nicer to be able to manually run the shell script to diagnose problems.

    #ADD_TEST(${ARGV})

    #SET_TESTS_PROPERTIES(${test_name} PROPERTIES ENVIRONMENT "${environs}")

    if(UNIX)

        set(binpath "${CMAKE_CURRENT_BINARY_DIR}:$PATH")

        #set both LD and DYLD paths to cover multiple UNIX OS library paths

        list(APPEND libpath "$LD_LIBRARY_PATH" "$DYLD_LIBRARY_PATH")

        list(APPEND pypath "$PYTHONPATH")

        #replace list separator with the path separator

        string(REPLACE ";" ":" libpath "${libpath}")

        string(REPLACE ";" ":" pypath "${pypath}")

        list(APPEND environs "PATH=${binpath}" "LD_LIBRARY_PATH=${libpath}" "DYLD_LIBRARY_PATH=${libpath}" "PYTHONPATH=${pypath}")

        #generate a bat file that sets the environment and runs the test

        find_program(SHELL sh)

        set(sh_file ${CMAKE_CURRENT_BINARY_DIR}/${test_name}_test.sh)

        file(WRITE ${sh_file} "#!${SHELL}\n")

        #each line sets an environment variable

        foreach(environ ${environs})

            file(APPEND ${sh_file} "export ${environ}\n")

        endforeach(environ)

        #load the command to run with its arguments

        foreach(arg ${ARGN})

            file(APPEND ${sh_file} "${arg} ")

        endforeach(arg)

        file(APPEND ${sh_file} "\n")

        #make the shell file executable

        execute_process(COMMAND chmod +x ${sh_file})

        add_test(${test_name} ${SHELL} ${sh_file})

    endif(UNIX)

    if(WIN32)

        list(APPEND libpath ${DLL_PATHS} "%PATH%")

        list(APPEND pypath "%PYTHONPATH%")

        #replace list separator with the path separator (escaped)

        string(REPLACE ";" "\\;" libpath "${libpath}")

        string(REPLACE ";" "\\;" pypath "${pypath}")

        list(APPEND environs "PATH=${libpath}" "PYTHONPATH=${pypath}")

        #generate a bat file that sets the environment and runs the test

        set(bat_file ${CMAKE_CURRENT_BINARY_DIR}/${test_name}_test.bat)

        file(WRITE ${bat_file} "@echo off\n")

        #each line sets an environment variable

        foreach(environ ${environs})

            file(APPEND ${bat_file} "SET ${environ}\n")

        endforeach(environ)

        #load the command to run with its arguments

        foreach(arg ${ARGN})

            file(APPEND ${bat_file} "${arg} ")

        endforeach(arg)

        file(APPEND ${bat_file} "\n")

        add_test(${test_name} ${bat_file})

    endif(WIN32)

endfunction(GR_ADD_TEST)