diff options
author | Harald Musum <musum@yahooinc.com> | 2023-01-02 09:13:26 +0100 |
---|---|---|
committer | Harald Musum <musum@yahooinc.com> | 2023-01-02 09:13:26 +0100 |
commit | 37bd63b9f4590d295cb82758f6b3f13325b5acfe (patch) | |
tree | f32832998bc5fefec78fb9420d58f9ac6f5a3d7d /configserver/src/test/java/com/yahoo | |
parent | 3f1f34a3a32ef7df98597619b9b26265a9a2dcce (diff) |
Always use lock when adding or deleting files from FileDirectory
Diffstat (limited to 'configserver/src/test/java/com/yahoo')
13 files changed, 15 insertions, 27 deletions
diff --git a/configserver/src/test/java/com/yahoo/vespa/config/server/ApplicationRepositoryTest.java b/configserver/src/test/java/com/yahoo/vespa/config/server/ApplicationRepositoryTest.java index 2e60a6c6cd0..61f447597a2 100644 --- a/configserver/src/test/java/com/yahoo/vespa/config/server/ApplicationRepositoryTest.java +++ b/configserver/src/test/java/com/yahoo/vespa/config/server/ApplicationRepositoryTest.java @@ -124,7 +124,7 @@ public class ApplicationRepositoryTest { .fileReferencesDir(temporaryFolder.newFolder().getAbsolutePath()) .build(); InMemoryFlagSource flagSource = new InMemoryFlagSource(); - fileDirectory = new FileDirectory(configserverConfig, flagSource); + fileDirectory = new FileDirectory(configserverConfig); tenantRepository = new TestTenantRepository.Builder() .withClock(clock) .withConfigserverConfig(configserverConfig) diff --git a/configserver/src/test/java/com/yahoo/vespa/config/server/ConfigServerBootstrapTest.java b/configserver/src/test/java/com/yahoo/vespa/config/server/ConfigServerBootstrapTest.java index 2f595d4ed06..fd2b7fe8a77 100644 --- a/configserver/src/test/java/com/yahoo/vespa/config/server/ConfigServerBootstrapTest.java +++ b/configserver/src/test/java/com/yahoo/vespa/config/server/ConfigServerBootstrapTest.java @@ -25,7 +25,6 @@ import com.yahoo.vespa.config.server.rpc.RpcServer; import com.yahoo.vespa.config.server.version.VersionState; import com.yahoo.vespa.curator.Curator; import com.yahoo.vespa.curator.mock.MockCurator; -import com.yahoo.vespa.flags.InMemoryFlagSource; import org.junit.Rule; import org.junit.Test; import org.junit.rules.TemporaryFolder; @@ -39,7 +38,6 @@ import java.util.Collections; import java.util.List; import java.util.Optional; import java.util.function.BooleanSupplier; -import java.util.stream.Collectors; import static com.yahoo.vespa.config.server.ConfigServerBootstrap.RedeployingApplicationsFails.CONTINUE; import static com.yahoo.vespa.config.server.ConfigServerBootstrap.VipStatusMode; @@ -189,7 +187,7 @@ public class ConfigServerBootstrapTest { stateMonitor, vipStatus, vipStatusMode, - new FileDirectory(tester.applicationRepository().configserverConfig(), new InMemoryFlagSource())); + new FileDirectory(tester.applicationRepository().configserverConfig())); } private void waitUntil(BooleanSupplier booleanSupplier, String messageIfWaitingFails) throws InterruptedException { diff --git a/configserver/src/test/java/com/yahoo/vespa/config/server/deploy/DeployTester.java b/configserver/src/test/java/com/yahoo/vespa/config/server/deploy/DeployTester.java index 1ed4fac10dc..e5b8a743b35 100644 --- a/configserver/src/test/java/com/yahoo/vespa/config/server/deploy/DeployTester.java +++ b/configserver/src/test/java/com/yahoo/vespa/config/server/deploy/DeployTester.java @@ -292,7 +292,7 @@ public class DeployTester { .withConfigserverConfig(configserverConfig) .withCurator(curator) .withFileDistributionFactory( - new MockFileDistributionFactory(configserverConfig, new FileDirectory(configserverConfig, flagSource))) + new MockFileDistributionFactory(configserverConfig, new FileDirectory(configserverConfig))) .withMetrics(Optional.ofNullable(metrics).orElse(Metrics.createTestMetrics())) .withModelFactoryRegistry((new ModelFactoryRegistry(modelFactories))) .withZone(zone); diff --git a/configserver/src/test/java/com/yahoo/vespa/config/server/filedistribution/FileDBRegistryTestCase.java b/configserver/src/test/java/com/yahoo/vespa/config/server/filedistribution/FileDBRegistryTestCase.java index 91ab5b4e61d..0aac0b022a4 100644 --- a/configserver/src/test/java/com/yahoo/vespa/config/server/filedistribution/FileDBRegistryTestCase.java +++ b/configserver/src/test/java/com/yahoo/vespa/config/server/filedistribution/FileDBRegistryTestCase.java @@ -3,7 +3,6 @@ package com.yahoo.vespa.config.server.filedistribution; import com.yahoo.config.FileReference; import com.yahoo.config.application.api.FileRegistry; -import com.yahoo.vespa.flags.InMemoryFlagSource; import org.junit.Test; import org.junit.rules.TemporaryFolder; import java.io.File; @@ -45,7 +44,7 @@ public class FileDBRegistryTestCase { TemporaryFolder tmpDir = new TemporaryFolder(); tmpDir.create(); AddFileInterface fileManager = - new ApplicationFileManager(new File(APP), new FileDirectory(tmpDir.newFolder(), new InMemoryFlagSource()), false); + new ApplicationFileManager(new File(APP), new FileDirectory(tmpDir.newFolder()), false); FileRegistry fileRegistry = new FileDBRegistry(fileManager); assertEquals(FOO_REF, fileRegistry.addFile(FOO_FILE)); try { diff --git a/configserver/src/test/java/com/yahoo/vespa/config/server/filedistribution/FileDirectoryTest.java b/configserver/src/test/java/com/yahoo/vespa/config/server/filedistribution/FileDirectoryTest.java index b346ef65bba..649d382ddb6 100644 --- a/configserver/src/test/java/com/yahoo/vespa/config/server/filedistribution/FileDirectoryTest.java +++ b/configserver/src/test/java/com/yahoo/vespa/config/server/filedistribution/FileDirectoryTest.java @@ -4,12 +4,10 @@ package com.yahoo.vespa.config.server.filedistribution; import com.yahoo.config.FileReference; import com.yahoo.io.IOUtils; -import com.yahoo.vespa.flags.InMemoryFlagSource; import org.junit.Before; import org.junit.Rule; import org.junit.Test; import org.junit.rules.TemporaryFolder; - import java.io.File; import java.io.IOException; import java.nio.file.Files; @@ -31,7 +29,7 @@ public class FileDirectoryTest { @Before public void setup() { - fileDirectory = new FileDirectory(temporaryFolder.getRoot(), new InMemoryFlagSource()); + fileDirectory = new FileDirectory(temporaryFolder.getRoot()); } @Test diff --git a/configserver/src/test/java/com/yahoo/vespa/config/server/filedistribution/FileServerTest.java b/configserver/src/test/java/com/yahoo/vespa/config/server/filedistribution/FileServerTest.java index cf86ff06333..3c9ea238479 100644 --- a/configserver/src/test/java/com/yahoo/vespa/config/server/filedistribution/FileServerTest.java +++ b/configserver/src/test/java/com/yahoo/vespa/config/server/filedistribution/FileServerTest.java @@ -41,7 +41,7 @@ public class FileServerTest { @Before public void setup() throws IOException { File rootDir = new File(temporaryFolder.newFolder("fileserver-root").getAbsolutePath()); - fileServer = new FileServer(new MockFileDownloader(rootDir), List.of(gzip, lz4), new FileDirectory(rootDir, new InMemoryFlagSource())); + fileServer = new FileServer(new MockFileDownloader(rootDir), List.of(gzip, lz4), new FileDirectory(rootDir)); } @Test @@ -86,7 +86,7 @@ public class FileServerTest { @Test public void requireThatWeCanReplayDirWithLz4() throws IOException, InterruptedException, ExecutionException { File rootDir = new File(temporaryFolder.newFolder("fileserver-root-3").getAbsolutePath()); - fileServer = new FileServer(new MockFileDownloader(rootDir), List.of(lz4, gzip), new FileDirectory(rootDir, new InMemoryFlagSource())); // prefer lz4 + fileServer = new FileServer(new MockFileDownloader(rootDir), List.of(lz4, gzip), new FileDirectory(rootDir)); // prefer lz4 File dir = getFileServerRootDir(); IOUtils.writeFile(dir + "/subdir/12z/f1", "dummy-data-2", true); CompletableFuture<byte []> content = new CompletableFuture<>(); @@ -143,7 +143,7 @@ public class FileServerTest { InMemoryFlagSource flagSource = new InMemoryFlagSource(); return new FileServer(new ConfigserverConfig(configBuilder), flagSource, - new FileDirectory(fileReferencesDir, flagSource)); + new FileDirectory(fileReferencesDir)); } private static class FileReceiver implements FileServer.Receiver { diff --git a/configserver/src/test/java/com/yahoo/vespa/config/server/http/v2/ApplicationHandlerTest.java b/configserver/src/test/java/com/yahoo/vespa/config/server/http/v2/ApplicationHandlerTest.java index b24d37f29fd..9251bbfcf39 100644 --- a/configserver/src/test/java/com/yahoo/vespa/config/server/http/v2/ApplicationHandlerTest.java +++ b/configserver/src/test/java/com/yahoo/vespa/config/server/http/v2/ApplicationHandlerTest.java @@ -43,7 +43,6 @@ import com.yahoo.vespa.config.server.session.PrepareParams; import com.yahoo.vespa.config.server.tenant.Tenant; import com.yahoo.vespa.config.server.tenant.TenantRepository; import com.yahoo.vespa.config.server.tenant.TestTenantRepository; -import com.yahoo.vespa.flags.InMemoryFlagSource; import org.junit.After; import org.junit.Before; import org.junit.Rule; @@ -126,7 +125,7 @@ public class ApplicationHandlerTest { tenantRepository = new TestTenantRepository.Builder() .withClock(clock) .withConfigserverConfig(configserverConfig) - .withFileDistributionFactory(new MockFileDistributionFactory(configserverConfig, new FileDirectory(configserverConfig, new InMemoryFlagSource()))) + .withFileDistributionFactory(new MockFileDistributionFactory(configserverConfig, new FileDirectory(configserverConfig))) .withHostProvisionerProvider(HostProvisionerProvider.withProvisioner(provisioner, false)) .withModelFactoryRegistry(new ModelFactoryRegistry(modelFactories)) .build(); diff --git a/configserver/src/test/java/com/yahoo/vespa/config/server/rpc/MockRpcServer.java b/configserver/src/test/java/com/yahoo/vespa/config/server/rpc/MockRpcServer.java index 8400f677124..0bf6012a668 100644 --- a/configserver/src/test/java/com/yahoo/vespa/config/server/rpc/MockRpcServer.java +++ b/configserver/src/test/java/com/yahoo/vespa/config/server/rpc/MockRpcServer.java @@ -11,8 +11,6 @@ import com.yahoo.vespa.config.server.filedistribution.FileServer; import com.yahoo.vespa.config.server.host.HostRegistry; import com.yahoo.vespa.config.server.monitoring.Metrics; import com.yahoo.vespa.config.server.rpc.security.NoopRpcAuthorizer; -import com.yahoo.vespa.flags.InMemoryFlagSource; - import java.io.File; import java.time.Duration; import java.time.Instant; @@ -39,7 +37,7 @@ public class MockRpcServer extends RpcServer { null, Metrics.createTestMetrics(), new HostRegistry(), - new FileServer(new FileDirectory(tempDir, new InMemoryFlagSource())), + new FileServer(new FileDirectory(tempDir)), new NoopRpcAuthorizer(), new RpcRequestHandlerProvider()); } diff --git a/configserver/src/test/java/com/yahoo/vespa/config/server/rpc/RpcTester.java b/configserver/src/test/java/com/yahoo/vespa/config/server/rpc/RpcTester.java index 651a56dfb25..e0b7909f032 100644 --- a/configserver/src/test/java/com/yahoo/vespa/config/server/rpc/RpcTester.java +++ b/configserver/src/test/java/com/yahoo/vespa/config/server/rpc/RpcTester.java @@ -9,7 +9,6 @@ import com.yahoo.jrt.Request; import com.yahoo.jrt.Spec; import com.yahoo.jrt.Supervisor; import com.yahoo.jrt.Transport; -import com.yahoo.net.HostName; import com.yahoo.test.ManualClock; import com.yahoo.vespa.config.server.ApplicationRepository; import com.yahoo.vespa.config.server.MemoryGenerationCounter; @@ -30,7 +29,6 @@ import com.yahoo.vespa.config.server.tenant.TestTenantRepository; import com.yahoo.vespa.flags.InMemoryFlagSource; import org.junit.After; import org.junit.rules.TemporaryFolder; - import java.io.IOException; import java.time.Duration; import java.time.Instant; @@ -49,7 +47,6 @@ import static org.junit.Assert.assertTrue; public class RpcTester implements AutoCloseable { private final ManualClock clock = new ManualClock(Instant.ofEpochMilli(100)); - private final String myHostname = HostName.getLocalhost(); private final Spec spec; private final RpcServer rpcServer; @@ -120,7 +117,7 @@ public class RpcTester implements AutoCloseable { flagSource)), Metrics.createTestMetrics(), hostRegistry, - new FileServer(new FileDirectory(temporaryFolder.newFolder(), flagSource)), + new FileServer(new FileDirectory(temporaryFolder.newFolder())), new NoopRpcAuthorizer(), new RpcRequestHandlerProvider()); rpcServer.setUpGetConfigHandlers(); diff --git a/configserver/src/test/java/com/yahoo/vespa/config/server/session/SessionPreparerTest.java b/configserver/src/test/java/com/yahoo/vespa/config/server/session/SessionPreparerTest.java index 79a6a3464ce..871338931fe 100644 --- a/configserver/src/test/java/com/yahoo/vespa/config/server/session/SessionPreparerTest.java +++ b/configserver/src/test/java/com/yahoo/vespa/config/server/session/SessionPreparerTest.java @@ -125,7 +125,7 @@ public class SessionPreparerTest { HostProvisionerProvider hostProvisionerProvider) { return new SessionPreparer( modelFactoryRegistry, - new MockFileDistributionFactory(configserverConfig, new FileDirectory(configserverConfig, flagSource)), + new MockFileDistributionFactory(configserverConfig, new FileDirectory(configserverConfig)), new InThreadExecutorService(), hostProvisionerProvider, configserverConfig, diff --git a/configserver/src/test/java/com/yahoo/vespa/config/server/session/SessionRepositoryTest.java b/configserver/src/test/java/com/yahoo/vespa/config/server/session/SessionRepositoryTest.java index af7754fa995..dc3441237ab 100644 --- a/configserver/src/test/java/com/yahoo/vespa/config/server/session/SessionRepositoryTest.java +++ b/configserver/src/test/java/com/yahoo/vespa/config/server/session/SessionRepositoryTest.java @@ -99,7 +99,7 @@ public class SessionRepositoryTest { .withCurator(curator) .withFlagSource(flagSource) .withFileDistributionFactory( - new MockFileDistributionFactory(configserverConfig, new FileDirectory(configserverConfig, flagSource))) + new MockFileDistributionFactory(configserverConfig, new FileDirectory(configserverConfig))) .withModelFactoryRegistry(modelFactoryRegistry) .build(); tenantRepository.addTenant(SessionRepositoryTest.tenantName); diff --git a/configserver/src/test/java/com/yahoo/vespa/config/server/tenant/TenantRepositoryTest.java b/configserver/src/test/java/com/yahoo/vespa/config/server/tenant/TenantRepositoryTest.java index 89279ceea1e..823466603b1 100644 --- a/configserver/src/test/java/com/yahoo/vespa/config/server/tenant/TenantRepositoryTest.java +++ b/configserver/src/test/java/com/yahoo/vespa/config/server/tenant/TenantRepositoryTest.java @@ -215,7 +215,7 @@ public class TenantRepositoryTest { Metrics.createTestMetrics(), new StripedExecutor<>(new InThreadExecutorService()), new StripedExecutor<>(new InThreadExecutorService()), - new FileDistributionFactory(configserverConfig, new FileDirectory(configserverConfig, flagSource)), + new FileDistributionFactory(configserverConfig, new FileDirectory(configserverConfig)), flagSource, new InThreadExecutorService(), new MockSecretStore(), diff --git a/configserver/src/test/java/com/yahoo/vespa/config/server/tenant/TestTenantRepository.java b/configserver/src/test/java/com/yahoo/vespa/config/server/tenant/TestTenantRepository.java index f6935c25903..dd982ccbd72 100644 --- a/configserver/src/test/java/com/yahoo/vespa/config/server/tenant/TestTenantRepository.java +++ b/configserver/src/test/java/com/yahoo/vespa/config/server/tenant/TestTenantRepository.java @@ -144,8 +144,7 @@ public class TestTenantRepository extends TenantRepository { public TenantRepository build() { if (fileDistributionFactory == null) - fileDistributionFactory = new FileDistributionFactory(configserverConfig, - new FileDirectory(configserverConfig, flagSource)); + fileDistributionFactory = new FileDistributionFactory(configserverConfig, new FileDirectory(configserverConfig)); return new TestTenantRepository(hostRegistry, curator, metrics, |