@@ -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