Use new feature flag architecture when uploading databases

lib/feature-flags.js

@@ -19,7 +19,7 @@ var __importStar = (this && this.__importStar) || function (mod) {
     return result;
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.GitHubFeatureFlags = void 0;
+exports.createFeatureFlags = exports.GitHubFeatureFlags = void 0;
 const api_client_1 = require("./api-client");
 const repository_1 = require("./repository");
 const util = __importStar(require("./util"));
@@ -76,4 +76,23 @@ class GitHubFeatureFlags {
     }
 }
 exports.GitHubFeatureFlags = GitHubFeatureFlags;
+/**
+ * Create a feature flags instance with the specified set of enabled flags.
+ *
+ * This should be only used within tests.
+ */
+function createFeatureFlags(enabledFlags) {
+    return {
+        getDatabaseUploadsEnabled: async () => {
+            return enabledFlags.includes("database_uploads_enabled");
+        },
+        getMlPoweredQueriesEnabled: async () => {
+            return enabledFlags.includes("ml_powered_queries_enabled");
+        },
+        getUploadsDomainEnabled: async () => {
+            return enabledFlags.includes("uploads_domain_enabled");
+        },
+    };
+}
+exports.createFeatureFlags = createFeatureFlags;
 //# sourceMappingURL=feature-flags.js.map