[9.0] [Rules migration][Integration test] Create migration API (#11232)…

… (#210999) (#211085) # Backport This will backport the following commits from `main` to `9.0`: - [[Rules migration][Integration test] Create migration API (#11232) (#210999)](#210999)  ### Questions ? Please refer to the [Backport tool documentation](https://github.com/sqren/backport)  Co-authored-by: Ievgen Sorokopud <[email protected]>
elastic · Feb 13, 2025 · e925dd8 · e925dd8
1 parent 74b8e15
commit e925dd8
Show file tree

Hide file tree

Showing 9 changed files with 395 additions and 206 deletions.
diff --git a/...curity_solution/server/lib/siem_migrations/rules/data/rule_migrations_data_base_client.ts b/...curity_solution/server/lib/siem_migrations/rules/data/rule_migrations_data_base_client.ts
@@ -41,11 +41,16 @@ export class RuleMigrationsDataBaseClient {
       return this.currentUser.profile_uid;
     }
     const username = this.currentUser.username;
-    const users = await this.esScopedClient.asCurrentUser.security.getUser({
-      username,
-      with_profile_uid: true,
-    });
-    return users[username].profile_uid;
+    try {
+      const users = await this.esScopedClient.asCurrentUser.security.getUser({
+        username,
+        with_profile_uid: true,
+      });
+      return users[username].profile_uid;
+    } catch (error) {
+      this.logger.error(`Error getting profile_uid for user ${username}: ${error}`);
+      return username;
+    }
   }
 
   protected processResponseHits<T extends object>(

diff --git a/...n_api_integration/test_suites/siem_migrations/rules/trial_license_complete_tier/create.ts b/...n_api_integration/test_suites/siem_migrations/rules/trial_license_complete_tier/create.ts
@@ -0,0 +1,105 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License
+ * 2.0; you may not use this file except in compliance with the Elastic License
+ * 2.0.
+ */
+
+import expect from 'expect';
+import { v4 as uuidv4 } from 'uuid';
+import { SiemMigrationStatus } from '@kbn/security-solution-plugin/common/siem_migrations/constants';
+import {
+  defaultOriginalRule,
+  deleteAllMigrationRules,
+  migrationResourcesRouteHelpersFactory,
+  migrationRulesRouteHelpersFactory,
+  splunkRuleWithResources,
+} from '../../utils';
+import { FtrProviderContext } from '../../../../ftr_provider_context';
+
+export default ({ getService }: FtrProviderContext) => {
+  const es = getService('es');
+  const supertest = getService('supertest');
+  const migrationRulesRoutes = migrationRulesRouteHelpersFactory(supertest);
+  const migrationResourcesRoutes = migrationResourcesRouteHelpersFactory(supertest);
+
+  describe('@ess @serverless @serverlessQA Create API', () => {
+    beforeEach(async () => {
+      await deleteAllMigrationRules(es);
+    });
+
+    it('should create migrations with provided id', async () => {
+      const migrationId = uuidv4();
+      await migrationRulesRoutes.create({ migrationId, body: [defaultOriginalRule] });
+
+      // fetch migration rule
+      const response = await migrationRulesRoutes.get({ migrationId });
+      expect(response.body.total).toEqual(1);
+
+      const migrationRule = response.body.data[0];
+      expect(migrationRule).toEqual(
+        expect.objectContaining({
+          migration_id: migrationId,
+          original_rule: defaultOriginalRule,
+          status: SiemMigrationStatus.PENDING,
+        })
+      );
+    });
+
+    it('should create migrations without provided id', async () => {
+      const {
+        body: { migration_id: migrationId },
+      } = await migrationRulesRoutes.create({ body: [defaultOriginalRule] });
+
+      // fetch migration rule
+      const response = await migrationRulesRoutes.get({ migrationId });
+      expect(response.body.total).toEqual(1);
+
+      const migrationRule = response.body.data[0];
+      expect(migrationRule).toEqual(
+        expect.objectContaining({
+          migration_id: migrationId,
+          original_rule: defaultOriginalRule,
+          status: SiemMigrationStatus.PENDING,
+        })
+      );
+    });
+
+    it('should create migrations with the rules that have resources', async () => {
+      const migrationId = uuidv4();
+      await migrationRulesRoutes.create({ migrationId, body: [splunkRuleWithResources] });
+
+      // fetch migration rule
+      const response = await migrationRulesRoutes.get({ migrationId });
+      expect(response.body.total).toEqual(1);
+
+      const migrationRule = response.body.data[0];
+      expect(migrationRule).toEqual(
+        expect.objectContaining({
+          migration_id: migrationId,
+          original_rule: splunkRuleWithResources,
+          status: SiemMigrationStatus.PENDING,
+        })
+      );
+
+      // fetch missing resources
+      const resourcesResponse = await migrationResourcesRoutes.getMissingResources({
+        migrationId,
+      });
+      expect(resourcesResponse.body).toEqual([
+        { type: 'macro', name: 'summariesonly' },
+        { type: 'macro', name: 'drop_dm_object_name(1)' },
+        { type: 'lookup', name: 'malware_tracker' },
+      ]);
+    });
+
+    it('should return no content error', async () => {
+      const migrationId = uuidv4();
+      await migrationRulesRoutes.create({ migrationId, body: [], expectStatusCode: 204 });
+
+      // fetch migration rule
+      const response = await migrationRulesRoutes.get({ migrationId });
+      expect(response.body.total).toEqual(0);
+    });
+  });
+};