Merge pull request #4113 from overleaf/ta-post-registration-analytics

Add Post Registration Analytics Job

GitOrigin-RevId: f0d83eeea2e32915782e916cb40a768d5c1b6116

services/web/test/unit/src/User/UserCreatorTests.js

@@ -48,6 +48,9 @@ describe('UserCreator', function () {
         './UserOnboardingEmailManager': (this.UserOnboardingEmailManager = {
           scheduleOnboardingEmail: sinon.stub(),
         }),
+        './UserPostRegistrationAnalyticsManager': (this.UserPostRegistrationAnalyticsManager = {
+          schedulePostRegistrationAnalytics: sinon.stub(),
+        }),
       },
     })
 
@@ -279,7 +282,7 @@ describe('UserCreator', function () {
         )
       })
 
-      it('should schedule an onboarding email on registration with saas feature', async function () {
+      it('should schedule post registration jobs on registration with saas feature', async function () {
         this.Features.hasFeature = sinon.stub().withArgs('saas').returns(true)
         const user = await this.UserCreator.promises.createNewUser({
           email: this.email,
@@ -289,14 +292,23 @@ describe('UserCreator', function () {
           this.UserOnboardingEmailManager.scheduleOnboardingEmail,
           user
         )
+        sinon.assert.calledWith(
+          this.UserPostRegistrationAnalyticsManager
+            .schedulePostRegistrationAnalytics,
+          user
+        )
       })
 
-      it('should not add schedule onboarding email when without saas feature', async function () {
+      it('should not schedule post registration checks when without saas feature', async function () {
         const attributes = { email: this.email }
         await this.UserCreator.promises.createNewUser(attributes)
         sinon.assert.notCalled(
           this.UserOnboardingEmailManager.scheduleOnboardingEmail
         )
+        sinon.assert.notCalled(
+          this.UserPostRegistrationAnalyticsManager
+            .schedulePostRegistrationAnalytics
+        )
       })
     })
   })

services/web/test/unit/src/User/UserPostRegistrationAnalyticsManagerTests.js

@@ -0,0 +1,122 @@
+const SandboxedModule = require('sandboxed-module')
+const path = require('path')
+const sinon = require('sinon')
+
+const MODULE_PATH = path.join(
+  __dirname,
+  '../../../../app/src/Features/User/UserPostRegistrationAnalyticsManager'
+)
+
+describe('UserPostRegistrationAnalyticsManager', function () {
+  beforeEach(function () {
+    this.fakeUserId = '123abc'
+    this.postRegistrationAnalyticsQueue = {
+      add: sinon.stub().resolves(),
+      process: callback => {
+        this.queueProcessFunction = callback
+      },
+    }
+    const self = this
+    this.Queues = {
+      getPostRegistrationAnalyticsQueue: () => {
+        return self.postRegistrationAnalyticsQueue
+      },
+    }
+    this.UserGetter = {
+      promises: {
+        getUser: sinon.stub().resolves({ _id: this.fakeUserId }),
+      },
+    }
+    this.InstitutionsAPI = {
+      promises: {
+        getUserAffiliations: sinon.stub().resolves([]),
+      },
+    }
+    this.AnalyticsManager = {
+      setUserProperty: sinon.stub().resolves(),
+    }
+    this.UserPostRegistrationAnalyticsManager = SandboxedModule.require(
+      MODULE_PATH,
+      {
+        globals: {
+          console: console,
+        },
+        requires: {
+          '../../infrastructure/Queues': this.Queues,
+          './UserGetter': this.UserGetter,
+          '../Institutions/InstitutionsAPI': this.InstitutionsAPI,
+          '../Analytics/AnalyticsManager': this.AnalyticsManager,
+        },
+      }
+    )
+  })
+
+  describe('schedule jobs', function () {
+    it('should schedule delayed job on queue', function () {
+      this.UserPostRegistrationAnalyticsManager.schedulePostRegistrationAnalytics(
+        {
+          _id: this.fakeUserId,
+        }
+      )
+      sinon.assert.calledWithMatch(
+        this.postRegistrationAnalyticsQueue.add,
+        { userId: this.fakeUserId },
+        { delay: 24 * 60 * 60 * 1000 }
+      )
+    })
+  })
+
+  describe('process jobs', function () {
+    it('stops without errors if user is not found', async function () {
+      this.UserGetter.promises.getUser.resolves(null)
+      await this.queueProcessFunction({ data: { userId: this.fakeUserId } })
+      sinon.assert.calledWith(this.UserGetter.promises.getUser, {
+        _id: this.fakeUserId,
+      })
+      sinon.assert.notCalled(this.InstitutionsAPI.promises.getUserAffiliations)
+      sinon.assert.notCalled(this.AnalyticsManager.setUserProperty)
+    })
+
+    it('sets user property if user has commons account affiliationd', async function () {
+      this.InstitutionsAPI.promises.getUserAffiliations.resolves([
+        {},
+        {
+          institution: {
+            commonsAccount: true,
+          },
+        },
+        {
+          institution: {
+            commonsAccount: false,
+          },
+        },
+      ])
+      await this.queueProcessFunction({ data: { userId: this.fakeUserId } })
+      sinon.assert.calledWith(this.UserGetter.promises.getUser, {
+        _id: this.fakeUserId,
+      })
+      sinon.assert.calledWith(
+        this.InstitutionsAPI.promises.getUserAffiliations,
+        this.fakeUserId
+      )
+      sinon.assert.calledWith(
+        this.AnalyticsManager.setUserProperty,
+        this.fakeUserId,
+        'registered-from-commons-account',
+        true
+      )
+    })
+
+    it('does not set user property if user has no commons account affiliation', async function () {
+      this.InstitutionsAPI.promises.getUserAffiliations.resolves([
+        {
+          institution: {
+            commonsAccount: false,
+          },
+        },
+      ])
+      await this.queueProcessFunction({ data: { userId: this.fakeUserId } })
+      sinon.assert.notCalled(this.AnalyticsManager.setUserProperty)
+    })
+  })
+})