Merge pull request #61579 from nate-chandler/test/basic-conveniences

[Test] Add a couple helpers.

Author: swift-ci

include/swift/SILOptimizer/Utils/ParseTestSpecification.h

@@ -125,6 +125,8 @@ struct Arguments {
     untakenIndex = 0;
   }
 
+  bool hasUntaken() { return untakenIndex < storage.size(); }
+
   Arguments() {}
   Arguments(Arguments const &) = delete;
   Arguments &operator=(Arguments const &) = delete;
@@ -159,8 +161,11 @@ struct Arguments {
 void getTestSpecifications(SILFunction *function,
                            SmallVectorImpl<std::string> &specifications);
 
+/// Given the string \p specification operand of a test_specification
+/// instruction from \p function, parse the arguments which it refers to into
+/// \p arguments and the component strings into \p argumentStrings.
 void parseTestArgumentsFromSpecification(
-    SILFunction *, StringRef specification, Arguments &arguments,
+    SILFunction *function, StringRef specification, Arguments &arguments,
     SmallVectorImpl<StringRef> &argumentStrings);
 
 } // namespace test

lib/SILOptimizer/UtilityPasses/UnitTestRunner.cpp

@@ -216,6 +216,14 @@ struct VisitAdjacentReborrowsOfPhiTest : UnitTest {
   }
 };
 
+// Arguments: NONE
+// Dumps:
+// - the function
+struct DumpFunction : UnitTest {
+  DumpFunction(UnitTestRunner *pass) : UnitTest(pass) {}
+  void invoke(Arguments &arguments) override { getFunction()->dump(); }
+};
+
 /// [new_tests] Add the new UnitTest subclass above this line.
 
 class UnitTestRunner : public SILFunctionTransform {
@@ -245,6 +253,8 @@ class UnitTestRunner : public SILFunctionTransform {
     return;                                                                    \
   }
 
+    ADD_UNIT_TEST_SUBCLASS("dump-function", DumpFunction)
+
     ADD_UNIT_TEST_SUBCLASS("test-specification-parsing", TestSpecificationTest)
     ADD_UNIT_TEST_SUBCLASS("canonicalize-ossa-lifetime",
                            CanonicalizeOSSALifetimeTest)