build.gradle

@@ -15,7 +15,7 @@ buildscript {
         classpath 'com.palantir.gradle.plugintesting:gradle-plugin-testing:0.6.0'
         classpath 'com.gradle.publish:plugin-publish-plugin:1.3.1'
         classpath 'com.palantir.baseline:gradle-baseline-java:6.16.0'
-        classpath 'com.palantir.javaformat:gradle-palantir-java-format:2.55.0'
+        classpath 'com.palantir.javaformat:gradle-palantir-java-format:2.56.0'
         classpath 'com.palantir.suppressible-error-prone:gradle-suppressible-error-prone:2.5.0'
     }
 }

gradle-baseline-java/src/main/java/com/palantir/baseline/plugins/BaselineTesting.java

@@ -141,7 +141,6 @@ private void configureTestTask(TestingExtension testingExtension, Test task) {
 
     private static Stream<SourceSet> testSourceSetsWhereJunitToolchainDepsHaveNotBeenAutomaticallyAdded(
             Project project) {
-
         SourceSetContainer sourceSets = project.getExtensions().getByType(SourceSetContainer.class);
         TestingExtension testingExtension = project.getExtensions().getByType(TestingExtension.class);
 
@@ -150,7 +149,7 @@ private static Stream<SourceSet> testSourceSetsWhereJunitToolchainDepsHaveNotBee
 
         return project.getTasks().withType(Test.class).getNames().stream().flatMap(testTaskName -> {
             if (gradleVersionLessThan8 || testTaskNotCreatedByJvmTestSuites(testingExtension, testTaskName)) {
-                return Stream.of(sourceSets.getByName(testTaskName));
+                return Stream.ofNullable(sourceSets.findByName(testTaskName));
             }
 
             return Stream.empty();

gradle-baseline-java/src/test/groovy/com/palantir/baseline/BaselineTestingIntegrationTest.groovy

@@ -294,6 +294,23 @@ class BaselineTestingIntegrationTest extends IntegrationSpec {
         gradleVersionNumber << GradleTestVersions.gradleVersionsForTests
     }
 
+    def '#gradleVersionNumber: test task without source set'() {
+        when:
+        gradleVersion = gradleVersionNumber
+
+        buildFile << standardBuildFile
+        buildFile << '''
+            tasks.register('otherTest', Test.class)
+        '''.stripIndent(true)
+
+        then:
+        // Run a task that will attempt to resolve dependencies
+        runTasksSuccessfully('compileJava')
+
+        where:
+        gradleVersionNumber << GradleTestVersions.gradleVersionsForTests
+    }
+
     def 'running -Drecreate=true will re-run tests even if no code changes'() {
         buildFile << standardBuildFile
         file('src/test/java/test/JUnit5Test.java') << junit5Test