29 lines
853 B
Crystal
29 lines
853 B
Crystal
module Spectator
|
|
# Indicates a call to exit the application was performed.
|
|
class SystemExit < Exception
|
|
# Status code passed to the exit call.
|
|
getter status : Int32
|
|
|
|
# Creates the exception.
|
|
def initialize(message : String? = nil, cause : Exception? = nil, @status : Int32 = 0)
|
|
super(message, cause)
|
|
end
|
|
end
|
|
|
|
# Allow Spectator to exit normally when needed.
|
|
private def self.exit(status = 0) : NoReturn
|
|
::Crystal::System::Process.exit(status)
|
|
end
|
|
end
|
|
|
|
class Process
|
|
# Replace the typically used exit method with a method that raises.
|
|
# This allows tests to catch attempts to exit the application.
|
|
def self.exit(status = 0) : NoReturn
|
|
# Re-enable log that is disabled from at-exit handlers.
|
|
::Log.setup_from_env(default_level: :none)
|
|
|
|
raise ::Spectator::SystemExit.new(status: status)
|
|
end
|
|
end
|