mirror of
https://code.briarproject.org/briar/briar.git
synced 2026-02-14 19:59:05 +01:00
Turn AttachmentReader into a proper class
and inject it where needed
This commit is contained in:
@@ -53,12 +53,12 @@ import static org.briarproject.bramble.test.TestUtils.getRandomId;
|
||||
import static org.briarproject.bramble.util.StringUtils.getRandomString;
|
||||
import static org.briarproject.briar.api.avatar.AvatarManager.CLIENT_ID;
|
||||
import static org.briarproject.briar.api.avatar.AvatarManager.MAJOR_VERSION;
|
||||
import static org.briarproject.briar.api.messaging.MessagingConstants.MAX_CONTENT_TYPE_BYTES;
|
||||
import static org.briarproject.briar.api.media.MediaConstants.MAX_CONTENT_TYPE_BYTES;
|
||||
import static org.briarproject.briar.avatar.AvatarConstants.GROUP_KEY_CONTACT_ID;
|
||||
import static org.briarproject.briar.avatar.AvatarConstants.MSG_KEY_VERSION;
|
||||
import static org.briarproject.briar.avatar.AvatarConstants.MSG_TYPE_UPDATE;
|
||||
import static org.briarproject.briar.media.MediaConstants.MSG_KEY_CONTENT_TYPE;
|
||||
import static org.briarproject.briar.media.MediaConstants.MSG_KEY_DESCRIPTOR_LENGTH;
|
||||
import static org.briarproject.briar.api.media.MediaConstants.MSG_KEY_CONTENT_TYPE;
|
||||
import static org.briarproject.briar.api.media.MediaConstants.MSG_KEY_DESCRIPTOR_LENGTH;
|
||||
import static org.junit.Assert.assertEquals;
|
||||
import static org.junit.Assert.assertFalse;
|
||||
|
||||
|
||||
@@ -4,6 +4,7 @@ import org.briarproject.bramble.test.TestDatabaseConfigModule;
|
||||
import org.briarproject.briar.api.avatar.AvatarManager;
|
||||
import org.briarproject.briar.api.media.Attachment;
|
||||
import org.briarproject.briar.api.media.AttachmentHeader;
|
||||
import org.briarproject.briar.api.media.AttachmentReader;
|
||||
import org.briarproject.briar.test.BriarIntegrationTest;
|
||||
import org.briarproject.briar.test.BriarIntegrationTestComponent;
|
||||
import org.briarproject.briar.test.DaggerBriarIntegrationTestComponent;
|
||||
@@ -17,7 +18,7 @@ import java.io.InputStream;
|
||||
import static org.briarproject.bramble.test.TestUtils.getRandomBytes;
|
||||
import static org.briarproject.bramble.util.IoUtils.copyAndClose;
|
||||
import static org.briarproject.bramble.util.StringUtils.getRandomString;
|
||||
import static org.briarproject.briar.api.messaging.MessagingConstants.MAX_CONTENT_TYPE_BYTES;
|
||||
import static org.briarproject.briar.api.media.MediaConstants.MAX_CONTENT_TYPE_BYTES;
|
||||
import static org.junit.Assert.assertArrayEquals;
|
||||
import static org.junit.Assert.assertEquals;
|
||||
import static org.junit.Assert.assertNotNull;
|
||||
@@ -27,6 +28,7 @@ public class AvatarManagerIntegrationTest
|
||||
extends BriarIntegrationTest<BriarIntegrationTestComponent> {
|
||||
|
||||
private AvatarManager avatarManager0, avatarManager1;
|
||||
private AttachmentReader attachmentReader0, attachmentReader1;
|
||||
|
||||
private final String contentType = getRandomString(MAX_CONTENT_TYPE_BYTES);
|
||||
|
||||
@@ -36,6 +38,8 @@ public class AvatarManagerIntegrationTest
|
||||
super.setUp();
|
||||
avatarManager0 = c0.getAvatarManager();
|
||||
avatarManager1 = c1.getAvatarManager();
|
||||
attachmentReader0 = c0.getAttachmentReader();
|
||||
attachmentReader1 = c1.getAttachmentReader();
|
||||
}
|
||||
|
||||
@Override
|
||||
@@ -90,7 +94,7 @@ public class AvatarManagerIntegrationTest
|
||||
assertNotNull(header0.getMessageId());
|
||||
|
||||
// 0 can retrieve their own avatar
|
||||
Attachment attachment0 = avatarManager0.getAvatar(header0);
|
||||
Attachment attachment0 = attachmentReader0.getAttachment(header0);
|
||||
assertEquals(contentType, attachment0.getHeader().getContentType());
|
||||
assertStreamMatches(avatar0bytes, attachment0.getStream());
|
||||
|
||||
@@ -105,7 +109,8 @@ public class AvatarManagerIntegrationTest
|
||||
assertNotNull(header0From1.getMessageId());
|
||||
|
||||
// 1 can retrieve 0's avatar
|
||||
Attachment attachment0From1 = avatarManager1.getAvatar(header0From1);
|
||||
Attachment attachment0From1 =
|
||||
attachmentReader1.getAttachment(header0From1);
|
||||
assertEquals(contentType,
|
||||
attachment0From1.getHeader().getContentType());
|
||||
assertStreamMatches(avatar0bytes, attachment0From1.getStream());
|
||||
@@ -127,7 +132,8 @@ public class AvatarManagerIntegrationTest
|
||||
assertNotNull(header1From0.getMessageId());
|
||||
|
||||
// 0 can retrieve 1's avatar
|
||||
Attachment attachment1From0 = avatarManager0.getAvatar(header1From0);
|
||||
Attachment attachment1From0 =
|
||||
attachmentReader0.getAttachment(header1From0);
|
||||
assertEquals(contentType1,
|
||||
attachment1From0.getHeader().getContentType());
|
||||
assertStreamMatches(avatar1bytes, attachment1From0.getStream());
|
||||
@@ -144,7 +150,7 @@ public class AvatarManagerIntegrationTest
|
||||
AttachmentHeader header0 = db0.transactionWithResult(true,
|
||||
txn -> avatarManager0.getMyAvatarHeader(txn));
|
||||
assertNotNull(header0);
|
||||
Attachment attachment0 = avatarManager0.getAvatar(header0);
|
||||
Attachment attachment0 = attachmentReader0.getAttachment(header0);
|
||||
assertStreamMatches(avatar0bytes, attachment0.getStream());
|
||||
|
||||
// send the avatar from 0 to 1
|
||||
@@ -154,7 +160,8 @@ public class AvatarManagerIntegrationTest
|
||||
AttachmentHeader header0From1 = db1.transactionWithNullableResult(true,
|
||||
txn -> avatarManager1.getAvatarHeader(txn, contact0From1));
|
||||
assertNotNull(header0From1);
|
||||
Attachment attachment0From1 = avatarManager1.getAvatar(header0From1);
|
||||
Attachment attachment0From1 =
|
||||
attachmentReader1.getAttachment(header0From1);
|
||||
assertStreamMatches(avatar0bytes, attachment0From1.getStream());
|
||||
|
||||
// 0 adds a new avatar
|
||||
@@ -167,7 +174,7 @@ public class AvatarManagerIntegrationTest
|
||||
header0 = db0.transactionWithResult(true,
|
||||
txn -> avatarManager0.getMyAvatarHeader(txn));
|
||||
assertNotNull(header0);
|
||||
attachment0 = avatarManager0.getAvatar(header0);
|
||||
attachment0 = attachmentReader0.getAttachment(header0);
|
||||
assertStreamMatches(avatar0bytes2, attachment0.getStream());
|
||||
|
||||
// send the new avatar from 0 to 1
|
||||
@@ -177,7 +184,7 @@ public class AvatarManagerIntegrationTest
|
||||
header0From1 = db1.transactionWithNullableResult(true,
|
||||
txn -> avatarManager1.getAvatarHeader(txn, contact0From1));
|
||||
assertNotNull(header0From1);
|
||||
attachment0From1 = avatarManager1.getAvatar(header0From1);
|
||||
attachment0From1 = attachmentReader1.getAttachment(header0From1);
|
||||
assertStreamMatches(avatar0bytes2, attachment0From1.getStream());
|
||||
}
|
||||
|
||||
|
||||
@@ -23,11 +23,11 @@ import static org.briarproject.bramble.test.TestUtils.getClientId;
|
||||
import static org.briarproject.bramble.test.TestUtils.getGroup;
|
||||
import static org.briarproject.bramble.test.TestUtils.getMessage;
|
||||
import static org.briarproject.bramble.util.StringUtils.getRandomString;
|
||||
import static org.briarproject.briar.api.messaging.MessagingConstants.MAX_CONTENT_TYPE_BYTES;
|
||||
import static org.briarproject.briar.api.media.MediaConstants.MAX_CONTENT_TYPE_BYTES;
|
||||
import static org.briarproject.briar.avatar.AvatarConstants.MSG_KEY_VERSION;
|
||||
import static org.briarproject.briar.avatar.AvatarConstants.MSG_TYPE_UPDATE;
|
||||
import static org.briarproject.briar.media.MediaConstants.MSG_KEY_CONTENT_TYPE;
|
||||
import static org.briarproject.briar.media.MediaConstants.MSG_KEY_DESCRIPTOR_LENGTH;
|
||||
import static org.briarproject.briar.api.media.MediaConstants.MSG_KEY_CONTENT_TYPE;
|
||||
import static org.briarproject.briar.api.media.MediaConstants.MSG_KEY_DESCRIPTOR_LENGTH;
|
||||
import static org.junit.Assert.assertEquals;
|
||||
|
||||
public class AvatarValidatorTest extends BrambleMockTestCase {
|
||||
|
||||
@@ -30,7 +30,7 @@ import static org.briarproject.briar.api.identity.AuthorInfo.Status.OURSELVES;
|
||||
import static org.briarproject.briar.api.identity.AuthorInfo.Status.UNKNOWN;
|
||||
import static org.briarproject.briar.api.identity.AuthorInfo.Status.UNVERIFIED;
|
||||
import static org.briarproject.briar.api.identity.AuthorInfo.Status.VERIFIED;
|
||||
import static org.briarproject.briar.api.messaging.MessagingConstants.MAX_CONTENT_TYPE_BYTES;
|
||||
import static org.briarproject.briar.api.media.MediaConstants.MAX_CONTENT_TYPE_BYTES;
|
||||
import static org.junit.Assert.assertEquals;
|
||||
import static org.junit.Assert.assertNull;
|
||||
|
||||
|
||||
@@ -7,6 +7,7 @@ import org.briarproject.briar.blog.BlogModule;
|
||||
import org.briarproject.briar.client.BriarClientModule;
|
||||
import org.briarproject.briar.forum.ForumModule;
|
||||
import org.briarproject.briar.identity.IdentityModule;
|
||||
import org.briarproject.briar.media.MediaModule;
|
||||
import org.briarproject.briar.messaging.MessagingModule;
|
||||
import org.briarproject.briar.privategroup.PrivateGroupModule;
|
||||
import org.briarproject.briar.privategroup.invitation.GroupInvitationModule;
|
||||
@@ -28,6 +29,7 @@ import dagger.Component;
|
||||
GroupInvitationModule.class,
|
||||
IdentityModule.class,
|
||||
IntroductionModule.class,
|
||||
MediaModule.class,
|
||||
MessagingModule.class,
|
||||
PrivateGroupModule.class,
|
||||
SharingModule.class
|
||||
|
||||
@@ -26,7 +26,7 @@ import static org.briarproject.bramble.test.TestUtils.getRandomId;
|
||||
import static org.briarproject.bramble.util.StringUtils.getRandomString;
|
||||
import static org.briarproject.briar.api.forum.ForumConstants.MAX_FORUM_POST_TEXT_LENGTH;
|
||||
import static org.briarproject.briar.api.messaging.MessagingConstants.MAX_ATTACHMENTS_PER_MESSAGE;
|
||||
import static org.briarproject.briar.api.messaging.MessagingConstants.MAX_CONTENT_TYPE_BYTES;
|
||||
import static org.briarproject.briar.api.media.MediaConstants.MAX_CONTENT_TYPE_BYTES;
|
||||
import static org.briarproject.briar.api.messaging.MessagingConstants.MAX_PRIVATE_MESSAGE_TEXT_LENGTH;
|
||||
import static org.junit.Assert.assertTrue;
|
||||
|
||||
|
||||
@@ -28,11 +28,11 @@ import static org.briarproject.bramble.test.TestUtils.getRandomBytes;
|
||||
import static org.briarproject.bramble.test.TestUtils.getRandomId;
|
||||
import static org.briarproject.bramble.util.StringUtils.getRandomString;
|
||||
import static org.briarproject.briar.api.messaging.MessagingConstants.MAX_ATTACHMENTS_PER_MESSAGE;
|
||||
import static org.briarproject.briar.api.messaging.MessagingConstants.MAX_CONTENT_TYPE_BYTES;
|
||||
import static org.briarproject.briar.api.media.MediaConstants.MAX_CONTENT_TYPE_BYTES;
|
||||
import static org.briarproject.briar.api.messaging.MessagingConstants.MAX_PRIVATE_MESSAGE_TEXT_LENGTH;
|
||||
import static org.briarproject.briar.client.MessageTrackerConstants.MSG_KEY_READ;
|
||||
import static org.briarproject.briar.media.MediaConstants.MSG_KEY_CONTENT_TYPE;
|
||||
import static org.briarproject.briar.media.MediaConstants.MSG_KEY_DESCRIPTOR_LENGTH;
|
||||
import static org.briarproject.briar.api.media.MediaConstants.MSG_KEY_CONTENT_TYPE;
|
||||
import static org.briarproject.briar.api.media.MediaConstants.MSG_KEY_DESCRIPTOR_LENGTH;
|
||||
import static org.briarproject.briar.messaging.MessageTypes.ATTACHMENT;
|
||||
import static org.briarproject.briar.messaging.MessageTypes.PRIVATE_MESSAGE;
|
||||
import static org.briarproject.briar.messaging.MessagingConstants.MSG_KEY_ATTACHMENT_HEADERS;
|
||||
|
||||
@@ -21,6 +21,7 @@ import org.briarproject.briar.api.conversation.ConversationManager;
|
||||
import org.briarproject.briar.api.forum.ForumManager;
|
||||
import org.briarproject.briar.api.forum.ForumSharingManager;
|
||||
import org.briarproject.briar.api.introduction.IntroductionManager;
|
||||
import org.briarproject.briar.api.media.AttachmentReader;
|
||||
import org.briarproject.briar.api.messaging.MessagingManager;
|
||||
import org.briarproject.briar.api.messaging.PrivateMessageFactory;
|
||||
import org.briarproject.briar.api.privategroup.PrivateGroupManager;
|
||||
@@ -31,6 +32,7 @@ import org.briarproject.briar.client.BriarClientModule;
|
||||
import org.briarproject.briar.forum.ForumModule;
|
||||
import org.briarproject.briar.identity.IdentityModule;
|
||||
import org.briarproject.briar.introduction.IntroductionModule;
|
||||
import org.briarproject.briar.media.MediaModule;
|
||||
import org.briarproject.briar.messaging.MessagingModule;
|
||||
import org.briarproject.briar.privategroup.PrivateGroupModule;
|
||||
import org.briarproject.briar.privategroup.invitation.GroupInvitationModule;
|
||||
@@ -51,6 +53,7 @@ import dagger.Component;
|
||||
GroupInvitationModule.class,
|
||||
IdentityModule.class,
|
||||
IntroductionModule.class,
|
||||
MediaModule.class,
|
||||
MessagingModule.class,
|
||||
PrivateGroupModule.class,
|
||||
SharingModule.class
|
||||
@@ -84,6 +87,8 @@ public interface BriarIntegrationTestComponent
|
||||
|
||||
IdentityManager getIdentityManager();
|
||||
|
||||
AttachmentReader getAttachmentReader();
|
||||
|
||||
AvatarManager getAvatarManager();
|
||||
|
||||
ClientHelper getClientHelper();
|
||||
|
||||
Reference in New Issue
Block a user