[unittest/Sema] NFC: Add a way to create a protocol with a given name and parent type

Author: xedin

unittests/Sema/SemaFixture.cpp

@@ -76,6 +76,17 @@ Type SemaTest::getStdlibType(StringRef name) const {
   return Type();
 }
 
+ProtocolType *SemaTest::createProtocol(llvm::StringRef protocolName,
+                                       Type parent) {
+  auto *PD = new (Context)
+      ProtocolDecl(DC, SourceLoc(), SourceLoc(),
+                   Context.getIdentifier(protocolName), /*Inherited=*/{},
+                   /*trailingWhere=*/nullptr);
+  PD->setImplicit();
+
+  return ProtocolType::get(PD, parent, Context);
+}
+
 ConstraintSystem::PotentialBindings
 SemaTest::inferBindings(ConstraintSystem &cs, TypeVariableType *typeVar) {
   llvm::SmallDenseMap<TypeVariableType *, ConstraintSystem::PotentialBindings>

unittests/Sema/SemaFixture.h

@@ -15,6 +15,7 @@
 #include "swift/AST/Module.h"
 #include "swift/AST/SourceFile.h"
 #include "swift/AST/Type.h"
+#include "swift/AST/Types.h"
 #include "swift/Basic/LangOptions.h"
 #include "swift/Basic/Platform.h"
 #include "swift/Basic/SourceManager.h"
@@ -66,6 +67,9 @@ class SemaTest : public SemaTestBase {
 protected:
   Type getStdlibType(StringRef name) const;
 
+  ProtocolType *createProtocol(llvm::StringRef protocolName,
+                               Type parent = Type());
+
   static ConstraintSystem::PotentialBindings
   inferBindings(ConstraintSystem &cs, TypeVariableType *typeVar);
 };