Skip to content

Commit

Permalink
Modified MessageDetailsEkmFlowTest.| #2137
Browse files Browse the repository at this point in the history
  • Loading branch information
@DenBond7
DenBond7 committed May 7, 2023
1 parent 40e3b3b commit 952d70e
Showing 1 changed file with 96 additions and 4 deletions.
100 changes: 96 additions & 4 deletions FlowCrypt/src/androidTest/java/com/flowcrypt/email/ui/MessageDetailsEkmFlowTest.kt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
package com.flowcrypt.email.ui

import androidx.test.espresso.Espresso.onView
import androidx.test.espresso.action.ViewActions.click
import androidx.test.espresso.assertion.ViewAssertions.matches
import androidx.test.espresso.matcher.ViewMatchers.isDisplayed
import androidx.test.espresso.matcher.ViewMatchers.withId
Expand All @@ -16,23 +17,38 @@ import com.flowcrypt.email.R
import com.flowcrypt.email.TestConstants
import com.flowcrypt.email.api.email.model.AttachmentInfo
import com.flowcrypt.email.api.retrofit.response.model.ClientConfiguration
import com.flowcrypt.email.database.entity.AccountEntity
import com.flowcrypt.email.database.entity.KeyEntity
import com.flowcrypt.email.extensions.kotlin.toHex
import com.flowcrypt.email.model.KeyImportDetails
import com.flowcrypt.email.rules.AddAccountToDatabaseRule
import com.flowcrypt.email.rules.AddPrivateKeyToDatabaseRule
import com.flowcrypt.email.rules.ClearAppSettingsRule
import com.flowcrypt.email.rules.FlowCryptMockWebServerRule
import com.flowcrypt.email.rules.GrantPermissionRuleChooser
import com.flowcrypt.email.rules.RetryRule
import com.flowcrypt.email.rules.ScreenshotTestRule
import com.flowcrypt.email.ui.base.BaseMessageDetailsFlowTest
import com.flowcrypt.email.util.AccountDaoManager
import com.flowcrypt.email.util.TestGeneralUtil
import com.google.api.client.json.gson.GsonFactory
import com.google.api.services.gmail.model.Message
import com.google.api.services.gmail.model.MessagePart
import com.google.api.services.gmail.model.MessagePartBody
import com.google.api.services.gmail.model.MessagePartHeader
import okhttp3.mockwebserver.Dispatcher
import okhttp3.mockwebserver.MockResponse
import okhttp3.mockwebserver.RecordedRequest
import org.hamcrest.Matchers.not
import org.junit.Rule
import org.junit.Test
import org.junit.rules.RuleChain
import org.junit.rules.TestRule
import org.junit.runner.RunWith
import java.math.BigInteger
import java.net.HttpURLConnection
import java.util.UUID
import kotlin.random.Random

/**
* @author Denys Bondarenko
Expand All @@ -44,8 +60,8 @@ class MessageDetailsEkmFlowTest : BaseMessageDetailsFlowTest() {
"messages/attachments/simple_att.json",
AttachmentInfo::class.java
)
private val userWithClientConfiguration = AccountDaoManager.getUserWithClientConfiguration(
ClientConfiguration(
private val userWithClientConfiguration = AccountDaoManager.getDefaultAccountDao().copy(
clientConfiguration = ClientConfiguration(
flags = listOf(
ClientConfiguration.ConfigurationProperty.NO_PRV_CREATE,
ClientConfiguration.ConfigurationProperty.NO_PRV_BACKUP,
Expand All @@ -56,8 +72,12 @@ class MessageDetailsEkmFlowTest : BaseMessageDetailsFlowTest() {
disallowAttesterSearchForDomains = null,
enforceKeygenAlgo = null,
enforceKeygenExpireMonths = null
)
),
accountType = AccountEntity.ACCOUNT_TYPE_GOOGLE,
useCustomerFesUrl = true,
useAPI = true
)

override val addAccountToDatabaseRule = AddAccountToDatabaseRule(userWithClientConfiguration)
private val addPrivateKeyToDatabaseRule = AddPrivateKeyToDatabaseRule(
accountEntity = addAccountToDatabaseRule.account,
Expand All @@ -67,11 +87,72 @@ class MessageDetailsEkmFlowTest : BaseMessageDetailsFlowTest() {
passphraseType = KeyEntity.PassphraseType.DATABASE
)

private val mockWebServerRule = FlowCryptMockWebServerRule(
TestConstants.MOCK_WEB_SERVER_PORT,
object : Dispatcher() {
override fun dispatch(request: RecordedRequest): MockResponse {
when (request.path) {
"/gmail/v1/users/me/messages/${simpleAttInfo?.uid?.toHex()}?fields=" +
"id,threadId,labelIds,snippet,sizeEstimate,historyId,internalDate,payload/partId," +
"payload/mimeType,payload/filename,payload/headers,payload/body,payload/" +
"parts(partId,mimeType,filename,headers,body/size,body/attachmentId)&format=full" -> {
return MockResponse()
.setResponseCode(HttpURLConnection.HTTP_OK)
.setBody(
Message().apply {
factory = GsonFactory.getDefaultInstance()
id = "5555555555555555"
threadId = "1111111111111111"
payload = MessagePart().apply {
partId = ""
mimeType = "multipart/mixed"
filename = ""
parts = listOf(
MessagePart().apply {
partId = "0"
mimeType = "image/png"
filename = "android.png"
headers = listOf(
MessagePartHeader().apply {
name = "Content-Type"
value = "image/png; name=\"android.png\""
}, MessagePartHeader().apply {
name = "Content-Disposition"
value = "attachment; filename=\"android.png\""
})
body = MessagePartBody().apply {
attachmentId = ATTACHMENT_ID
}
}
)
}
historyId = BigInteger.valueOf(Random.nextLong())
}.toString()
)
}

"/gmail/v1/users/me/messages/${simpleAttInfo?.uid?.toHex()}/attachments/$ATTACHMENT_ID?fields=data&prettyPrint=false" -> {
return MockResponse()
.setResponseCode(HttpURLConnection.HTTP_OK)
.setBody(
MessagePartBody().apply {
factory = GsonFactory.getDefaultInstance()
data = "we don't care about this content"
}.toString()
)
}

else -> return MockResponse().setResponseCode(HttpURLConnection.HTTP_NOT_FOUND)
}
}
})

@get:Rule
var ruleChain: TestRule = RuleChain
.outerRule(RetryRule.DEFAULT)
.around(ClearAppSettingsRule())
.around(GrantPermissionRuleChooser.grant(android.Manifest.permission.POST_NOTIFICATIONS))
.around(mockWebServerRule)
.around(addAccountToDatabaseRule)
.around(addPrivateKeyToDatabaseRule)
.around(activeActivityRule)
Expand All @@ -94,14 +175,25 @@ class MessageDetailsEkmFlowTest : BaseMessageDetailsFlowTest() {
}

@Test
fun testVisibilityOfPreviewAttachmentButton() {
fun testPreviewAttachmentButton() {
baseCheckWithAtt(
getMsgInfo(
"messages/info/standard_msg_info_plaintext_with_one_att.json",
"messages/mime/standard_msg_info_plaintext_with_one_att.txt", simpleAttInfo
), simpleAttInfo
)

unregisterCountingIdlingResource()

onView(withId(R.id.imageButtonPreviewAtt))
.check(matches(isDisplayed()))
.perform(click())
Thread.sleep(2000)
//as the Content app is not installed on a device the app should show the warning
isDialogWithTextDisplayed(decorView, getResString(R.string.warning_don_not_have_content_app))
}

companion object {
val ATTACHMENT_ID = UUID.randomUUID().toString()
}
}

0 comments on commit 952d70e

Please sign in to comment.