1 /*
2  * Copyright (c) 2021-2023 The Khronos Group Inc.
3  * Copyright (c) 2021-2023 Valve Corporation
4  * Copyright (c) 2021-2023 LunarG, Inc.
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining a copy
7  * of this software and/or associated documentation files (the "Materials"), to
8  * deal in the Materials without restriction, including without limitation the
9  * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10  * sell copies of the Materials, and to permit persons to whom the Materials are
11  * furnished to do so, subject to the following conditions:
12  *
13  * The above copyright notice(s) and this permission notice shall be included in
14  * all copies or substantial portions of the Materials.
15  *
16  * THE MATERIALS ARE PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
19  *
20  * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
21  * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
22  * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE MATERIALS OR THE
23  * USE OR OTHER DEALINGS IN THE MATERIALS.
24  *
25  * Author: Charles Giessen <charles@lunarg.com>
26  */
27 
28 #include "test_environment.h"
29 
30 // Makes any failed assertion throw, allowing for graceful cleanup of resources instead of hard aborts
31 class ThrowListener : public testing::EmptyTestEventListener {
32     void OnTestPartResult(const testing::TestPartResult& result) override {
33         if (result.type() == testing::TestPartResult::kFatalFailure) {
34             // We need to make sure an exception wasn't already thrown so we dont throw another exception at the same time
35             std::exception_ptr ex = std::current_exception();
36             if (ex) {
37                 return;
38             }
39             throw testing::AssertionException(result);
40         }
41     }
42 };
43 
main(int argc, char** argv)44 int main(int argc, char** argv) {
45 #if defined(_WIN32)
46     // Avoid "Abort, Retry, Ignore" dialog boxes
47     _set_error_mode(_OUT_TO_STDERR);
48     _set_abort_behavior(0, _WRITE_ABORT_MSG | _CALL_REPORTFAULT);
49     SetErrorMode(SEM_FAILCRITICALERRORS | SEM_NOGPFAULTERRORBOX | SEM_NOOPENFILEERRORBOX);
50     _CrtSetReportMode(_CRT_ASSERT, _CRTDBG_MODE_FILE);
51     _CrtSetReportFile(_CRT_ASSERT, _CRTDBG_FILE_STDERR);
52     _CrtSetReportMode(_CRT_WARN, _CRTDBG_MODE_FILE);
53     _CrtSetReportFile(_CRT_WARN, _CRTDBG_FILE_STDERR);
54     _CrtSetReportMode(_CRT_ERROR, _CRTDBG_MODE_FILE);
55     _CrtSetReportFile(_CRT_ERROR, _CRTDBG_FILE_STDERR);
56 #endif
57 
58     // clean up folders from old test
59     fs::delete_folder(fs::path(FRAMEWORK_BUILD_DIRECTORY) / "null_dir");
60     fs::delete_folder(fs::path(FRAMEWORK_BUILD_DIRECTORY) / "icd_manifests");
61     fs::delete_folder(fs::path(FRAMEWORK_BUILD_DIRECTORY) / "icd_env_vars_manifests");
62     fs::delete_folder(fs::path(FRAMEWORK_BUILD_DIRECTORY) / "explicit_layer_manifests");
63     fs::delete_folder(fs::path(FRAMEWORK_BUILD_DIRECTORY) / "explicit_env_var_layer_folder");
64     fs::delete_folder(fs::path(FRAMEWORK_BUILD_DIRECTORY) / "explicit_add_env_var_layer_folder");
65     fs::delete_folder(fs::path(FRAMEWORK_BUILD_DIRECTORY) / "implicit_layer_manifests");
66     fs::delete_folder(fs::path(FRAMEWORK_BUILD_DIRECTORY) / "override_layer_manifests");
67     fs::delete_folder(fs::path(FRAMEWORK_BUILD_DIRECTORY) / "app_package_manifests");
68     fs::delete_folder(fs::path(FRAMEWORK_BUILD_DIRECTORY) / "macos_bundle");
69 
70     // make sure the tests don't find these env-vars if they were set on the system
71     EnvVarWrapper vk_icd_filenames_env_var{"VK_ICD_FILENAMES"};
72     EnvVarWrapper vk_driver_files_env_var{"VK_DRIVER_FILES"};
73     EnvVarWrapper vk_add_driver_files_env_var{"VK_ADD_DRIVER_FILES"};
74     EnvVarWrapper vk_layer_path_env_var{"VK_LAYER_PATH"};
75     EnvVarWrapper vk_add_layer_path_env_var{"VK_ADD_LAYER_PATH"};
76     EnvVarWrapper vk_instance_layers_env_var{"VK_INSTANCE_LAYERS"};
77     EnvVarWrapper vk_loader_drivers_select_env_var{"VK_LOADER_DRIVERS_SELECT"};
78     EnvVarWrapper vk_loader_drivers_disable_env_var{"VK_LOADER_DRIVERS_DISABLE"};
79     EnvVarWrapper vk_loader_layers_enable_env_var{"VK_LOADER_LAYERS_ENABLE"};
80     EnvVarWrapper vk_loader_layers_disable_env_var{"VK_LOADER_LAYERS_DISABLE"};
81     EnvVarWrapper vk_loader_debug_env_var{"VK_LOADER_DEBUG"};
82     EnvVarWrapper vk_loader_disable_inst_ext_filter_env_var{"VK_LOADER_DISABLE_INST_EXT_FILTER"};
83 
84 #if COMMON_UNIX_PLATFORMS
85     // Set only one of the 4 XDG variables to /etc, let everything else be empty
86     EnvVarWrapper xdg_config_home_env_var{"XDG_CONFIG_HOME", ETC_DIR};
87     EnvVarWrapper xdg_config_dirs_env_var{"XDG_CONFIG_DIRS"};
88     EnvVarWrapper xdg_data_home_env_var{"XDG_DATA_HOME"};
89     EnvVarWrapper xdg_data_dirs_env_var{"XDG_DATA_DIRS"};
90     EnvVarWrapper home_env_var{"HOME", HOME_DIR};
91 #endif
92     ::testing::InitGoogleTest(&argc, argv);
93     ::testing::UnitTest::GetInstance()->listeners().Append(new ThrowListener);
94     int result = RUN_ALL_TESTS();
95 
96     return result;
97 }
98