Closes #6152: Add license type to what is reported to Mixpanel

Adds the license type to data sent to Mixpanel.

inc/Engine/Admin/Settings/Page.php

@@ -260,25 +260,7 @@ public function customer_data() {
 			'is_from_one_dot_com' => false,
 		];
 
-		if (
-			false === $user
-			||
-			! isset( $user->licence_account, $user->licence_expiration )
-		) {
-			return $data;
-		}
-
-		if (
-			1 <= $user->licence_account
-			&&
-			$user->licence_account < 3
-		) {
-			$data['license_type'] = 'Single';
-		} elseif ( -1 === (int) $user->licence_account ) {
-			$data['license_type'] = 'Infinite';
-		} else {
-			$data['license_type'] = 'Plus';
-		}
+		$data['license_type'] = get_license_type($user);
 
 		$data['license_class']       = time() < $user->licence_expiration ? 'wpr-isValid' : 'wpr-isInvalid';
 		$data['license_expiration']  = date_i18n( get_option( 'date_format' ), (int) $user->licence_expiration );

inc/admin/admin.php

@@ -324,6 +324,11 @@ function rocket_analytics_data() {
 		$data['cdn_cnames'] = 0;
 	}
 
+	$customer_data = get_transient( 'wp_rocket_customer_data' );
+	if ( false !== $customer_data ) {
+		$data['license_type'] = get_license_type($customer_data);
+	}
+
 	return $data;
 }
 

inc/functions/admin.php

@@ -526,3 +526,28 @@ function rocket_create_options_hash( $value ) {
 
 	return md5( wp_json_encode( $value_diff ) );
 }
+
+/**
+ * This function returns the license type for a customer.
+ *
+ * @param object $customer_data
+ * @return string
+ */
+function get_license_type($customer_data) {
+    $type = '';
+
+	if ( false === $customer_data || ! isset( $customer_data->licence_account ) ) {
+	    return __( 'Unavailable', 'rocket' );
+    }
+
+    if ( 1 <= $customer_data->licence_account && $customer_data->licence_account < 3
+    ) {
+	    $type = 'Single';
+    } elseif ( -1 === (int) $customer_data->licence_account ) {
+	    $type = 'Infinite';
+    } else {
+	    $type = 'Plus';
+    }
+
+    return $type;
+}

tests/Fixtures/inc/functions/admin/getLicenseType.php

@@ -0,0 +1,28 @@
+<?php
+
+return [
+   'singleLicense' => [
+     'config' => [
+        'customer_data' => (object)['licence_account' => 1],
+     ],
+     'expected' => 'Single',
+   ],
+   'infiniteLicense' => [
+     'config' => [
+        'customer_data' => (object)['licence_account' => -1],
+     ],
+     'expected' => 'Infinite',
+   ],
+   'plusLicense' => [
+     'config' => [
+        'customer_data' => (object)['licence_account' => 3],
+     ],
+     'expected' => 'Plus',
+   ],
+   'unavailableLicense' => [
+     'config' => [
+        'customer_data' => null,
+     ],
+     'expected' => 'Unavailable',
+   ],
+];

tests/Unit/inc/functions/admin/getLicenseType.php

@@ -0,0 +1,24 @@
+<?php
+
+namespace WP_Rocket\Tests\Unit\inc\functions\admin;
+
+use WP_Rocket\Tests\Unit\TestCase;
+
+class Test_GetLicenseType extends TestCase {
+
+    public function setUp(): void {
+        parent::setUp();
+    }
+
+    /**
+     * @dataProvider configTestData
+     */
+    public function testReturnAsExpected($config, $expected) {
+        $this->stubTranslationFunctions();
+        // Extract 'customer_data' from the config
+        $customer_data = $config['customer_data'];
+        $result = get_license_type($customer_data);
+
+        $this->assertEquals($expected, $result);
+    }
+}