add tests for error_stop and optional returncode

Author: scivision

src/CMakeLists.txt

@@ -1,4 +1,5 @@
 set(SRC
+    lib/mod_stdlib.f90
     stdlib_experimental_ascii.f90
     stdlib_experimental_io.f90
     stdlib_experimental_error.f90

src/stdlib_experimental_error.f90

@@ -1,41 +1,53 @@
 module stdlib_experimental_error
+use, intrinsic :: iso_fortran_env, only: stderr=>error_unit
 implicit none
 private
 public :: assert, error_stop
 
 contains
 
-subroutine assert(condition)
+subroutine assert(condition, code)
 ! If condition == .false., it aborts the program.
 !
 ! Arguments
 ! ---------
 !
 logical, intent(in) :: condition
+integer, intent(in), optional :: code
 !
 ! Example
 ! -------
 !
 ! call assert(a == 5)
 
-if (.not. condition) call error_stop("Assert failed.")
+if (.not. condition) call error_stop("Assert failed.", code)
 end subroutine
 
-subroutine error_stop(msg)
+subroutine error_stop(msg, code)
 ! Aborts the program with nonzero exit code
 !
-! The statement "stop msg" will return 0 exit code when compiled using
-! gfortran. error_stop() uses the statement "stop 1" which returns an exit code
-! 1 and a print statement to print the message.
+! The "stop <character>" statement generally has return code 0.
+! To allow non-zero return code termination with character message,
+! error_stop() uses the statement "error stop", which by default
+! has exit code 1 and prints the message to stderr.
+! An optional integer return code "code" may be specified.
 !
 ! Example
 ! -------
 !
 ! call error_stop("Invalid argument")
 
-character(len=*) :: msg ! Message to print on stdout
-print *, msg
-stop 1
+character(len=*) :: msg ! Message to print on stderr
+integer, intent(in), optional :: code
+
+integer :: returncode
+
+if(present(code)) then
+  write(stderr,*) msg
+  error stop code
+else
+  error stop msg
+endif
 end subroutine
 
 end module

src/tests/CMakeLists.txt

@@ -1,3 +1,16 @@
 add_subdirectory(ascii)
 add_subdirectory(loadtxt)
 
+add_executable(test_dummy test_dummy.f90)
+target_link_libraries(test_dummy fortran_stdlib)
+add_test(NAME Dummy COMMAND $<TARGET_FILE:test_dummy>)
+
+add_executable(test_skip test_skip.f90)
+target_link_libraries(test_skip fortran_stdlib)
+add_test(NAME AlwaysSkip COMMAND $<TARGET_FILE:test_skip>)
+set_tests_properties(AlwaysSkip PROPERTIES SKIP_RETURN_CODE 77)
+
+add_executable(test_fail test_fail.f90)
+target_link_libraries(test_fail fortran_stdlib)
+add_test(NAME AlwaysFail COMMAND $<TARGET_FILE:test_fail>)
+set_tests_properties(AlwaysFail PROPERTIES WILL_FAIL true)

src/tests/test_fail.f90

@@ -0,0 +1,8 @@
+program AlwaysFail
+
+use stdlib_experimental_error, only : assert
+implicit none
+
+call assert(.false.)
+
+end program

src/tests/test_skip.f90

@@ -0,0 +1,8 @@
+program AlwaysSkip
+
+use stdlib_experimental_error, only : assert
+implicit none
+
+call assert(.false., 77)
+
+end program