From d3ce8ae568bfd763ece653a9a3c45c77e8374c31 Mon Sep 17 00:00:00 2001
From: Pierre Gayvallet <pierre.gayvallet@elastic.co>
Date: Tue, 15 Dec 2020 13:15:12 +0100
Subject: [PATCH] properly recognize enterprise licenses (#85849) (#85921)

---
 .../plugins/licensing/server/plugin.test.ts   | 19 ++++++++++++++
 x-pack/plugins/licensing/server/plugin.ts     |  2 +-
 x-pack/test/licensing_plugin/scenario.ts      | 25 +++++++++++++++++++
 .../test/licensing_plugin/server/updates.ts   |  8 ++++++
 4 files changed, 53 insertions(+), 1 deletion(-)

diff --git a/x-pack/plugins/licensing/server/plugin.test.ts b/x-pack/plugins/licensing/server/plugin.test.ts
index b1669db00f227..7aec81f84f978 100644
--- a/x-pack/plugins/licensing/server/plugin.test.ts
+++ b/x-pack/plugins/licensing/server/plugin.test.ts
@@ -81,6 +81,25 @@ describe('licensing plugin', () => {
         expect(license.isAvailable).toBe(true);
       });
 
+      it('calls `callAsInternalUser` with the correct parameters', async () => {
+        const esClient = elasticsearchServiceMock.createLegacyClusterClient();
+        esClient.callAsInternalUser.mockResolvedValue({
+          license: buildRawLicense(),
+          features: {},
+        });
+
+        const coreSetup = createCoreSetupWith(esClient);
+        await plugin.setup(coreSetup);
+        const { license$ } = await plugin.start();
+        await license$.pipe(take(1)).toPromise();
+
+        expect(esClient.callAsInternalUser).toHaveBeenCalledTimes(1);
+        expect(esClient.callAsInternalUser).toHaveBeenCalledWith('transport.request', {
+          method: 'GET',
+          path: '/_xpack?accept_enterprise=true',
+        });
+      });
+
       it('observable receives updated licenses', async () => {
         const types: LicenseType[] = ['basic', 'gold', 'platinum'];
 
diff --git a/x-pack/plugins/licensing/server/plugin.ts b/x-pack/plugins/licensing/server/plugin.ts
index 2ee8d26419571..3823f251fb7fb 100644
--- a/x-pack/plugins/licensing/server/plugin.ts
+++ b/x-pack/plugins/licensing/server/plugin.ts
@@ -182,7 +182,7 @@ export class LicensingPlugin implements Plugin<LicensingPluginSetup, LicensingPl
     try {
       const response = await clusterClient.callAsInternalUser('transport.request', {
         method: 'GET',
-        path: '/_xpack',
+        path: '/_xpack?accept_enterprise=true',
       });
 
       const normalizedLicense = response.license
diff --git a/x-pack/test/licensing_plugin/scenario.ts b/x-pack/test/licensing_plugin/scenario.ts
index 9b73a7793c527..b351c7443cbb6 100644
--- a/x-pack/test/licensing_plugin/scenario.ts
+++ b/x-pack/test/licensing_plugin/scenario.ts
@@ -67,6 +67,31 @@ export function createScenario({ getService, getPageObjects }: FtrProviderContex
       expect(response.body.trial_was_started).to.be(true);
     },
 
+    async startEnterprise() {
+      const response = await esSupertestWithoutAuth
+        .post('/_license/?acknowledge=true')
+        .send({
+          license: {
+            uid: '00000000-d3ad-7357-c0d3-000000000000',
+            type: 'enterprise',
+            issue_date_in_millis: 1577836800000,
+            start_date_in_millis: 1577836800000,
+            // expires 2022-12-31
+            expiry_date_in_millis: 1672531199999,
+            max_resource_units: 250,
+            max_nodes: null,
+            issued_to: 'Elastic Internal Use (development environments)',
+            issuer: 'Elastic',
+            signature:
+              'AAAABQAAAA1gHUVis7hel8b8nNCAAAAAIAo5/x6hrsGh1GqqrJmy4qgmEC7gK0U4zQ6q5ZEMhm4jAAABAKMR+w3KZsMJfG5jNWgZXJLwRmiNqN7k94vKFgRdj1yM+gA9ufhXIn9d01OvFhPjilIqm+fxVjCxXwGKbFRiwtTWnTYjXPuNml+qCFGgUWguWEcVoIW6VU7/lYOqMJ4EB4zOMLe93P267iaDm542aelQrW1OJ69lGGuPBik8v9r1bNZzKBQ99VUr/qoosGDAm0udh2HxWzYoCL5lDML5Niy87xlVCubSSBXdUXzUgdZKKk6pKaMdHswB1gjvEfnwqPxEWAyrV0BCr/T1WehXd7U4p6/zt6sJ6cPh+34AZe9g4+3WPKrZhX4iaSHMDDHn4HNjO72CZ2oi42ZDNnJ37tA=',
+          },
+        })
+        .auth('license_manager_user', 'license_manager_user-password')
+        .expect(200);
+
+      expect(response.body.license_status).to.be('valid');
+    },
+
     async deleteLicense() {
       const response = await esSupertestWithoutAuth
         .delete('/_license')
diff --git a/x-pack/test/licensing_plugin/server/updates.ts b/x-pack/test/licensing_plugin/server/updates.ts
index e24b71939213c..ecfaae5f46620 100644
--- a/x-pack/test/licensing_plugin/server/updates.ts
+++ b/x-pack/test/licensing_plugin/server/updates.ts
@@ -62,5 +62,13 @@ export default function (ftrContext: FtrProviderContext) {
       // banner shown only when license expired not just deleted
       await testSubjects.missingOrFail('licenseExpiredBanner');
     });
+
+    it('properly recognize an enterprise license', async () => {
+      await scenario.startEnterprise();
+      await scenario.waitForPluginToDetectLicenseUpdate();
+
+      const enterpriseLicense = await scenario.getLicense();
+      expect(enterpriseLicense.license?.type).to.be('enterprise');
+    });
   });
 }