15
15
16
16
import static org .assertj .core .api .Assertions .assertThat ;
17
17
import static org .assertj .core .api .Assertions .fail ;
18
+ import static tech .pegasys .teku .validator .api .GraffitiManager .GRAFFITI_MANAGEMENT_DIR ;
18
19
19
20
import java .io .File ;
20
21
import java .io .IOException ;
@@ -149,6 +150,21 @@ void deleteGraffiti_shouldThrowExceptionWhenUnableToWriteFile(@TempDir final Pat
149
150
.hasValue ("java.nio.file.AccessDeniedException: " + file );
150
151
}
151
152
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
+
152
168
private void checkGraffitiFile (final BLSPublicKey publicKey , final String graffiti ) {
153
169
final Path filePath = getGraffitiManagementDir ().resolve (getFileName (publicKey ));
154
170
try {
@@ -161,7 +177,7 @@ private void checkGraffitiFile(final BLSPublicKey publicKey, final String graffi
161
177
}
162
178
163
179
private Path getGraffitiManagementDir () {
164
- return dataDirLayout .getValidatorDataDirectory ().resolve ("graffiti-management" );
180
+ return dataDirLayout .getValidatorDataDirectory ().resolve (GRAFFITI_MANAGEMENT_DIR );
165
181
}
166
182
167
183
private String getFileName (final BLSPublicKey publicKey ) {
0 commit comments