1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.hosted.controller.maintenance;
import com.yahoo.config.provision.ApplicationId;
import com.yahoo.config.provision.zone.ZoneId;
import com.yahoo.vespa.hosted.controller.api.integration.configserver.ApplicationReindexing;
import com.yahoo.vespa.hosted.controller.api.integration.configserver.ApplicationReindexing.Cluster;
import com.yahoo.vespa.hosted.controller.api.integration.configserver.ApplicationReindexing.Status;
import org.junit.jupiter.api.Test;
import java.time.Duration;
import java.time.Instant;
import java.time.ZonedDateTime;
import java.util.List;
import java.util.Map;
import static com.yahoo.vespa.hosted.controller.maintenance.ReindexingTriggerer.inWindowOfOpportunity;
import static com.yahoo.vespa.hosted.controller.maintenance.ReindexingTriggerer.reindexingIsReady;
import static com.yahoo.vespa.hosted.controller.maintenance.ReindexingTriggerer.reindexingPeriod;
import static java.time.DayOfWeek.FRIDAY;
import static java.time.DayOfWeek.MONDAY;
import static java.time.DayOfWeek.THURSDAY;
import static java.time.DayOfWeek.TUESDAY;
import static java.time.DayOfWeek.WEDNESDAY;
import static org.junit.jupiter.api.Assertions.assertFalse;
import static org.junit.jupiter.api.Assertions.assertTrue;
public class ReindexingTriggererTest {
@Test
void testWindowOfOpportunity() {
Duration interval = Duration.ofHours(1);
Instant now = Instant.now();
Instant doom = now.plus(ReindexingTriggerer.reindexingPeriod);
int triggered = 0;
while (now.isBefore(doom)) {
if (inWindowOfOpportunity(now, ApplicationId.defaultId(), ZoneId.defaultId())) {
triggered++;
ZonedDateTime time = ZonedDateTime.ofInstant(now, java.time.ZoneId.of("Europe/Oslo"));
assertTrue(List.of(MONDAY, TUESDAY, WEDNESDAY, THURSDAY, FRIDAY).contains(time.getDayOfWeek()));
assertTrue(List.of(8, 9, 10, 11).contains(time.getHour()));
}
now = now.plus(interval);
}
// Summer/winter time :'(
assertTrue(3 <= triggered && triggered <= 5, "Should be in window of opportunity three to five times each period");
}
@Test
void testReindexingIsReady() {
Instant then = Instant.now();
ApplicationReindexing reindexing = new ApplicationReindexing(true,
Map.of("c", new Cluster(Map.of(), Map.of("d", new Status(then)))));
Instant now = then;
assertFalse(reindexingIsReady(reindexing, now),
"Should not be ready less than one half-period after last triggering");
now = now.plus(reindexingPeriod.dividedBy(2));
assertFalse(reindexingIsReady(reindexing, now),
"Should not be ready one half-period after last triggering");
now = now.plusMillis(1);
assertTrue(reindexingIsReady(reindexing, now),
"Should be ready more than one half-period after last triggering");
reindexing = new ApplicationReindexing(true,
Map.of("cluster",
new Cluster(Map.of(),
Map.of("type",
new Status(then, then, null, null, null, null, 1.0, null)))));
assertFalse(reindexingIsReady(reindexing, now),
"Should not be ready when reindexing is already running");
reindexing = new ApplicationReindexing(true,
Map.of("cluster",
new Cluster(Map.of("type", 123L),
Map.of("type",
new Status(then, then, now, null, null, null, 1.0, null)))));
assertTrue(reindexingIsReady(reindexing, now),
"Should be ready when reindexing is no longer running");
}
}
|
