libc/test/integration/startup/gpu/args_test.cpp

//===-- Loader test to check args to main ---------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#include "test/IntegrationTest/test.h"

TEST_MAIN(int argc, char **argv, char **envp) {
  ASSERT_TRUE(argc == 4);
  ASSERT_STREQ(argv[1], "1");
  ASSERT_STREQ(argv[2], "2");
  ASSERT_STREQ(argv[3], "3");

  bool found_france = false;
  bool found_germany = false;
  for (; *envp != nullptr; ++envp) {
    if (LIBC_NAMESPACE::cpp::string_view(*envp) == "FRANCE=Paris")
      found_france = true;
    if (LIBC_NAMESPACE::cpp::string_view(*envp) == "GERMANY=Berlin")
      found_germany = true;
  }

  ASSERT_TRUE(found_france && found_germany);

  return 0;
}