@@ -40,18 +40,21 @@ test.describe('Digital Letters - Handle TTL', () => {
4040 test ( 'should handle withdrawn item' , async ( ) => {
4141 const letterId = uuidv4 ( ) ;
4242 const messageUri = `https://example.com/ttl/resource/${ letterId } ` ;
43+ const messageReference = letterId ;
44+ const { senderId } = baseEvent . data ;
4345
4446 const event = {
4547 ...baseEvent ,
4648 id : letterId ,
4749 data : {
4850 ...baseEvent . data ,
4951 messageUri,
52+ messageReference,
5053 } ,
5154 } satisfies MESHInboxMessageDownloaded ;
5255
5356 const ttlItem = {
54- PK : messageUri ,
57+ PK : ` ${ senderId } _ ${ messageReference } ` ,
5558 SK : 'TTL' ,
5659 dateOfExpiry : '2023-12-31#0' ,
5760 event,
@@ -62,7 +65,7 @@ test.describe('Digital Letters - Handle TTL', () => {
6265 const putResponseCode = await putTtl ( ttlItem ) ;
6366 expect ( putResponseCode ) . toBe ( 200 ) ;
6467
65- const deleteResponseCode = await deleteTtl ( messageUri ) ;
68+ const deleteResponseCode = await deleteTtl ( senderId , messageReference ) ;
6669 expect ( deleteResponseCode ) . toBe ( 200 ) ;
6770
6871 await expectToPassEventually ( async ( ) => {
@@ -81,18 +84,21 @@ test.describe('Digital Letters - Handle TTL', () => {
8184 test ( 'should handle expired item' , async ( ) => {
8285 const letterId = uuidv4 ( ) ;
8386 const messageUri = `https://example.com/ttl/resource/${ letterId } ` ;
87+ const messageReference = letterId ;
88+ const { senderId } = baseEvent . data ;
8489
8590 const event = {
8691 ...baseEvent ,
8792 id : letterId ,
8893 data : {
8994 ...baseEvent . data ,
9095 messageUri,
96+ messageReference,
9197 } ,
9298 } satisfies MESHInboxMessageDownloaded ;
9399
94100 const ttlItem = {
95- PK : messageUri ,
101+ PK : ` ${ senderId } _ ${ messageReference } ` ,
96102 SK : 'TTL' ,
97103 dateOfExpiry : '2023-12-31#0' ,
98104 event,
@@ -102,7 +108,7 @@ test.describe('Digital Letters - Handle TTL', () => {
102108 const putResponseCode = await putTtl ( ttlItem ) ;
103109 expect ( putResponseCode ) . toBe ( 200 ) ;
104110
105- const deleteResponseCode = await deleteTtl ( messageUri ) ;
111+ const deleteResponseCode = await deleteTtl ( senderId , messageReference ) ;
106112 expect ( deleteResponseCode ) . toBe ( 200 ) ;
107113
108114 await expectToPassEventually ( async ( ) => {
@@ -123,7 +129,8 @@ test.describe('Digital Letters - Handle TTL', () => {
123129 test . setTimeout ( 160_000 ) ;
124130
125131 const letterId = uuidv4 ( ) ;
126- const messageUri = `https://example.com/ttl/resource/${ letterId } ` ;
132+ const messageReference = letterId ;
133+ const { senderId } = baseEvent . data ;
127134
128135 const eventWithNoMessageUri = {
129136 ...baseEvent ,
@@ -135,7 +142,7 @@ test.describe('Digital Letters - Handle TTL', () => {
135142 } ;
136143
137144 const ttlItem = {
138- PK : messageUri ,
145+ PK : ` ${ senderId } _ ${ messageReference } ` ,
139146 SK : 'TTL' ,
140147 dateOfExpiry : '2023-12-31#0' ,
141148 event : eventWithNoMessageUri ,
@@ -145,7 +152,7 @@ test.describe('Digital Letters - Handle TTL', () => {
145152 const putResponseCode = await putTtl ( ttlItem ) ;
146153 expect ( putResponseCode ) . toBe ( 200 ) ;
147154
148- const deleteResponseCode = await deleteTtl ( messageUri ) ;
155+ const deleteResponseCode = await deleteTtl ( senderId , messageReference ) ;
149156 expect ( deleteResponseCode ) . toBe ( 200 ) ;
150157
151158 await expectMessageContainingString ( HANDLE_TTL_DLQ_NAME , letterId , 150 ) ;
0 commit comments