def test_transfer_message_files(self) -> None: bucket = create_s3_buckets(settings.S3_AUTH_UPLOADS_BUCKET)[0] hamlet = self.example_user('hamlet') othello = self.example_user('othello') upload_message_file('dummy1.txt', len(b'zulip1!'), 'text/plain', b'zulip1!', hamlet) upload_message_file('dummy2.txt', len(b'zulip2!'), 'text/plain', b'zulip2!', othello) transfer_message_files_to_s3(1) attachments = Attachment.objects.all() self.assertEqual(len(bucket.get_all_keys()), 2) self.assertEqual(bucket.get_key(attachments[0].path_id).get_contents_as_string(), b'zulip1!') self.assertEqual(bucket.get_key(attachments[1].path_id).get_contents_as_string(), b'zulip2!')
def test_transfer_message_files(self) -> None: bucket = create_s3_buckets(settings.S3_AUTH_UPLOADS_BUCKET)[0] hamlet = self.example_user('hamlet') othello = self.example_user('othello') upload_message_file(u'dummy1.txt', len(b'zulip1!'), u'text/plain', b'zulip1!', hamlet) upload_message_file(u'dummy2.txt', len(b'zulip2!'), u'text/plain', b'zulip2!', othello) transfer_message_files_to_s3(1) attachments = Attachment.objects.all() self.assertEqual(len(bucket.get_all_keys()), 2) self.assertEqual(bucket.get_key(attachments[0].path_id).get_contents_as_string(), b'zulip1!') self.assertEqual(bucket.get_key(attachments[1].path_id).get_contents_as_string(), b'zulip2!')
def test_transfer_message_files(self) -> None: bucket = create_s3_buckets(settings.S3_AUTH_UPLOADS_BUCKET)[0] hamlet = self.example_user("hamlet") othello = self.example_user("othello") upload_message_file("dummy1.txt", len(b"zulip1!"), "text/plain", b"zulip1!", hamlet) upload_message_file("dummy2.txt", len(b"zulip2!"), "text/plain", b"zulip2!", othello) with self.assertLogs(level="INFO"): transfer_message_files_to_s3(1) attachments = Attachment.objects.all().order_by("id") self.assert_length(list(bucket.objects.all()), 2) self.assertEqual(bucket.Object(attachments[0].path_id).get()["Body"].read(), b"zulip1!") self.assertEqual(bucket.Object(attachments[1].path_id).get()["Body"].read(), b"zulip2!")
def test_transfer_message_files(self) -> None: bucket = create_s3_buckets(settings.S3_AUTH_UPLOADS_BUCKET)[0] hamlet = self.example_user('hamlet') othello = self.example_user('othello') upload_message_file('dummy1.txt', len(b'zulip1!'), 'text/plain', b'zulip1!', hamlet) upload_message_file('dummy2.txt', len(b'zulip2!'), 'text/plain', b'zulip2!', othello) with self.assertLogs(level="INFO"): transfer_message_files_to_s3(1) attachments = Attachment.objects.all().order_by("id") self.assertEqual(len(list(bucket.objects.all())), 2) self.assertEqual( bucket.Object(attachments[0].path_id).get()['Body'].read(), b'zulip1!') self.assertEqual( bucket.Object(attachments[1].path_id).get()['Body'].read(), b'zulip2!')