Skip to content

Commit 2cb4983

Browse files
committed
Add test for when management structure already exists
1 parent cf19ffb commit 2cb4983

File tree

1 file changed

+17
-1
lines changed

1 file changed

+17
-1
lines changed

Diff for: validator/api/src/test/java/tech/pegasys/teku/validator/api/GraffitiManagerTest.java

+17-1
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515

1616
import static org.assertj.core.api.Assertions.assertThat;
1717
import static org.assertj.core.api.Assertions.fail;
18+
import static tech.pegasys.teku.validator.api.GraffitiManager.GRAFFITI_MANAGEMENT_DIR;
1819

1920
import java.io.File;
2021
import java.io.IOException;
@@ -149,6 +150,21 @@ void deleteGraffiti_shouldThrowExceptionWhenUnableToWriteFile(@TempDir final Pat
149150
.hasValue("java.nio.file.AccessDeniedException: " + file);
150151
}
151152

153+
@Test
154+
void shouldSetAndDeleteGraffitiWhenManagementPreexisting(@TempDir final Path tempDir) {
155+
dataDirLayout = new SimpleDataDirLayout(tempDir);
156+
final Path managementDir = getGraffitiManagementDir();
157+
assertThat(managementDir.toFile().mkdirs()).isTrue();
158+
159+
manager = new GraffitiManager(dataDirLayout);
160+
161+
assertThat(manager.setGraffiti(publicKey, graffiti)).isEmpty();
162+
checkGraffitiFile(publicKey, graffiti);
163+
164+
assertThat(manager.deleteGraffiti(publicKey)).isEmpty();
165+
checkGraffitiFile(publicKey, "");
166+
}
167+
152168
private void checkGraffitiFile(final BLSPublicKey publicKey, final String graffiti) {
153169
final Path filePath = getGraffitiManagementDir().resolve(getFileName(publicKey));
154170
try {
@@ -161,7 +177,7 @@ private void checkGraffitiFile(final BLSPublicKey publicKey, final String graffi
161177
}
162178

163179
private Path getGraffitiManagementDir() {
164-
return dataDirLayout.getValidatorDataDirectory().resolve("graffiti-management");
180+
return dataDirLayout.getValidatorDataDirectory().resolve(GRAFFITI_MANAGEMENT_DIR);
165181
}
166182

167183
private String getFileName(final BLSPublicKey publicKey) {

0 commit comments

Comments
 (0)