// Part of the Crubit project, under the Apache License v2.0 with LLVM
// Exceptions. See /LICENSE for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

#include "nullability/test/check_diagnostics.h"
#include "external/llvm-project/third-party/unittest/googletest/include/gtest/gtest.h"

namespace clang::tidy::nullability {
namespace {

TEST(PointerNullabilityTest, Arrays) {
  EXPECT_TRUE(checkDiagnostics(R"cc(
    int *_Nullable x[2][2];
    int *_Nonnull y[2][2];

    void target() {
      __assert_nullability<NK_nullable>(x[0][1]);
      __assert_nullability<NK_nonnull>(y[0][1]);
    }
  )cc"));
}

TEST(PointerNullabilityTest, ArrayArgumentSyntax) {
  EXPECT_TRUE(checkDiagnostics(R"cc(
    // Though declared as arrays, these parameters are actually pointers.
    // Their nullability syntax is unusual.
    void target(int unknown[], int nonnull[_Nonnull], int nullable[_Nullable]) {
      *unknown;
      *nonnull;
      *nullable;  // [[unsafe]]
    }
  )cc"));
}

TEST(PointerNullabilityTest, VectorsArentPointers) {
  // We used to crash here: generally in x[0], x is (decayed to) a pointer,
  // and x[0] is a dereference and thus has a shorter nullability vector.
  // However x may also be a vector type, then x[0] has the same vector.
  EXPECT_TRUE(checkDiagnostics(R"cc(
    using char4 = __attribute__((ext_vector_type(4))) char;
    void target(char __attribute__((vector_size(4))) x, char4 y) {
      __assert_nullability<>(x);
      __assert_nullability<>(x[0]);
      __assert_nullability<>(y[0]);
    }
  )cc"));
}
}  // namespace
}  // namespace clang::tidy::nullability