@atproto/ozone 0.0.16 → 0.0.17-next.1

package/tests/moderation-statuses.test.ts

@@ -1,32 +1,28 @@
 import assert from 'node:assert'
-import { TestNetwork, SeedClient, basicSeed } from '@atproto/dev-env'
-import AtpAgent, {
-  ComAtprotoAdminDefs,
-  ComAtprotoAdminQueryModerationStatuses,
+import {
+  TestNetwork,
+  SeedClient,
+  basicSeed,
+  ModeratorClient,
+} from '@atproto/dev-env'
+import {
+  ToolsOzoneModerationDefs,
+  ToolsOzoneModerationQueryStatuses,
 } from '@atproto/api'
 import { forSnapshot } from './_util'
 import {
   REASONMISLEADING,
   REASONSPAM,
 } from '../src/lexicon/types/com/atproto/moderation/defs'
+import {
+  REVIEWOPEN,
+  REVIEWNONE,
+} from '../src/lexicon/types/tools/ozone/moderation/defs'
 
 describe('moderation-statuses', () => {
   let network: TestNetwork
-  let agent: AtpAgent
-  let pdsAgent: AtpAgent
   let sc: SeedClient
-
-  const emitModerationEvent = async (eventData) => {
-    return pdsAgent.api.com.atproto.admin.emitModerationEvent(eventData, {
-      encoding: 'application/json',
-      headers: network.ozone.adminAuthHeaders('moderator'),
-    })
-  }
-
-  const queryModerationStatuses = (statusQuery) =>
-    agent.api.com.atproto.admin.queryModerationStatuses(statusQuery, {
-      headers: network.ozone.adminAuthHeaders('moderator'),
-    })
+  let modClient: ModeratorClient
 
   const seedEvents = async () => {
     const bobsAccount = {
@@ -49,25 +45,19 @@ describe('moderation-statuses', () => {
     }
 
     for (let i = 0; i < 4; i++) {
-      await emitModerationEvent({
-        event: {
-          $type: 'com.atproto.admin.defs#modEventReport',
-          reportType: i % 2 ? REASONSPAM : REASONMISLEADING,
-          comment: 'X',
-        },
+      await sc.createReport({
+        reasonType: i % 2 ? REASONSPAM : REASONMISLEADING,
+        reason: 'X',
         //   Report bob's account by alice and vice versa
         subject: i % 2 ? bobsAccount : carlasAccount,
-        createdBy: i % 2 ? sc.dids.alice : sc.dids.bob,
+        reportedBy: i % 2 ? sc.dids.alice : sc.dids.bob,
       })
-      await emitModerationEvent({
-        event: {
-          $type: 'com.atproto.admin.defs#modEventReport',
-          reportType: REASONSPAM,
-          comment: 'X',
-        },
+      await sc.createReport({
+        reasonType: REASONSPAM,
+        reason: 'X',
         //   Report bob's post by alice and vice versa
         subject: i % 2 ? bobsPost : alicesPost,
-        createdBy: i % 2 ? sc.dids.alice : sc.dids.bob,
+        reportedBy: i % 2 ? sc.dids.alice : sc.dids.bob,
       })
     }
   }
@@ -76,9 +66,8 @@ describe('moderation-statuses', () => {
     network = await TestNetwork.create({
       dbPostgresSchema: 'ozone_moderation_statuses',
     })
-    agent = network.ozone.getClient()
-    pdsAgent = network.pds.getClient()
     sc = network.getSeedClient()
+    modClient = network.ozone.getModClient()
     await basicSeed(sc)
     await network.processAll()
     await seedEvents()
@@ -90,44 +79,44 @@ describe('moderation-statuses', () => {
 
   describe('query statuses', () => {
     it('returns statuses for subjects that received moderation events', async () => {
-      const response = await queryModerationStatuses({})
+      const response = await modClient.queryStatuses({})
 
-      expect(forSnapshot(response.data.subjectStatuses)).toMatchSnapshot()
+      expect(forSnapshot(response.subjectStatuses)).toMatchSnapshot()
     })
 
     it('returns statuses filtered by subject language', async () => {
-      const klingonQueue = await queryModerationStatuses({
+      const klingonQueue = await modClient.queryStatuses({
         tags: ['lang:i'],
       })
 
-      expect(forSnapshot(klingonQueue.data.subjectStatuses)).toMatchSnapshot()
+      expect(forSnapshot(klingonQueue.subjectStatuses)).toMatchSnapshot()
 
-      const nonKlingonQueue = await queryModerationStatuses({
+      const nonKlingonQueue = await modClient.queryStatuses({
         excludeTags: ['lang:i'],
       })
 
       // Verify that the klingon tagged subject is not returned when excluding klingon
-      expect(
-        nonKlingonQueue.data.subjectStatuses.map((s) => s.id),
-      ).not.toContain(klingonQueue.data.subjectStatuses[0].id)
+      expect(nonKlingonQueue.subjectStatuses.map((s) => s.id)).not.toContain(
+        klingonQueue.subjectStatuses[0].id,
+      )
     })
 
     it('returns paginated statuses', async () => {
       // We know there will be exactly 4 statuses in db
       const getPaginatedStatuses = async (
-        params: ComAtprotoAdminQueryModerationStatuses.QueryParams,
+        params: ToolsOzoneModerationQueryStatuses.QueryParams,
       ) => {
         let cursor: string | undefined = ''
-        const statuses: ComAtprotoAdminDefs.SubjectStatusView[] = []
+        const statuses: ToolsOzoneModerationDefs.SubjectStatusView[] = []
         let count = 0
         do {
-          const results = await queryModerationStatuses({
+          const results = await modClient.queryStatuses({
             limit: 1,
             cursor,
             ...params,
           })
-          cursor = results.data.cursor
-          statuses.push(...results.data.subjectStatuses)
+          cursor = results.cursor
+          statuses.push(...results.subjectStatuses)
           count++
           // The count is just a brake-check to prevent infinite loop
         } while (cursor && count < 10)
@@ -139,13 +128,12 @@ describe('moderation-statuses', () => {
       expect(list[0].id).toEqual(7)
       expect(list[list.length - 1].id).toEqual(1)
 
-      await emitModerationEvent({
+      await modClient.emitEvent({
         subject: list[1].subject,
         event: {
-          $type: 'com.atproto.admin.defs#modEventAcknowledge',
+          $type: 'tools.ozone.moderation.defs#modEventAcknowledge',
           comment: 'X',
         },
-        createdBy: sc.dids.bob,
       })
 
       const listReviewedFirst = await getPaginatedStatuses({
@@ -160,13 +148,97 @@ describe('moderation-statuses', () => {
     })
   })
 
+  describe('reviewState changes', () => {
+    it('only sets state to #reviewNone on first non-impactful event', async () => {
+      const bobsAccount = {
+        $type: 'com.atproto.admin.defs#repoRef',
+        did: sc.dids.bob,
+      }
+      const alicesPost = {
+        $type: 'com.atproto.repo.strongRef',
+        uri: sc.posts[sc.dids.alice][0].ref.uriStr,
+        cid: sc.posts[sc.dids.alice][0].ref.cidStr,
+      }
+      const getBobsAccountStatus = async () => {
+        const data = await modClient.queryStatuses({
+          subject: bobsAccount.did,
+        })
+
+        return data.subjectStatuses[0]
+      }
+      // Since bob's account already had a reviewState, it won't be changed by non-impactful events
+      const bobsAccountStatusBeforeTag = await getBobsAccountStatus()
+
+      await Promise.all([
+        modClient.emitEvent({
+          subject: bobsAccount,
+          event: {
+            $type: 'tools.ozone.moderation.defs#modEventTag',
+            add: ['newTag'],
+            remove: [],
+            comment: 'X',
+          },
+          createdBy: sc.dids.alice,
+        }),
+        modClient.emitEvent({
+          subject: bobsAccount,
+          event: {
+            $type: 'tools.ozone.moderation.defs#modEventComment',
+            comment: 'X',
+          },
+          createdBy: sc.dids.alice,
+        }),
+      ])
+      const bobsAccountStatusAfterTag = await getBobsAccountStatus()
+
+      expect(bobsAccountStatusBeforeTag.reviewState).toEqual(
+        bobsAccountStatusAfterTag.reviewState,
+      )
+
+      // Since alice's post didn't have a reviewState it is set to reviewNone on first non-impactful event
+      const getAlicesPostStatus = async () => {
+        const data = await modClient.queryStatuses({
+          subject: alicesPost.uri,
+        })
+
+        return data.subjectStatuses[0]
+      }
+
+      const alicesPostStatusBeforeTag = await getAlicesPostStatus()
+      expect(alicesPostStatusBeforeTag).toBeUndefined()
+
+      await modClient.emitEvent({
+        subject: alicesPost,
+        event: {
+          $type: 'tools.ozone.moderation.defs#modEventComment',
+          comment: 'X',
+        },
+        createdBy: sc.dids.alice,
+      })
+      const alicesPostStatusAfterTag = await getAlicesPostStatus()
+      expect(alicesPostStatusAfterTag.reviewState).toEqual(REVIEWNONE)
+
+      await modClient.emitEvent({
+        subject: alicesPost,
+        event: {
+          $type: 'tools.ozone.moderation.defs#modEventReport',
+          reportType: REASONMISLEADING,
+          comment: 'X',
+        },
+        createdBy: sc.dids.alice,
+      })
+      const alicesPostStatusAfterReport = await getAlicesPostStatus()
+      expect(alicesPostStatusAfterReport.reviewState).toEqual(REVIEWOPEN)
+    })
+  })
+
   describe('blobs', () => {
     it('are tracked on takendown subject', async () => {
       const post = sc.posts[sc.dids.carol][0]
       assert(post.images.length > 1)
-      await emitModerationEvent({
+      await modClient.emitEvent({
         event: {
-          $type: 'com.atproto.admin.defs#modEventTakedown',
+          $type: 'tools.ozone.moderation.defs#modEventTakedown',
         },
         subject: {
           $type: 'com.atproto.repo.strongRef',
@@ -176,11 +248,9 @@ describe('moderation-statuses', () => {
         subjectBlobCids: [post.images[0].image.ref.toString()],
         createdBy: sc.dids.alice,
       })
-      const { data: result } =
-        await pdsAgent.api.com.atproto.admin.queryModerationStatuses(
-          { subject: post.ref.uriStr },
-          { headers: network.ozone.adminAuthHeaders('moderator') },
-        )
+      const result = await modClient.queryStatuses({
+        subject: post.ref.uriStr,
+      })
       expect(result.subjectStatuses.length).toBe(1)
       expect(result.subjectStatuses[0]).toMatchObject({
         takendown: true,
@@ -190,22 +260,19 @@ describe('moderation-statuses', () => {
 
     it('are tracked on reverse-takendown subject based on previous status', async () => {
       const post = sc.posts[sc.dids.carol][0]
-      await emitModerationEvent({
+      await modClient.emitEvent({
         event: {
-          $type: 'com.atproto.admin.defs#modEventReverseTakedown',
+          $type: 'tools.ozone.moderation.defs#modEventReverseTakedown',
         },
         subject: {
           $type: 'com.atproto.repo.strongRef',
           uri: post.ref.uriStr,
           cid: post.ref.cidStr,
         },
-        createdBy: sc.dids.alice,
       })
-      const { data: result } =
-        await pdsAgent.api.com.atproto.admin.queryModerationStatuses(
-          { subject: post.ref.uriStr },
-          { headers: network.ozone.adminAuthHeaders('moderator') },
-        )
+      const result = await modClient.queryStatuses({
+        subject: post.ref.uriStr,
+      })
       expect(result.subjectStatuses.length).toBe(1)
       expect(result.subjectStatuses[0]).toMatchObject({
         takendown: false,