sqlancer
diff --git a/‎src/sqlancer/postgres/gen/PostgresTableSpaceGenerator.java‎
Lines changed: 5 additions & 2 deletions b/‎src/sqlancer/postgres/gen/PostgresTableSpaceGenerator.java‎
Lines changed: 5 additions & 2 deletions
diff --git a/‎test/sqlancer/postgres/gen/TestPostgresTableSpaceGenerator.java‎
Lines changed: 25 additions & 0 deletions b/‎test/sqlancer/postgres/gen/TestPostgresTableSpaceGenerator.java‎
Lines changed: 25 additions & 0 deletions
@@ -1,5 +1,6 @@
 package sqlancer.postgres.gen;
 
+import sqlancer.IgnoreMeException;
 import sqlancer.common.query.ExpectedErrors;
 import sqlancer.common.query.SQLQueryAdapter;
 import sqlancer.postgres.PostgresGlobalState;
@@ -19,10 +20,12 @@ public PostgresTableSpaceGenerator(PostgresGlobalState globalState) {
     }
 
     public static SQLQueryAdapter generate(PostgresGlobalState globalState) {
-        // Skip tablespace generation if the option is disabled
+        // PostgresProvider.mapActions does not schedule this action when the option is disabled, but QPG
+        // selects actions by index without consulting the schedule, so the generator has to report that
+        // it has nothing to generate.
         PostgresOptions options = globalState.getDbmsSpecificOptions();
         if (!options.isTestTablespaces()) {
-            return null;
+            throw new IgnoreMeException();
         }
         return new PostgresTableSpaceGenerator(globalState).generateTableSpace();
     }
 
@@ -0,0 +1,25 @@
+package sqlancer.postgres.gen;
+
+import static org.junit.jupiter.api.Assertions.assertThrows;
+
+import org.junit.jupiter.api.Test;
+
+import sqlancer.IgnoreMeException;
+import sqlancer.postgres.PostgresGlobalState;
+import sqlancer.postgres.PostgresOptions;
+
+class TestPostgresTableSpaceGenerator {
+
+    @Test
+    void generateIsSkippedWhenTablespacesAreDisabled() {
+        PostgresGlobalState state = new PostgresGlobalState();
+        state.setDbmsSpecificOptions(new PostgresOptions() {
+            @Override
+            public boolean isTestTablespaces() {
+                return false;
+            }
+        });
+
+        assertThrows(IgnoreMeException.class, () -> PostgresTableSpaceGenerator.generate(state));
+    }
+}
Original file line number	Diff line number	Diff line change
`@@ -1,5 +1,6 @@`
`1`	`1`	`package sqlancer.postgres.gen;`
`2`	`2`
	`3`	`+import sqlancer.IgnoreMeException;`
`3`	`4`	`import sqlancer.common.query.ExpectedErrors;`
`4`	`5`	`import sqlancer.common.query.SQLQueryAdapter;`
`5`	`6`	`import sqlancer.postgres.PostgresGlobalState;`
`@@ -19,10 +20,12 @@ public PostgresTableSpaceGenerator(PostgresGlobalState globalState) {`
`19`	`20`	`}`
`20`	`21`
`21`	`22`	`public static SQLQueryAdapter generate(PostgresGlobalState globalState) {`
`22`		`- // Skip tablespace generation if the option is disabled`
	`23`	`+ // PostgresProvider.mapActions does not schedule this action when the option is disabled, but QPG`
	`24`	`+ // selects actions by index without consulting the schedule, so the generator has to report that`
	`25`	`+ // it has nothing to generate.`
`23`	`26`	`PostgresOptions options = globalState.getDbmsSpecificOptions();`
`24`	`27`	`if (!options.isTestTablespaces()) {`
`25`		`- return null;`
	`28`	`+ throw new IgnoreMeException();`
`26`	`29`	`}`
`27`	`30`	`return new PostgresTableSpaceGenerator(globalState).generateTableSpace();`
`28`	`31`	`}`