Skip to content

Commit b033366

Browse files
committed
update screen capture
1 parent de34b90 commit b033366

1 file changed

Lines changed: 26 additions & 3 deletions

File tree

app/src/androidTest/java/com/example/util/simpletimetracker/ScreenCaptureTest.kt

Lines changed: 26 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,10 @@ class ScreenCaptureTest : BaseUiTest() {
4444
val isMainTabComplex = false
4545
val showGoals = true
4646
val showMainButtons = true
47+
// false for series calendar.
48+
val addAdditionalRecords = true
49+
// true for series.
50+
val addRecordsForStatsDetailSeries = false
4751

4852
val colors = ColorMapper.getAvailableColors()
4953
val icons = iconImageMapper
@@ -172,15 +176,34 @@ class ScreenCaptureTest : BaseUiTest() {
172176
}.timeInMillis
173177
val additionalRecords = 550 - durationsMinutes.size
174178
val additionalRecordsDuration = TimeUnit.MINUTES.toMillis(30)
175-
repeat(additionalRecords) {
176-
testUtils.addRecord(guitarType, timeEnded - additionalRecordsDuration, timeEnded)
179+
if (addAdditionalRecords) {
180+
repeat(additionalRecords) {
181+
testUtils.addRecord(guitarType, timeEnded - additionalRecordsDuration, timeEnded)
182+
}
177183
}
178184
val desiredDuration = 495 * 60 + 25
179185
val additionalRecordDuration = desiredDuration -
180186
durationsMinutes.sum() -
181187
additionalRecords * 30
182188
val additionalRecordDurationMillis = TimeUnit.MINUTES.toMillis(additionalRecordDuration.toLong())
183-
testUtils.addRecord(guitarType, timeEnded - additionalRecordDurationMillis, timeEnded)
189+
if (addAdditionalRecords) {
190+
testUtils.addRecord(guitarType, timeEnded - additionalRecordDurationMillis, timeEnded)
191+
}
192+
if (addRecordsForStatsDetailSeries) {
193+
val timeEndedForSeries = calendar.apply {
194+
timeInMillis = timeEnded
195+
add(Calendar.DATE, 30)
196+
}.timeInMillis
197+
repeat(3) { it ->
198+
val day = it.toLong()
199+
val hourInMillis = TimeUnit.HOURS.toMillis(1)
200+
testUtils.addRecord(
201+
guitarType,
202+
timeEndedForSeries - TimeUnit.DAYS.toMillis(day),
203+
timeEndedForSeries - TimeUnit.DAYS.toMillis(day) + hourInMillis,
204+
)
205+
}
206+
}
184207

185208
// Main tab
186209
val hour = TimeUnit.HOURS.toMillis(1)

0 commit comments

Comments
 (0)