aboutsummaryrefslogtreecommitdiff
path: root/unittests/clang-tidy/ObjCModuleTest.cpp
blob: 92ae8e1038f07e48a1dcc9430b10d881e8b0bdfe (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
//===---- ObjCModuleTest.cpp - clang-tidy ---------------------------------===//
//
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//

#include "ClangTidyTest.h"
#include "gtest/gtest.h"
#include "objc/ForbiddenSubclassingCheck.h"

using namespace clang::tidy::objc;

namespace clang {
namespace tidy {
namespace test {

TEST(ObjCForbiddenSubclassing, AllowedSubclass) {
  std::vector<ClangTidyError> Errors;
  runCheckOnCode<ForbiddenSubclassingCheck>(
      "@interface Foo\n"
      "@end\n"
      "@interface Bar : Foo\n"
      "@end\n",
      &Errors,
      "input.m");
  EXPECT_EQ(0ul, Errors.size());
}

TEST(ObjCForbiddenSubclassing, ForbiddenSubclass) {
  std::vector<ClangTidyError> Errors;
  runCheckOnCode<ForbiddenSubclassingCheck>(
      "@interface UIImagePickerController\n"
      "@end\n"
      "@interface Foo : UIImagePickerController\n"
      "@end\n",
      &Errors,
      "input.m");
  EXPECT_EQ(1ul, Errors.size());
  EXPECT_EQ(
      "Objective-C interface 'Foo' subclasses 'UIImagePickerController', which is not intended to be subclassed",
      Errors[0].Message.Message);
}

} // namespace test
} // namespace tidy
} // namespace clang