From 9c1c6590e89e2ea4ca388c90b79e3018905fac79 Mon Sep 17 00:00:00 2001
From: Mehmood Ahmad <31419912+mehmoodak@users.noreply.github.com>
Date: Fri, 15 Nov 2024 12:42:34 +0500
Subject: [PATCH] Adds Tracks Analytics on backend (#155)

Adds Tracks Analytics by using the Telemetry library available in MU Plugins which will only work if site is hosted on VIP Platform or if the analytics is enabled via filter.
---
 composer.json                                 |   3 +-
 composer.lock                                 |  43 ++-
 inc/Analytics/EnvironmentConfig.php           |  81 +++++
 inc/Analytics/TracksAnalytics.php             | 170 +++++++++++
 inc/Config/DataSource/HttpDataSource.php      |   9 +-
 inc/Editor/BlockManagement/ConfigStore.php    |  32 +-
 inc/REST/DataSourceController.php             |  61 +++-
 psalm.xml                                     |   2 +
 remote-data-blocks.php                        |   3 +
 src/data-sources/hooks/useDataSources.ts      |  12 +-
 tests/inc/Analytics/EnvironmentConfigTest.php |  77 +++++
 tests/inc/Analytics/TracksAnalyticsTest.php   | 277 ++++++++++++++++++
 tests/inc/Config/HttpDataSourceTest.php       |  23 ++
 tests/inc/Editor/ConfigStoreTest.php          |  32 ++
 tests/inc/bootstrap.php                       |   1 +
 tests/inc/stubs.php                           |  22 +-
 tests/inc/test-utils.php                      |  61 ++++
 17 files changed, 891 insertions(+), 18 deletions(-)
 create mode 100644 inc/Analytics/EnvironmentConfig.php
 create mode 100644 inc/Analytics/TracksAnalytics.php
 create mode 100644 tests/inc/Analytics/EnvironmentConfigTest.php
 create mode 100644 tests/inc/Analytics/TracksAnalyticsTest.php
 create mode 100644 tests/inc/Config/HttpDataSourceTest.php
 create mode 100644 tests/inc/Editor/ConfigStoreTest.php
 create mode 100644 tests/inc/test-utils.php

diff --git a/composer.json b/composer.json
index 06279533..65ef6b5f 100644
--- a/composer.json
+++ b/composer.json
@@ -41,7 +41,8 @@
     "slevomat/coding-standard": "^8.15",
     "php-stubs/wordpress-stubs": "^6.6",
     "psalm/phar": "^5.26",
-    "mockery/mockery": "^1.6"
+    "mockery/mockery": "^1.6",
+    "php-stubs/wordpress-globals": "^0.2.0"
   },
   "config": {
     "allow-plugins": {
diff --git a/composer.lock b/composer.lock
index 51eb7b72..78bcf23e 100644
--- a/composer.lock
+++ b/composer.lock
@@ -4,7 +4,7 @@
         "Read more about it at https://getcomposer.org/doc/01-basic-usage.md#installing-dependencies",
         "This file is @generated automatically"
     ],
-    "content-hash": "829c7d99f898737644ae2dd3d45313e0",
+    "content-hash": "5e0d80162087cfd0477746d29ec55e89",
     "packages": [
         {
             "name": "galbar/jsonpath",
@@ -1292,6 +1292,47 @@
             },
             "time": "2022-02-21T01:04:05+00:00"
         },
+        {
+            "name": "php-stubs/wordpress-globals",
+            "version": "v0.2.0",
+            "source": {
+                "type": "git",
+                "url": "https://github.com/php-stubs/wordpress-globals.git",
+                "reference": "748a1fb2ae8fda94844bd0545935095dbf404b32"
+            },
+            "dist": {
+                "type": "zip",
+                "url": "https://api.github.com/repos/php-stubs/wordpress-globals/zipball/748a1fb2ae8fda94844bd0545935095dbf404b32",
+                "reference": "748a1fb2ae8fda94844bd0545935095dbf404b32",
+                "shasum": ""
+            },
+            "require-dev": {
+                "php": "~7.1"
+            },
+            "suggest": {
+                "php-stubs/wordpress-stubs": "Up-to-date WordPress function and class declaration stubs",
+                "szepeviktor/phpstan-wordpress": "WordPress extensions for PHPStan"
+            },
+            "type": "library",
+            "notification-url": "https://packagist.org/downloads/",
+            "license": [
+                "MIT"
+            ],
+            "description": "Global variables and global constants from WordPress core.",
+            "homepage": "https://github.com/php-stubs/wordpress-globals",
+            "keywords": [
+                "PHPStan",
+                "constants",
+                "globals",
+                "static analysis",
+                "wordpress"
+            ],
+            "support": {
+                "issues": "https://github.com/php-stubs/wordpress-globals/issues",
+                "source": "https://github.com/php-stubs/wordpress-globals/tree/master"
+            },
+            "time": "2020-01-13T06:12:59+00:00"
+        },
         {
             "name": "php-stubs/wordpress-stubs",
             "version": "v6.6.2",
diff --git a/inc/Analytics/EnvironmentConfig.php b/inc/Analytics/EnvironmentConfig.php
new file mode 100644
index 00000000..fe52b93e
--- /dev/null
+++ b/inc/Analytics/EnvironmentConfig.php
@@ -0,0 +1,81 @@
+<?php declare(strict_types = 1);
+
+namespace RemoteDataBlocks\Analytics;
+
+defined( 'ABSPATH' ) || exit();
+
+use Automattic\VIP\Telemetry\Tracks;
+use function Automattic\VIP\Telemetry\Tracks\get_tracks_core_properties;
+
+/**
+ * Class for environment configuration.
+ *
+ * This class abstracts WordPress-specific functions for easy mocking.
+ */
+class EnvironmentConfig {
+	/**
+	 * Tracks analytics core properties.
+	 * 
+	 * This is set by the Tracks library available in MU Plugins.
+	 *
+	 * @var array<string, mixed>
+	 */
+	private array $tracks_core_props = [];
+
+	public function __construct() {
+		if ( function_exists( 'Automattic\VIP\Telemetry\Tracks\get_tracks_core_properties' ) ) {
+			add_action( 'init', function (): void {
+				/** @psalm-suppress UndefinedFunction */
+				$this->tracks_core_props = get_tracks_core_properties();
+			} );
+		}
+	}
+
+	public function is_enabled_via_filter(): bool {
+		return apply_filters( 'remote_data_blocks_enable_tracks_analytics', false ) ?? false;
+	}
+
+	public function get_tracks_lib_class(): ?string {
+		if ( ! class_exists( 'Automattic\VIP\Telemetry\Tracks' ) ) {
+			return null;
+		}
+
+		return Tracks::class;
+	}
+
+	public function is_wpvip_site(): bool {
+		if ( ! isset( $this->tracks_core_props['hosting_provider'] ) ) {
+			return false;
+		}
+
+		return 'wpvip' === $this->tracks_core_props['hosting_provider'];
+	}
+
+	public function is_local_env(): bool {
+		if ( ! isset( $this->tracks_core_props['vipgo_env'] ) ) {
+			return false;
+		}
+
+		return 'local' === $this->tracks_core_props['vipgo_env'];
+	}
+
+	public function is_remote_data_blocks_plugin( string|null $plugin_path ): bool {
+		return 'remote-data-blocks/remote-data-blocks.php' === $plugin_path;
+	}
+
+	public function should_track_post_having_remote_data_blocks( int $post_id ): bool {
+		// Ensure this is not an auto-save or revision.
+		if ( wp_is_post_revision( $post_id ) || wp_is_post_autosave( $post_id ) ) {
+			return false;
+		}
+
+		return true;
+	}
+
+	/**
+	 * Get the core properties to be sent with each event.
+	 */
+	public function get_tracks_core_properties(): array {
+		return $this->tracks_core_props;
+	}
+}
diff --git a/inc/Analytics/TracksAnalytics.php b/inc/Analytics/TracksAnalytics.php
new file mode 100644
index 00000000..f6fa05e2
--- /dev/null
+++ b/inc/Analytics/TracksAnalytics.php
@@ -0,0 +1,170 @@
+<?php declare(strict_types = 1);
+
+namespace RemoteDataBlocks\Analytics;
+
+defined( 'ABSPATH' ) || exit();
+
+use RemoteDataBlocks\Editor\BlockManagement\ConfigStore;
+
+/**
+ * Class to implement Tracks Analytics.
+ */
+class TracksAnalytics {
+	/**
+	 * The tracks instance (not using Tracks as type because it is present in MU Plugins codebase).
+	 */
+	private static object|null $instance = null;
+
+	/**
+	 * Environment configuration.
+	 */
+	private static ?EnvironmentConfig $env_config = null;
+
+	/**
+	 * Initialize Tracks Analytics based on the environment configuration.
+	 *
+	 * @param EnvironmentConfig $env_config Environment configuration.
+	 */
+	public static function init( EnvironmentConfig $env_config ): void {
+		self::$env_config = $env_config;
+
+		// Do not track on local environment.
+		if ( self::$env_config->is_local_env() ) {
+			return;
+		}
+
+		$tracks_class = self::$env_config->get_tracks_lib_class();
+		if ( ! $tracks_class ) {
+			return;
+		}
+
+		if ( self::$env_config->is_wpvip_site() || self::$env_config->is_enabled_via_filter() ) {
+			self::$instance = new $tracks_class( '', self::get_global_properties() );
+			self::setup_tracking_via_hooks();
+		}
+	}
+
+	/**
+	 * Get `Tracks` global properties to be sent with each event.
+	 */
+	public static function get_global_properties(): array {
+		$rdb_specific_props = [
+			'plugin_version' => defined( 'REMOTE_DATA_BLOCKS__PLUGIN_VERSION' ) ? constant( 'REMOTE_DATA_BLOCKS__PLUGIN_VERSION' ) : '',
+		];
+
+		return array_merge( $rdb_specific_props, self::$env_config->get_tracks_core_properties() );
+	}
+
+	private static function setup_tracking_via_hooks(): void {
+		// WordPress Dashboard Hooks.
+		add_action( 'activated_plugin', [ __CLASS__, 'track_plugin_activation' ] );
+		add_action( 'deactivated_plugin', [ __CLASS__, 'track_plugin_deactivation' ] );
+		add_action( 'save_post', [ __CLASS__, 'track_remote_data_blocks_usage' ], 10, 2 );
+	}
+
+	/**
+	 * Activation hook.
+	 *
+	 * @param string $plugin_path Path of the plugin that was activated.
+	 */
+	public static function track_plugin_activation( string $plugin_path ): void {
+		if ( ! self::$env_config->is_remote_data_blocks_plugin( $plugin_path ) ) {
+			return;
+		}
+
+		self::record_event( 'remotedatablocks_plugin_toggle', [ 'action' => 'activate' ] );
+	}
+
+	/**
+	 * Deactivation hook.
+	 *
+	 * @param string $plugin_path Path of the plugin that was deactivated.
+	 */
+	public static function track_plugin_deactivation( string $plugin_path ): void {
+		if ( ! self::$env_config->is_remote_data_blocks_plugin( $plugin_path ) ) {
+			return;
+		}
+
+		self::record_event( 'remotedatablocks_plugin_toggle', [ 'action' => 'deactivate' ] );
+	}
+
+	/**
+	 * Track usage of Remote Data Blocks.
+	 *
+	 * @param int      $post_id Post ID.
+	 * @param \WP_Post $post Post object.
+	 */
+	public static function track_remote_data_blocks_usage( int $post_id, object $post ): void {
+		if ( ! self::$env_config->should_track_post_having_remote_data_blocks( $post_id ) ) {
+			return;
+		}
+
+		$post_status = $post->post_status;
+		if ( 'publish' !== $post_status ) {
+			return;
+		}
+
+		// Regular expression to match all remote data blocks present in the post content.
+		$reg_exp = '/<!--\s{1}wp:remote-data-blocks\/([^\s]+)\s/';
+		preg_match_all( $reg_exp, $post->post_content, $matches );
+		if ( count( $matches[1] ) === 0 ) {
+			return;
+		}
+
+		// Get data source and track usage.
+		$track_props = [
+			'post_status' => $post_status,
+			'post_type' => $post->post_type,
+		];
+		foreach ( $matches[1] as $match ) {
+			$data_source_type = ConfigStore::get_data_source_type( 'remote-data-blocks/' . $match );
+			if ( ! $data_source_type ) {
+				continue;
+			}
+
+			// Calculate stats of each remote data source.
+			$key = $data_source_type . '_data_source_count';
+			$track_props[ $key ] = ( $track_props[ $key ] ?? 0 ) + 1;
+			$track_props['remote_data_blocks_total_count'] = ( $track_props['remote_data_blocks_total_count'] ?? 0 ) + 1;
+		}
+
+		self::record_event( 'remotedatablocks_blocks_usage_stats', $track_props );
+	}
+
+	/**
+	 * Record an event with Tracks.
+	 *
+	 * @param string $event_name The name of the event.
+	 * @param array  $props      The properties to send with the event.
+	 *
+	 * @return bool True if the event was recorded, false otherwise.
+	 */
+	public static function record_event( string $event_name, array $props ): bool {
+		if ( ! isset( self::$instance ) ) {
+			return false;
+		}
+
+		self::$instance->record_event( $event_name, $props );
+
+		return true;
+	}
+
+	/**
+	 * Get the instance of Tracks.
+	 */
+	public static function get_instance(): ?object {
+		return self::$instance;
+	}
+
+	public static function get_env_config(): ?EnvironmentConfig {
+		return self::$env_config;
+	}
+
+	/**
+	 * Reset class properties.
+	 */
+	public static function reset(): void {
+		self::$instance = null;
+		self::$env_config = null;
+	}
+}
diff --git a/inc/Config/DataSource/HttpDataSource.php b/inc/Config/DataSource/HttpDataSource.php
index ac83d713..b0eb25f0 100644
--- a/inc/Config/DataSource/HttpDataSource.php
+++ b/inc/Config/DataSource/HttpDataSource.php
@@ -36,10 +36,17 @@ public function get_image_url(): ?string {
 		return null;
 	}
 
+	/**
+	 * Get the service name.
+	 */
+	public function get_service(): ?string {
+		return $this->config['service'] ?? null;
+	}
+
 	public function get_slug(): string {
 		return $this->config['slug'];
 	}
-	
+
 	/**
 	 * @inheritDoc
 	 */
diff --git a/inc/Editor/BlockManagement/ConfigStore.php b/inc/Editor/BlockManagement/ConfigStore.php
index f934deb1..d2d6caa0 100644
--- a/inc/Editor/BlockManagement/ConfigStore.php
+++ b/inc/Editor/BlockManagement/ConfigStore.php
@@ -27,7 +27,6 @@ public static function init( ?LoggerInterface $logger = null ): void {
 	 * Convert a block title to a block name. Mainly this is to reduce the burden
 	 * of configuration and to ensure that block names are unique (since block
 	 * titles must be unique).
-	 *
 	 */
 	public static function get_block_name( string $block_title ): string {
 		return 'remote-data-blocks/' . sanitize_title( $block_title );
@@ -69,6 +68,37 @@ public static function is_registered_block( string $block_name ): bool {
 		return isset( self::$configurations[ $block_name ] );
 	}
 
+	/**
+	 * Get data source type from block name.
+	 *
+	 * @param string $block_name Name of the block.
+	 */
+	public static function get_data_source_type( string $block_name ): ?string {
+		$config = self::get_configuration( $block_name );
+		if ( ! $config ) {
+			return null;
+		}
+
+		$queries = $config['queries'];
+		if ( count( $queries ) === 0 ) {
+			return null;
+		}
+
+		$data_source_type = null;
+		foreach ( $queries as $query ) {
+			if ( ! $query instanceof HttpQueryContext ) {
+				continue;
+			}
+
+			$data_source_type = $query->get_data_source()->get_service();
+			if ( $data_source_type ) {
+				break;
+			}
+		}
+
+		return $data_source_type;
+	}
+
 	/**
 	 * Return an unprivileged representation of the data sources that can be
 	 * displayed in settings screens.
diff --git a/inc/REST/DataSourceController.php b/inc/REST/DataSourceController.php
index 2485e22b..d696ec5f 100644
--- a/inc/REST/DataSourceController.php
+++ b/inc/REST/DataSourceController.php
@@ -2,6 +2,7 @@
 
 namespace RemoteDataBlocks\REST;
 
+use RemoteDataBlocks\Analytics\TracksAnalytics;
 use RemoteDataBlocks\Editor\BlockManagement\ConfigStore;
 use RemoteDataBlocks\WpdbStorage\DataSourceCrud;
 use WP_REST_Controller;
@@ -105,7 +106,14 @@ public function register_routes(): void {
 	 * @return WP_REST_Response|WP_Error Response object on success, or WP_Error object on failure.
 	 */
 	public function create_item( $request ) {
-		$item = DataSourceCrud::register_new_data_source( $request->get_json_params() );
+		$data_source_properties = $request->get_json_params();
+		$item = DataSourceCrud::register_new_data_source( $data_source_properties );
+
+		TracksAnalytics::record_event( 'remotedatablocks_data_source_interaction', array_merge( [
+			'data_source_type' => $data_source_properties['service'],
+			'action' => 'add',
+		], $this->get_data_source_interaction_track_props( $data_source_properties ) ) );
+
 		return rest_ensure_response( $item );
 	}
 
@@ -120,10 +128,10 @@ public function get_items( $request ) {
 		$ui_configured_data_sources = DataSourceCrud::get_data_sources_list();
 
 		/**
-		 * quick and dirty deduplication of data sources by slug
+		 * Quick and dirty de-duplication of data sources by slug.
 		 *
-		 * ui configured data sources take precedence over code configured ones
-		 * here due to the ordering of the two arrays passed to array_reduce
+		 * UI configured data sources take precedence over code configured ones
+		 * here due to the ordering of the two arrays passed to array_reduce.
 		 *
 		 * @todo: refactor this out in the near future in favor of an upstream
 		 * single source of truth for data source configurations
@@ -136,7 +144,21 @@ function ( $acc, $item ) {
 			},
 			[]
 		));
-	
+
+		// Tracks Analytics. Only once per day to reduce noise.
+		$track_transient_key = 'remotedatablocks_view_data_sources_tracked';
+		if ( ! get_transient( $track_transient_key ) ) {
+			$code_configured_data_sources_count = count( $code_configured_data_sources );
+			$ui_configured_data_sources_count = count( $ui_configured_data_sources );
+
+			TracksAnalytics::record_event( 'remotedatablocks_view_data_sources', [
+				'total_data_sources_count' => $code_configured_data_sources_count + $ui_configured_data_sources_count,
+				'code_configured_data_sources_count' => $code_configured_data_sources_count,
+				'ui_configured_data_sources_count' => $ui_configured_data_sources_count,
+			] );
+			set_transient( $track_transient_key, true, DAY_IN_SECONDS );
+		}
+
 		return rest_ensure_response( $data_sources );
 	}
 
@@ -158,7 +180,14 @@ public function get_item( $request ) {
 	 * @return WP_REST_Response|WP_Error Response object on success, or WP_Error object on failure.
 	 */
 	public function update_item( $request ) {
-		$item = DataSourceCrud::update_item_by_uuid( $request->get_param( 'uuid' ), $request->get_json_params() );
+		$data_source_properties = $request->get_json_params();
+		$item = DataSourceCrud::update_item_by_uuid( $request->get_param( 'uuid' ), $data_source_properties );
+
+		TracksAnalytics::record_event( 'remotedatablocks_data_source_interaction', array_merge( [
+			'data_source_type' => $data_source_properties['service'],
+			'action' => 'update',
+		], $this->get_data_source_interaction_track_props( $data_source_properties ) ) );
+
 		return rest_ensure_response( $item );
 	}
 
@@ -169,7 +198,15 @@ public function update_item( $request ) {
 	 * @return WP_REST_Response|WP_Error Response object on success, or WP_Error object on failure.
 	 */
 	public function delete_item( $request ) {
+		$data_source_properties = $request->get_json_params();
 		$result = DataSourceCrud::delete_item_by_uuid( $request->get_param( 'uuid' ) );
+
+		// Tracks Analytics.
+		TracksAnalytics::record_event( 'remotedatablocks_data_source_interaction', [
+			'data_source_type' => $data_source_properties['service'],
+			'action' => 'delete',
+		] );
+
 		return rest_ensure_response( $result );
 	}
 
@@ -215,4 +252,16 @@ public function delete_item_permissions_check( $request ) {
 	public function item_slug_conflicts_permissions_check() {
 		return current_user_can( 'manage_options' );
 	}
+
+	private function get_data_source_interaction_track_props( $data_source_properties ): array {
+		$props = [];
+
+		if ( 'generic-http' === $data_source_properties['service'] ) {
+			$auth = $data_source_properties['auth'];
+			$props['authentication_type'] = $auth['type'] ?? '';
+			$props['api_key_location'] = $auth['addTo'] ?? '';
+		}
+
+		return $props;
+	}
 }
diff --git a/psalm.xml b/psalm.xml
index 231f1867..32058bee 100644
--- a/psalm.xml
+++ b/psalm.xml
@@ -30,7 +30,9 @@
         </UnusedClass>
     </issueHandlers>
     <stubs>
+        <file name="vendor/php-stubs/wordpress-globals/wordpress-globals.php"/>
         <file name="vendor/php-stubs/wordpress-stubs/wordpress-stubs.php"/>
+        <file name="tests/inc/test-utils.php"/>
         <file name="remote-data-blocks.php"/>
     </stubs>
 </psalm>
diff --git a/remote-data-blocks.php b/remote-data-blocks.php
index afd79d83..12d82814 100644
--- a/remote-data-blocks.php
+++ b/remote-data-blocks.php
@@ -34,6 +34,9 @@
 Editor\BlockManagement\ConfigRegistry::init();
 Editor\PatternEditor\PatternEditor::init();
 
+// Analytics
+Analytics\TracksAnalytics::init( new Analytics\EnvironmentConfig() );
+
 // Example API
 ExampleApi\ExampleApi::init();
 
diff --git a/src/data-sources/hooks/useDataSources.ts b/src/data-sources/hooks/useDataSources.ts
index 6eeef9e5..919e3b5f 100644
--- a/src/data-sources/hooks/useDataSources.ts
+++ b/src/data-sources/hooks/useDataSources.ts
@@ -49,14 +49,14 @@ export const useDataSources = ( loadOnMount = true ) => {
 		[ setLoadingSlugConflicts, setSlugConflicts, createErrorNotice ]
 	);
 
-	async function updateDataSource( source: DataSourceConfig ) {
+	async function updateDataSource( sourceConfig: DataSourceConfig ) {
 		let result: DataSourceConfig;
 
 		try {
 			result = await apiFetch( {
-				path: `${ REST_BASE_DATA_SOURCES }/${ source.uuid }`,
+				path: `${ REST_BASE_DATA_SOURCES }/${ sourceConfig.uuid }`,
 				method: 'PUT',
-				data: source,
+				data: sourceConfig,
 			} );
 		} catch ( error ) {
 			showSnackbar( 'error', __( 'Failed to update data source.', 'remote-data-blocks' ) );
@@ -65,7 +65,10 @@ export const useDataSources = ( loadOnMount = true ) => {
 
 		showSnackbar(
 			'success',
-			__( sprintf( '"%s" has been successfully updated.', 'remote-data-blocks' ), source.slug )
+			__(
+				sprintf( '"%s" has been successfully updated.', 'remote-data-blocks' ),
+				sourceConfig.slug
+			)
 		);
 		return result;
 	}
@@ -96,6 +99,7 @@ export const useDataSources = ( loadOnMount = true ) => {
 			await apiFetch( {
 				path: `${ REST_BASE_DATA_SOURCES }/${ source.uuid }`,
 				method: 'DELETE',
+				data: source,
 			} );
 		} catch ( error ) {
 			showSnackbar( 'error', __( 'Failed to delete data source.', 'remote-data-blocks' ) );
diff --git a/tests/inc/Analytics/EnvironmentConfigTest.php b/tests/inc/Analytics/EnvironmentConfigTest.php
new file mode 100644
index 00000000..fc1e2833
--- /dev/null
+++ b/tests/inc/Analytics/EnvironmentConfigTest.php
@@ -0,0 +1,77 @@
+<?php declare(strict_types = 1);
+
+namespace RemoteDataBlocks\Tests\Analytics;
+
+use PHPUnit\Framework\TestCase;
+use RemoteDataBlocks\Analytics\EnvironmentConfig;
+
+class EnvironmentConfigTest extends TestCase {
+	public function setUp(): void {
+		$GLOBALS['__wordpress_filters'] = [];
+	}
+
+	public function testIsEnabledViaFilterReturnsFalse(): void {
+		$obj = new EnvironmentConfig();
+
+		$this->assertEquals( false, $obj->is_enabled_via_filter() );
+	}
+
+	public function testIsEnabledViaFilterReturnsTrue(): void {
+		$GLOBALS['__wordpress_filters']['remote_data_blocks_enable_tracks_analytics'] = true;
+		$obj = new EnvironmentConfig();
+
+		$this->assertEquals( true, $obj->is_enabled_via_filter() );
+	}
+
+	public function testIsWpvipSiteReturnsFalse(): void {
+		$obj = new EnvironmentConfig();
+
+		$this->assertEquals( false, $obj->is_wpvip_site() );
+	}
+
+	public function testIsWpvipSiteReturnsTrue(): void {
+		$obj = new EnvironmentConfig();
+		set_private_property( EnvironmentConfig::class, $obj, 'tracks_core_props', [ 'hosting_provider' => 'wpvip' ] );
+
+		$this->assertEquals( true, $obj->is_wpvip_site() );
+	}
+
+	public function testIsLocalEnvReturnsFalse(): void {
+		$obj = new EnvironmentConfig();
+
+		$this->assertEquals( false, $obj->is_local_env() );
+	}
+
+	public function testIsLocalEnvReturnsTrue(): void {
+		$obj = new EnvironmentConfig();
+		set_private_property( EnvironmentConfig::class, $obj, 'tracks_core_props', [ 'vipgo_env' => 'local' ] );
+
+		$this->assertEquals( true, $obj->is_local_env() );
+	}
+
+	public function testIsRemoteDataBlocksPluginReturnsFalse(): void {
+		$obj = new EnvironmentConfig();
+
+		$this->assertEquals( false, $obj->is_remote_data_blocks_plugin( '' ) );
+	}
+
+	public function testIsRemoteDataBlocksPluginReturnsTrue(): void {
+		$obj = new EnvironmentConfig();
+
+		$this->assertEquals( true, $obj->is_remote_data_blocks_plugin( 'remote-data-blocks/remote-data-blocks.php' ) );
+	}
+
+	public function testGetTracksCoreProperties(): void {
+		$obj = new EnvironmentConfig();
+		$this->assertEquals( [], $obj->get_tracks_core_properties() );
+
+		set_private_property( EnvironmentConfig::class, $obj, 'tracks_core_props', [
+			'hosting_provider' => 'wpvip',
+			'vipgo_env' => 'local',
+		] );
+		$this->assertEquals( [
+			'hosting_provider' => 'wpvip',
+			'vipgo_env' => 'local',
+		], $obj->get_tracks_core_properties() );
+	}
+}
diff --git a/tests/inc/Analytics/TracksAnalyticsTest.php b/tests/inc/Analytics/TracksAnalyticsTest.php
new file mode 100644
index 00000000..b7560a64
--- /dev/null
+++ b/tests/inc/Analytics/TracksAnalyticsTest.php
@@ -0,0 +1,277 @@
+<?php declare(strict_types = 1);
+
+namespace RemoteDataBlocks\Tests\Analytics;
+
+use Mockery;
+use Mockery\MockInterface;
+use PHPUnit\Framework\TestCase;
+use PHPUnit\Framework\MockObject\MockObject;
+use RemoteDataBlocks\Analytics\TracksAnalytics;
+use RemoteDataBlocks\Analytics\EnvironmentConfig;
+use RemoteDataBlocks\Config\QueryContext\HttpQueryContext;
+use RemoteDataBlocks\Editor\BlockManagement\ConfigStore;
+use RemoteDataBlocks\ExampleApi\Queries\ExampleApiDataSource;
+use RemoteDataBlocks\Integrations\Shopify\ShopifyDataSource;
+
+// Define a mock class for Tracks.
+if ( ! class_exists( 'Automattic\VIP\Telemetry\Tracks' ) ) {
+	class MockTracks {
+		/**
+		 * @psalm-suppress PossiblyUnusedParam
+		 */
+		public function record_event( $_name, $_props ) {}
+	}
+}
+
+class TracksAnalyticsTest extends TestCase {
+	public function setUp(): void {
+		$GLOBALS['__wordpress_actions'] = [];
+	}
+
+	public function tearDown(): void {
+		Mockery::close();
+		TracksAnalytics::reset();
+	}
+
+	public function testInitDoesNotSetTracksIfLibraryIsNotPresent(): void {
+		TracksAnalytics::init( new EnvironmentConfig() );
+
+		$this->assertEquals( null, TracksAnalytics::get_instance() );
+	}
+
+	public function testInitDoesNotSetTracksOnLocalEnvironment(): void {
+		/** @var MockObject|EnvironmentConfig */
+		$env_config_mock = $this->getMockBuilder( EnvironmentConfig::class )->onlyMethods( [ 'get_tracks_lib_class', 'is_local_env', 'is_enabled_via_filter' ] )->getMock();
+		$env_config_mock->method( 'get_tracks_lib_class' )->with()->willReturn( MockTracks::class );
+		$env_config_mock->method( 'is_local_env' )->with()->willReturn( true );
+		$env_config_mock->method( 'is_enabled_via_filter' )->with()->willReturn( true );
+
+		/** @var TracksAnalytics|MockInterface */
+		Mockery::mock( TracksAnalytics::class )->makePartial();
+		TracksAnalytics::init( $env_config_mock );
+
+		$this->assertEquals( null, TracksAnalytics::get_instance() );
+	}
+
+	public function testInitDoesSetTracksIfTrackingIsEnabledViaFilter(): void {
+		/** @var MockObject|EnvironmentConfig */
+		$env_config_mock = $this->getMockBuilder( EnvironmentConfig::class )->onlyMethods( [ 'is_enabled_via_filter', 'get_tracks_lib_class' ] )->getMock();
+		$env_config_mock->method( 'is_enabled_via_filter' )->with()->willReturn( true );
+		$env_config_mock->method( 'get_tracks_lib_class' )->with()->willReturn( MockTracks::class );
+
+		/** @var TracksAnalytics|MockInterface */
+		Mockery::mock( TracksAnalytics::class )->makePartial();
+		TracksAnalytics::init( $env_config_mock );
+
+		$this->assertInstanceOf( MockTracks::class, TracksAnalytics::get_instance() );
+		$this->assertEquals( [ 'activated_plugin', 'deactivated_plugin', 'save_post' ], array_keys( $GLOBALS['__wordpress_actions'] ) );
+	}
+
+	public function testInitDoesSetTracksIfTrackingIsEnabledOnVipSite(): void {
+		/** @var MockObject|EnvironmentConfig */
+		$env_config_mock = $this->getMockBuilder( EnvironmentConfig::class )->onlyMethods( [ 'is_wpvip_site', 'get_tracks_lib_class' ] )->getMock();
+		$env_config_mock->method( 'is_wpvip_site' )->with()->willReturn( true );
+		$env_config_mock->method( 'get_tracks_lib_class' )->with()->willReturn( MockTracks::class );
+
+		/** @var TracksAnalytics|MockInterface */
+		Mockery::mock( TracksAnalytics::class )->makePartial();
+		TracksAnalytics::init( $env_config_mock );
+
+		$this->assertInstanceOf( MockTracks::class, TracksAnalytics::get_instance() );
+		$this->assertEquals( [ 'activated_plugin', 'deactivated_plugin', 'save_post' ], array_keys( $GLOBALS['__wordpress_actions'] ) );
+	}
+
+	public function testGetGlobalProperties(): void {
+		/** @var MockObject|EnvironmentConfig */
+		$env_config_mock = $this->getMockBuilder( EnvironmentConfig::class )->onlyMethods( [ 'get_tracks_core_properties' ] )->getMock();
+		$env_config_mock->expects( $this->once() )->method( 'get_tracks_core_properties' )->with()->willReturn( [ 'vipgo_env' => '123' ] );
+
+		TracksAnalytics::init( $env_config_mock );
+
+		$this->assertEquals( [
+			'plugin_version' => '',
+			'vipgo_env' => '123',
+		], TracksAnalytics::get_global_properties() );
+	}
+
+	public function testTrackPluginActivationDoesNotRecordEventIfPluginIsNotRDB(): void {
+		/** @var MockObject|EnvironmentConfig */
+		$env_config_mock = $this->getMockBuilder( EnvironmentConfig::class )->onlyMethods( [ 'is_remote_data_blocks_plugin' ] )->getMock();
+		$env_config_mock->method( 'is_remote_data_blocks_plugin' )->with()->willReturn( false );
+
+		/** @var MockTracks|MockObject */
+		$tracks_mock = $this->getMockBuilder( MockTracks::class )->onlyMethods( [ 'record_event' ] )->getMock();
+		$tracks_mock->expects( $this->exactly( 0 ) )->method( 'record_event' );
+
+		set_private_property( TracksAnalytics::class, null, 'instance', $tracks_mock );
+		TracksAnalytics::init( $env_config_mock );
+		TracksAnalytics::track_plugin_activation( 'plugin_path' );
+	}
+
+	public function testTrackPluginActivationDoesRecordEventIfPluginIsRDB(): void {
+		/** @var MockObject|EnvironmentConfig */
+		$env_config_mock = $this->getMockBuilder( EnvironmentConfig::class )->onlyMethods( [ 'is_remote_data_blocks_plugin' ] )->getMock();
+		$env_config_mock->method( 'is_remote_data_blocks_plugin' )->with()->willReturn( true );
+
+		/** @var MockTracks|MockObject */
+		$tracks_mock = $this->getMockBuilder( MockTracks::class )->onlyMethods( [ 'record_event' ] )->getMock();
+		$tracks_mock->expects( $this->exactly( 1 ) )->method( 'record_event' )->with( 'remotedatablocks_plugin_toggle', $this->isType( 'array' ) );
+
+		set_private_property( TracksAnalytics::class, null, 'instance', $tracks_mock );
+		TracksAnalytics::init( $env_config_mock );
+		TracksAnalytics::track_plugin_activation( 'plugin_path' );
+	}
+
+	public function testTrackPluginDeactivationDoesNotRecordEventIfPluginIsNotRDB(): void {
+		/** @var MockObject|EnvironmentConfig */
+		$env_config_mock = $this->getMockBuilder( EnvironmentConfig::class )->onlyMethods( [ 'is_remote_data_blocks_plugin' ] )->getMock();
+		$env_config_mock->method( 'is_remote_data_blocks_plugin' )->with()->willReturn( false );
+
+		/** @var MockTracks|MockObject */
+		$tracks_mock = $this->getMockBuilder( MockTracks::class )->onlyMethods( [ 'record_event' ] )->getMock();
+		$tracks_mock->expects( $this->exactly( 0 ) )->method( 'record_event' );
+
+		set_private_property( TracksAnalytics::class, null, 'instance', $tracks_mock );
+		TracksAnalytics::init( $env_config_mock );
+		TracksAnalytics::track_plugin_deactivation( 'plugin_path' );
+	}
+
+	public function testTrackPluginDeactivationDoesRecordEventIfPluginIsRDB(): void {
+		/** @var MockObject|EnvironmentConfig */
+		$env_config_mock = $this->getMockBuilder( EnvironmentConfig::class )->onlyMethods( [ 'is_remote_data_blocks_plugin' ] )->getMock();
+		$env_config_mock->method( 'is_remote_data_blocks_plugin' )->with()->willReturn( true );
+	
+		/** @var MockTracks|MockObject */
+		$tracks_mock = $this->getMockBuilder( MockTracks::class )->onlyMethods( [ 'record_event' ] )->getMock();
+		$tracks_mock->expects( $this->exactly( 1 ) )->method( 'record_event' )->with( 'remotedatablocks_plugin_toggle', $this->isType( 'array' ) );
+
+		set_private_property( TracksAnalytics::class, null, 'instance', $tracks_mock );
+		TracksAnalytics::init( $env_config_mock );
+		TracksAnalytics::track_plugin_deactivation( 'plugin_path' );
+	}
+
+	public function testTrackRemoteDataBlocksUsageDoesNotTrackEventIfUsageShouldNotBeTracked(): void {
+		/** @var MockObject|EnvironmentConfig */
+		$env_config_mock = $this->getMockBuilder( EnvironmentConfig::class )->onlyMethods( [ 'should_track_post_having_remote_data_blocks' ] )->getMock();
+		$env_config_mock->method( 'should_track_post_having_remote_data_blocks' )->with( 1 )->willReturn( false );
+
+		/** @var MockTracks|MockObject */
+		$tracks_mock = $this->getMockBuilder( MockTracks::class )->onlyMethods( [ 'record_event' ] )->getMock();
+		$tracks_mock->expects( $this->exactly( 0 ) )->method( 'record_event' );
+
+		set_private_property( TracksAnalytics::class, null, 'instance', $tracks_mock );
+		TracksAnalytics::init( $env_config_mock );
+		TracksAnalytics::track_remote_data_blocks_usage( 1, (object) [] );
+	}
+
+	public function testTrackRemoteDataBlocksUsageDoesNotTrackEventIfPostStatusIsNotPublish(): void {
+			/** @var MockObject|EnvironmentConfig */
+			$env_config_mock = $this->getMockBuilder( EnvironmentConfig::class )->onlyMethods( [ 'should_track_post_having_remote_data_blocks' ] )->getMock();
+			$env_config_mock->method( 'should_track_post_having_remote_data_blocks' )->with( 1 )->willReturn( true );
+	
+			/** @var MockTracks|MockObject */
+			$tracks_mock = $this->getMockBuilder( MockTracks::class )->onlyMethods( [ 'record_event' ] )->getMock();
+			$tracks_mock->expects( $this->exactly( 0 ) )->method( 'record_event' );
+	
+			set_private_property( TracksAnalytics::class, null, 'instance', $tracks_mock );
+			TracksAnalytics::init( $env_config_mock );
+			TracksAnalytics::track_remote_data_blocks_usage( 1, (object) [ 'post_status' => 'draft' ] );
+	}
+
+	public function testTrackRemoteDataBlocksUsageDoesNotTrackEventIfPostContentHaveNoRemoteBlocks(): void {
+		/** @var MockObject|EnvironmentConfig */
+		$env_config_mock = $this->getMockBuilder( EnvironmentConfig::class )->onlyMethods( [ 'should_track_post_having_remote_data_blocks' ] )->getMock();
+		$env_config_mock->method( 'should_track_post_having_remote_data_blocks' )->with( 1 )->willReturn( true );
+
+		// Setup data sources.
+		ConfigStore::init();
+		ConfigStore::set_configuration( 'remote-data-blocks/shopify-vip-store', [
+			'queries' => [ new HttpQueryContext( ShopifyDataSource::create( 'access_token', 'name' ) ) ],
+		] );
+
+		/** @var MockTracks|MockObject */
+		$tracks_mock = $this->getMockBuilder( MockTracks::class )->onlyMethods( [ 'record_event' ] )->getMock();
+		$tracks_mock->expects( $this->exactly( 0 ) )->method( 'record_event' );
+
+		set_private_property( TracksAnalytics::class, null, 'instance', $tracks_mock );
+		TracksAnalytics::init( $env_config_mock );
+		TracksAnalytics::track_remote_data_blocks_usage( 1, (object) [
+			'post_type' => 'post',
+			'post_status' => 'publish',
+			'post_content' => '<p>No remote data blocks</p>',
+		] );
+	}
+
+	public function testTrackRemoteDataBlocksUsageDoesTrackEventIfPostContentHaveRemoteBlocks(): void {
+		/** @var MockObject|EnvironmentConfig */
+		$env_config_mock = $this->getMockBuilder( EnvironmentConfig::class )->onlyMethods( [ 'should_track_post_having_remote_data_blocks' ] )->getMock();
+		$env_config_mock->method( 'should_track_post_having_remote_data_blocks' )->with( 1 )->willReturn( true );
+
+		// Setup data sources.
+		ConfigStore::init();
+		ConfigStore::set_configuration( 'remote-data-blocks/shopify-vip-store', [
+			'queries' => [ new HttpQueryContext( ShopifyDataSource::create( 'access_token', 'name' ) ) ],
+		] );
+		ConfigStore::set_configuration( 'remote-data-blocks/conference-event', [
+			'queries' => [
+				new HttpQueryContext( ExampleApiDataSource::from_array( [
+					'slug' => 'example-api',
+					'service' => 'example_api',
+				] ) ),
+			],
+		] );
+
+		/** @var MockTracks|MockObject */
+		$tracks_mock = $this->getMockBuilder( MockTracks::class )->onlyMethods( [ 'record_event' ] )->getMock();
+		$tracks_mock->expects( $this->exactly( 1 ) )->method( 'record_event' )->with( 'remotedatablocks_blocks_usage_stats', [
+			'post_status' => 'publish',
+			'post_type' => 'post',
+			'shopify_data_source_count' => 2,
+			'remote_data_blocks_total_count' => 3,
+			'example_api_data_source_count' => 1,
+		] );
+
+		set_private_property( TracksAnalytics::class, null, 'instance', $tracks_mock );
+		TracksAnalytics::init( $env_config_mock );
+		TracksAnalytics::track_remote_data_blocks_usage( 1, (object) [
+			'post_type' => 'post',
+			'post_status' => 'publish',
+			'post_content' => '<!-- wp:remote-data-blocks/shopify-vip-store {"remoteData":{"blockName":"remote-data-blocks/shopify-vip-store","isCollection":false,"metadata":{"last_updated":{"name":"Last updated","type":"string","value":"2024-10-07 14:43:51"},"total_count":{"name":"Total count","type":"number","value":1}},"queryInput":{"id":"gid://shopify/Product/8642689958112"},"resultId":"d9583b2d-b79f-4af7-8adc-a723c53d7f67","results":[{"description":"Our floating shelf is the perfect way to store all your Tonal accessories. Its sleek, versatile design makes this an easy fit with any style room. Available in Coffee Oak (seen here), as well as Matte Black and Light Aged Ash.\nMade in the U.S.","title":"Tonal Accessories Shelf (Coffee Oak)","image_url":"https://cdn.shopify.com/s/files/1/0680/3456/0224/files/Coffee-Oak-with-products.webp?v=1721748682","image_alt_text":"","price":"$272.99","variant_id":"gid://shopify/ProductVariant/46081522368736"}]}} --> <div class="wp-block-remote-data-blocks-shopify-vip-store rdb-container"><!-- wp:group {"metadata":{"categories":["Remote Data Blocks"],"patternName":"remote-data-blocks/shopify-vip-store/pattern","name":"Shopify (vip-store) Data"},"layout":{"type":"constrained"}} --> <div class="wp-block-group"><!-- wp:columns --> <div class="wp-block-columns"><!-- wp:column {"width":"33%"} --> <div class="wp-block-column" style="flex-basis:33%"><!-- wp:image {"metadata":{"bindings":{"alt":{"source":"remote-data/binding","args":{"block":"remote-data-blocks/shopify-vip-store","field":"image_alt_text"}},"url":{"source":"remote-data/binding","args":{"block":"remote-data-blocks/shopify-vip-store","field":"image_url"}}},"name":"Image URL"}} --> <figure class="wp-block-image"><img src="https://cdn.shopify.com/s/files/1/0680/3456/0224/files/Coffee-Oak-with-products.webp?v=1721748682" alt=""/></figure> <!-- /wp:image --></div> <!-- /wp:column --> <!-- wp:column --> <div class="wp-block-column"><!-- wp:heading {"metadata":{"bindings":{"content":{"source":"remote-data/binding","args":{"block":"remote-data-blocks/shopify-vip-store","field":"title"}}},"name":"Title"}} --> <h2 class="wp-block-heading">Tonal Accessories Shelf (Coffee Oak)</h2> <!-- /wp:heading --> <!-- wp:paragraph {"metadata":{"bindings":{"content":{"source":"remote-data/binding","args":{"block":"remote-data-blocks/shopify-vip-store","field":"description"}}},"name":"Product description"}} --> <p>Our floating shelf is the perfect way to store all your Tonal accessories. Its sleek, versatile design makes this an easy fit with any style room. Available in Coffee Oak (seen here), as well as Matte Black and Light Aged Ash. Made in the U.S.</p> <!-- /wp:paragraph --> <!-- wp:paragraph {"metadata":{"bindings":{"content":{"source":"remote-data/binding","args":{"block":"remote-data-blocks/shopify-vip-store","field":"price"}}},"name":"Item price"}} --> <p>$272.99</p> <!-- /wp:paragraph --></div> <!-- /wp:column --></div> <!-- /wp:columns --></div> <!-- /wp:group --></div> <!-- /wp:remote-data-blocks/shopify-vip-store --> <!-- wp:remote-data-blocks/shopify-vip-store {"remoteData":{"blockName":"remote-data-blocks/shopify-vip-store","isCollection":false,"metadata":{"last_updated":{"name":"Last updated","type":"string","value":"2024-10-07 14:44:01"},"total_count":{"name":"Total count","type":"number","value":1}},"queryInput":{"id":"gid://shopify/Product/8642627207392"},"resultId":"ea7d7dae-4888-42c5-b64c-2b69ccbb958e","results":[{"description":"No detail is too small. Tonal’s proprietary T-Locks let you swap out Tonal accessories with a quick push and twist to lock everything in place.","title":"T-Locks (Pack of 4)","image_url":"https://cdn.shopify.com/s/files/1/0680/3456/0224/files/T-Locks.webp?v=1721746444","image_alt_text":"","price":"$42.99","variant_id":"gid://shopify/ProductVariant/46081259307232"}]}} --> <div class="wp-block-remote-data-blocks-shopify-vip-store rdb-container"><!-- wp:group {"metadata":{"categories":["Remote Data Blocks"],"patternName":"remote-data-blocks/shopify-vip-store/pattern","name":"Shopify (vip-store) Data"},"layout":{"type":"constrained"}} --> <div class="wp-block-group"><!-- wp:columns --> <div class="wp-block-columns"><!-- wp:column {"width":"33%"} --> <div class="wp-block-column" style="flex-basis:33%"><!-- wp:image {"metadata":{"bindings":{"alt":{"source":"remote-data/binding","args":{"block":"remote-data-blocks/shopify-vip-store","field":"image_alt_text"}},"url":{"source":"remote-data/binding","args":{"block":"remote-data-blocks/shopify-vip-store","field":"image_url"}}},"name":"Image URL"}} --> <figure class="wp-block-image"><img src="https://cdn.shopify.com/s/files/1/0680/3456/0224/files/T-Locks.webp?v=1721746444" alt=""/></figure> <!-- /wp:image --></div> <!-- /wp:column --> <!-- wp:column --> <div class="wp-block-column"><!-- wp:heading {"metadata":{"bindings":{"content":{"source":"remote-data/binding","args":{"block":"remote-data-blocks/shopify-vip-store","field":"title"}}},"name":"Title"}} --> <h2 class="wp-block-heading">T-Locks (Pack of 4)</h2> <!-- /wp:heading --> <!-- wp:paragraph {"metadata":{"bindings":{"content":{"source":"remote-data/binding","args":{"block":"remote-data-blocks/shopify-vip-store","field":"description"}}},"name":"Product description"}} --> <p>No detail is too small. Tonal’s proprietary T-Locks let you swap out Tonal accessories with a quick push and twist to lock everything in place.</p> <!-- /wp:paragraph --> <!-- wp:paragraph {"metadata":{"bindings":{"content":{"source":"remote-data/binding","args":{"block":"remote-data-blocks/shopify-vip-store","field":"price"}}},"name":"Item price"}} --> <p>$42.99</p> <!-- /wp:paragraph --></div> <!-- /wp:column --></div> <!-- /wp:columns --></div> <!-- /wp:group --></div> <!-- /wp:remote-data-blocks/shopify-vip-store --> <!-- wp:remote-data-blocks/conference-event {"remoteData":{"blockName":"remote-data-blocks/conference-event","isCollection":false,"metadata":{"last_updated":{"name":"Last updated","type":"string","value":"2024-10-07 14:44:10"},"total_count":{"name":"Total count","type":"number","value":1}},"queryInput":{"record_id":"rec1eYD2JFtevz39u"},"resultId":"268c2a0f-2322-4197-b915-e3cede1d1c83","results":[{"id":"rec1eYD2JFtevz39u","title":"Break","location":"Pearl room","type":"Panel"}]}} --> <div class="wp-block-remote-data-blocks-conference-event rdb-container"><!-- wp:heading {"metadata":{"bindings":{"content":{"source":"remote-data/binding","args":{"block":"remote-data-blocks/conference-event","field":"title"}}},"name":"Title"}} --> <h2 class="wp-block-heading">Break</h2> <!-- /wp:heading --> <!-- wp:paragraph {"metadata":{"bindings":{"content":{"source":"remote-data/binding","args":{"block":"remote-data-blocks/conference-event","field":"location"}}},"name":"Location"}} --> <p>Pearl room</p> <!-- /wp:paragraph --> <!-- wp:paragraph {"metadata":{"bindings":{"content":{"source":"remote-data/binding","args":{"block":"remote-data-blocks/conference-event","field":"type"}}},"name":"Type"}} --> <p>Panel</p> <!-- /wp:paragraph --></div> <!-- /wp:remote-data-blocks/conference-event -->',
+		] );
+	}
+
+	public function testRecordEventDoesNothingIfInstanceIsNotSet(): void {
+		/** @var TracksAnalytics|MockObject */
+		$obj = new TracksAnalytics();
+
+		$result = $obj->record_event( 'name', [] );
+
+		$this->assertEquals( false, $result );
+	}
+
+	public function testRecordEventTracksTheEventIfInstanceIsSet(): void {
+		$mock_tracks = $this->getMockBuilder( MockTracks::class )->onlyMethods( [ 'record_event' ] )->getMock();
+		$mock_tracks->expects( $this->exactly( 1 ) )->method( 'record_event' )->with( 'event_name', [ 'event_props' ] );
+		/** @var TracksAnalytics|MockObject */
+		$obj = new TracksAnalytics();
+		set_private_property( TracksAnalytics::class, $obj, 'instance', $mock_tracks );
+
+		$result = $obj->record_event( 'event_name', [ 'event_props' ] );
+
+		$this->assertEquals( true, $result );
+	}
+
+	public function testResetMethod(): void {
+		$obj = new TracksAnalytics();
+		set_private_property( TracksAnalytics::class, $obj, 'instance', new MockTracks() );
+		TracksAnalytics::init( new EnvironmentConfig() );
+
+		$this->assertInstanceOf( MockTracks::class, TracksAnalytics::get_instance() );
+		$this->assertInstanceOf( EnvironmentConfig::class, TracksAnalytics::get_env_config() );
+
+		TracksAnalytics::reset();
+
+		$this->assertEquals( null, TracksAnalytics::get_instance() );
+		$this->assertEquals( null, TracksAnalytics::get_env_config() );
+	}
+}
diff --git a/tests/inc/Config/HttpDataSourceTest.php b/tests/inc/Config/HttpDataSourceTest.php
new file mode 100644
index 00000000..bf51c0a4
--- /dev/null
+++ b/tests/inc/Config/HttpDataSourceTest.php
@@ -0,0 +1,23 @@
+<?php declare(strict_types = 1);
+
+namespace RemoteDataBlocks\Tests\Config;
+
+use PHPUnit\Framework\TestCase;
+use RemoteDataBlocks\Tests\Mocks\MockDataSource;
+use RemoteDataBlocks\Tests\Mocks\MockValidator;
+
+class HttpDataSourceTest extends TestCase {
+	private MockDataSource $http_data_source;
+
+	public function testGetServiceMethodReturnsNull(): void {
+		$this->http_data_source = MockDataSource::from_array( [], new MockValidator() );
+
+		$this->assertNull( $this->http_data_source->get_service() );
+	}
+
+	public function testGetServiceMethodReturnsCorrectValue(): void {
+		$this->http_data_source = MockDataSource::from_array( MockDataSource::MOCK_CONFIG, new MockValidator() );
+
+		$this->assertEquals( 'mock', $this->http_data_source->get_service() );
+	}
+}
diff --git a/tests/inc/Editor/ConfigStoreTest.php b/tests/inc/Editor/ConfigStoreTest.php
new file mode 100644
index 00000000..76a7e739
--- /dev/null
+++ b/tests/inc/Editor/ConfigStoreTest.php
@@ -0,0 +1,32 @@
+<?php declare(strict_types = 1);
+
+namespace RemoteDataBlocks\Tests\Editor\BlockPatterns;
+
+use PHPUnit\Framework\TestCase;
+use RemoteDataBlocks\Config\QueryContext\HttpQueryContext;
+use RemoteDataBlocks\Editor\BlockManagement\ConfigStore;
+use RemoteDataBlocks\Integrations\Airtable\AirtableDataSource;
+
+class ConfigStoreTest extends TestCase {
+	public function testGetDataSourceReturnsNullIfConfigIsNotFound(): void {
+		ConfigStore::init();
+
+		$this->assertNull( ConfigStore::get_data_source_type( 'block_name' ) );
+	}
+
+	public function testGetDataSourceReturnsNullIfThereAreNoQueries(): void {
+		ConfigStore::init();
+		ConfigStore::set_configuration( 'block_name', [ 'queries' => [] ] );
+
+		$this->assertNull( ConfigStore::get_data_source_type( 'block_name' ) );
+	}
+
+	public function testGetDataSourceReturnsDataSource(): void {
+		ConfigStore::init();
+		ConfigStore::set_configuration( 'airtable_remote_blocks', [
+			'queries' => [ new HttpQueryContext( AirtableDataSource::create( 'access_token', 'base_id', [], 'Name' ) ) ],
+		] );
+
+		$this->assertEquals( 'airtable', ConfigStore::get_data_source_type( 'airtable_remote_blocks' ) );
+	}
+}
diff --git a/tests/inc/bootstrap.php b/tests/inc/bootstrap.php
index b6fdaa02..967645bd 100644
--- a/tests/inc/bootstrap.php
+++ b/tests/inc/bootstrap.php
@@ -10,3 +10,4 @@
 require_once __DIR__ . '/../../inc/Integrations/constants.php';
 require_once __DIR__ . '/../../functions.php';
 require_once __DIR__ . '/stubs.php';
+require_once __DIR__ . '/test-utils.php';
diff --git a/tests/inc/stubs.php b/tests/inc/stubs.php
index e0f08c43..64ae441e 100644
--- a/tests/inc/stubs.php
+++ b/tests/inc/stubs.php
@@ -1,11 +1,17 @@
 <?php declare(strict_types = 1);
 
-function apply_filters( string $_filter, mixed $thing ): mixed {
-	return $thing;
+$GLOBALS['__wordpress_filters'] = [];
+function apply_filters( string $filter, mixed $thing ): mixed {
+	return $GLOBALS['__wordpress_filters'][ $filter ] ?? $thing;
 }
 
-function add_action( string $_action, mixed ...$_args ): void {}
-function add_filter( string $_filter, mixed ...$_args ): void {}
+$GLOBALS['__wordpress_actions'] = [];
+function add_action( string $action, mixed ...$args ): void {
+	$GLOBALS['__wordpress_actions'][ $action ][] = $args;
+}
+function add_filter( string $filter, mixed ...$args ): void {
+	$GLOBALS['__wordpress_filters'][ $filter ][] = $args;
+}
 
 $GLOBALS['__wordpress_done_actions'] = [];
 function do_action( string $action, mixed ...$args ): void {
@@ -21,6 +27,14 @@ function register_block_pattern( string $_name, array $_options ): void {
 	// Do nothing
 }
 
+function is_multisite(): void {
+	// Do nothing
+}
+
+function get_bloginfo( $_property ): void {
+	// Do nothing
+}
+
 function sanitize_title( string $title ): string {
 	return str_replace( ' ', '-', strtolower( $title ) );
 }
diff --git a/tests/inc/test-utils.php b/tests/inc/test-utils.php
new file mode 100644
index 00000000..3d3f1bbf
--- /dev/null
+++ b/tests/inc/test-utils.php
@@ -0,0 +1,61 @@
+<?php declare(strict_types = 1);
+
+/**
+ * Gets private property of a class.
+ *
+ * @param class-string $class_name Name of the class.
+ * @param string       $property_name Name of the property.
+ * @return ReflectionProperty
+ */
+function get_private_property( string $class_name, string $property_name ): ReflectionProperty {
+	$reflector = new ReflectionClass( $class_name );
+	$property = $reflector->getProperty( $property_name );
+
+	/**
+	 * @psalm-suppress UnusedMethodCall
+	 */
+	$property->setAccessible( true );
+
+	return $property;
+}
+
+/**
+ * Overrides the value of a private property on a given object. This is
+ * useful when mocking the internals of a class.
+ *
+ * Note that the property will no longer be private after setAccessible is
+ * called.
+ *
+ * @param class-string $class_name The fully qualified class name, including namespace.
+ * @param object       $object_instance The object instance on which to set the value.
+ * @param string       $property_name The name of the private property to override.
+ * @param mixed        $value The value to set.
+ */
+function set_private_property(
+	string $class_name,
+	$object_instance,
+	string $property_name,
+	$value
+): void {
+	$property = get_private_property( $class_name, $property_name );
+	$property->setValue( $object_instance, $value );
+}
+
+/**
+ * Gets private method of a class.
+ *
+ * @param class-string $class_name Name of the class.
+ * @param string       $method Name of the method.
+ * @return ReflectionMethod
+ */
+function get_private_method( string $class_name, string $method ): ReflectionMethod {
+	$reflector = new ReflectionClass( $class_name );
+	$method = $reflector->getMethod( $method );
+
+	/**
+	 * @psalm-suppress UnusedMethodCall
+	 */
+	$method->setAccessible( true );
+
+	return $method;
+}