diff --git a/momento-sdk/src/intTest/java/momento/sdk/AuthClientCacheTests.java b/momento-sdk/src/intTest/java/momento/sdk/AuthClientCacheTests.java index 6cfc36e0..dd4ae7e8 100644 --- a/momento-sdk/src/intTest/java/momento/sdk/AuthClientCacheTests.java +++ b/momento-sdk/src/intTest/java/momento/sdk/AuthClientCacheTests.java @@ -28,7 +28,7 @@ import org.junit.jupiter.api.BeforeAll; import org.junit.jupiter.api.Test; -public class AuthClientCacheTests extends BaseTestClass { +public class AuthClientCacheTests extends BaseCacheTestClass { private static AuthClient authClient; String key = "test-key"; diff --git a/momento-sdk/src/intTest/java/momento/sdk/AuthClientTopicTests.java b/momento-sdk/src/intTest/java/momento/sdk/AuthClientTopicTests.java index 08933926..096b8a04 100644 --- a/momento-sdk/src/intTest/java/momento/sdk/AuthClientTopicTests.java +++ b/momento-sdk/src/intTest/java/momento/sdk/AuthClientTopicTests.java @@ -13,7 +13,7 @@ import org.junit.jupiter.api.BeforeAll; import org.junit.jupiter.api.Test; -public class AuthClientTopicTests extends BaseTestClass { +public class AuthClientTopicTests extends BaseCacheTestClass { private static AuthClient authClient; private static String topicName = "topic"; diff --git a/momento-sdk/src/intTest/java/momento/sdk/BaseTestClass.java b/momento-sdk/src/intTest/java/momento/sdk/BaseCacheTestClass.java similarity index 98% rename from momento-sdk/src/intTest/java/momento/sdk/BaseTestClass.java rename to momento-sdk/src/intTest/java/momento/sdk/BaseCacheTestClass.java index 6d682a8e..76209592 100644 --- a/momento-sdk/src/intTest/java/momento/sdk/BaseTestClass.java +++ b/momento-sdk/src/intTest/java/momento/sdk/BaseCacheTestClass.java @@ -9,7 +9,7 @@ import org.junit.jupiter.api.AfterAll; import org.junit.jupiter.api.BeforeAll; -public class BaseTestClass { +public class BaseCacheTestClass { protected static final Duration DEFAULT_TTL_SECONDS = Duration.ofSeconds(60); protected static final Duration FIVE_SECONDS = Duration.ofSeconds(5); protected static CredentialProvider credentialProvider; diff --git a/momento-sdk/src/intTest/java/momento/sdk/BaseStorageTestClass.java b/momento-sdk/src/intTest/java/momento/sdk/BaseStorageTestClass.java new file mode 100644 index 00000000..1172da5d --- /dev/null +++ b/momento-sdk/src/intTest/java/momento/sdk/BaseStorageTestClass.java @@ -0,0 +1,55 @@ +package momento.sdk; + +import java.time.Duration; +import java.util.UUID; +import momento.sdk.auth.CredentialProvider; +import momento.sdk.config.StorageConfigurations; +import momento.sdk.responses.storage.CreateStoreResponse; +import momento.sdk.responses.storage.DeleteStoreResponse; +import org.junit.jupiter.api.AfterAll; +import org.junit.jupiter.api.BeforeAll; + +public class BaseStorageTestClass { + protected static final Duration TEN_SECONDS = Duration.ofSeconds(10); + protected static CredentialProvider credentialProvider; + protected static PreviewStorageClient storageClient; + protected static String storeName; + + @BeforeAll + static void beforeAll() { + credentialProvider = CredentialProvider.fromEnvVar("TEST_AUTH_TOKEN"); + storageClient = + new PreviewStorageClientBuilder() + .withCredentialProvider(credentialProvider) + .withConfiguration(StorageConfigurations.Laptop.latest()) + .build(); + storeName = testStoreName(); + ensureTestStoreExists(storeName); + } + + @AfterAll + static void afterAll() { + cleanupTestStore(storeName); + storageClient.close(); + } + + protected static void ensureTestStoreExists(String storeName) { + CreateStoreResponse response = storageClient.createStore(storeName).join(); + if (response instanceof CreateStoreResponse.Error) { + throw new RuntimeException( + "Failed to test create store: " + ((CreateStoreResponse.Error) response).getMessage()); + } + } + + public static void cleanupTestStore(String storeName) { + DeleteStoreResponse response = storageClient.deleteStore(storeName).join(); + if (response instanceof DeleteStoreResponse.Error) { + throw new RuntimeException( + "Failed to test delete store: " + ((DeleteStoreResponse.Error) response).getMessage()); + } + } + + public static String testStoreName() { + return "java-integration-test-default-" + UUID.randomUUID(); + } +} diff --git a/momento-sdk/src/intTest/java/momento/sdk/CacheClientTest.java b/momento-sdk/src/intTest/java/momento/sdk/CacheClientTest.java index 7a4c39fe..1018a13a 100644 --- a/momento-sdk/src/intTest/java/momento/sdk/CacheClientTest.java +++ b/momento-sdk/src/intTest/java/momento/sdk/CacheClientTest.java @@ -33,7 +33,7 @@ import org.junit.jupiter.api.Test; /** Just includes a happy test path that interacts with both control and data plane clients. */ -final class CacheClientTest extends BaseTestClass { +final class CacheClientTest extends BaseCacheTestClass { private static final String JWT_HEADER_BASE64 = "eyJhbGciOiJIUzUxMiJ9"; private static final String JWT_INVALID_SIGNATURE_BASE64 = "gdghdjjfjyehhdkkkskskmmls76573jnajhjjjhjdhnndy"; diff --git a/momento-sdk/src/intTest/java/momento/sdk/CacheControlPlaneTest.java b/momento-sdk/src/intTest/java/momento/sdk/CacheControlPlaneTest.java index fc79fc73..5c5e03d2 100644 --- a/momento-sdk/src/intTest/java/momento/sdk/CacheControlPlaneTest.java +++ b/momento-sdk/src/intTest/java/momento/sdk/CacheControlPlaneTest.java @@ -21,7 +21,7 @@ import org.assertj.core.api.InstanceOfAssertFactories; import org.junit.jupiter.api.Test; -final class CacheControlPlaneTest extends BaseTestClass { +final class CacheControlPlaneTest extends BaseCacheTestClass { @Test public void createListRevokeSigningKeyWorks() { final SigningKeyCreateResponse signingKeyCreateResponse = diff --git a/momento-sdk/src/intTest/java/momento/sdk/CacheDataPlaneClientSideTest.java b/momento-sdk/src/intTest/java/momento/sdk/CacheDataPlaneClientSideTest.java index 4489b974..586e5f47 100644 --- a/momento-sdk/src/intTest/java/momento/sdk/CacheDataPlaneClientSideTest.java +++ b/momento-sdk/src/intTest/java/momento/sdk/CacheDataPlaneClientSideTest.java @@ -10,7 +10,7 @@ import org.junit.jupiter.api.Test; /** Tests client side exceptions */ -final class CacheDataPlaneClientSideTest extends BaseTestClass { +final class CacheDataPlaneClientSideTest extends BaseCacheTestClass { @Test public void nullKeyGetReturnsError() { final GetResponse stringResponse = cacheClient.get(cacheName, (String) null).join(); diff --git a/momento-sdk/src/intTest/java/momento/sdk/CacheDataPlaneEagerConnectionTest.java b/momento-sdk/src/intTest/java/momento/sdk/CacheDataPlaneEagerConnectionTest.java index eb4c555e..60008a21 100644 --- a/momento-sdk/src/intTest/java/momento/sdk/CacheDataPlaneEagerConnectionTest.java +++ b/momento-sdk/src/intTest/java/momento/sdk/CacheDataPlaneEagerConnectionTest.java @@ -1,6 +1,5 @@ package momento.sdk; -import static momento.sdk.BaseTestClass.credentialProvider; import static momento.sdk.TestUtils.randomString; import static org.assertj.core.api.Assertions.assertThat; @@ -10,7 +9,7 @@ import momento.sdk.responses.cache.SetResponse; import org.junit.jupiter.api.Test; -public class CacheDataPlaneEagerConnectionTest extends BaseTestClass { +public class CacheDataPlaneEagerConnectionTest extends BaseCacheTestClass { @Test void getReturnsHitAfterSet() { CacheClient client = diff --git a/momento-sdk/src/intTest/java/momento/sdk/CacheDataPlaneTest.java b/momento-sdk/src/intTest/java/momento/sdk/CacheDataPlaneTest.java index 4772c18a..ab7b92f9 100644 --- a/momento-sdk/src/intTest/java/momento/sdk/CacheDataPlaneTest.java +++ b/momento-sdk/src/intTest/java/momento/sdk/CacheDataPlaneTest.java @@ -16,7 +16,7 @@ import org.junit.jupiter.api.Test; /** Tests with Async APIs. */ -final class CacheDataPlaneTest extends BaseTestClass { +final class CacheDataPlaneTest extends BaseCacheTestClass { @Test void getReturnsHitAfterSet() { final String key = randomString("key"); diff --git a/momento-sdk/src/intTest/java/momento/sdk/DictionaryTest.java b/momento-sdk/src/intTest/java/momento/sdk/DictionaryTest.java index 3a643749..5a7aa62c 100644 --- a/momento-sdk/src/intTest/java/momento/sdk/DictionaryTest.java +++ b/momento-sdk/src/intTest/java/momento/sdk/DictionaryTest.java @@ -23,7 +23,7 @@ import org.assertj.core.api.InstanceOfAssertFactories; import org.junit.jupiter.api.Test; -public class DictionaryTest extends BaseTestClass { +public class DictionaryTest extends BaseCacheTestClass { Map stringStringMap = new HashMap<>(); Map stringBytesMap = new HashMap<>(); Map bytesStringMap = new HashMap<>(); diff --git a/momento-sdk/src/intTest/java/momento/sdk/ListTest.java b/momento-sdk/src/intTest/java/momento/sdk/ListTest.java index ac217c2b..bc06ebd8 100644 --- a/momento-sdk/src/intTest/java/momento/sdk/ListTest.java +++ b/momento-sdk/src/intTest/java/momento/sdk/ListTest.java @@ -22,7 +22,7 @@ import org.assertj.core.api.InstanceOfAssertFactories; import org.junit.jupiter.api.Test; -public class ListTest extends BaseTestClass { +public class ListTest extends BaseCacheTestClass { private final List values = Arrays.asList("val1", "val2", "val3", "val4"); @Test diff --git a/momento-sdk/src/intTest/java/momento/sdk/MomentoBatchUtilsIntegrationTest.java b/momento-sdk/src/intTest/java/momento/sdk/MomentoBatchUtilsIntegrationTest.java index 53ff2bb7..937f116b 100644 --- a/momento-sdk/src/intTest/java/momento/sdk/MomentoBatchUtilsIntegrationTest.java +++ b/momento-sdk/src/intTest/java/momento/sdk/MomentoBatchUtilsIntegrationTest.java @@ -17,7 +17,7 @@ import org.junit.jupiter.api.BeforeAll; import org.junit.jupiter.api.Test; -public class MomentoBatchUtilsIntegrationTest extends BaseTestClass { +public class MomentoBatchUtilsIntegrationTest extends BaseCacheTestClass { private static MomentoBatchUtils momentoBatchUtils; @BeforeAll diff --git a/momento-sdk/src/intTest/java/momento/sdk/SetTest.java b/momento-sdk/src/intTest/java/momento/sdk/SetTest.java index 6fe9b3c4..42258126 100644 --- a/momento-sdk/src/intTest/java/momento/sdk/SetTest.java +++ b/momento-sdk/src/intTest/java/momento/sdk/SetTest.java @@ -17,7 +17,7 @@ import org.assertj.core.api.InstanceOfAssertFactories; import org.junit.jupiter.api.Test; -public class SetTest extends BaseTestClass { +public class SetTest extends BaseCacheTestClass { @Test public void setAddElementStringHappyPath() { final String setName = randomString(); diff --git a/momento-sdk/src/intTest/java/momento/sdk/SortedSetTest.java b/momento-sdk/src/intTest/java/momento/sdk/SortedSetTest.java index 7b19051a..08e8d381 100644 --- a/momento-sdk/src/intTest/java/momento/sdk/SortedSetTest.java +++ b/momento-sdk/src/intTest/java/momento/sdk/SortedSetTest.java @@ -29,7 +29,7 @@ import org.assertj.core.api.InstanceOfAssertFactories; import org.junit.jupiter.api.Test; -public class SortedSetTest extends BaseTestClass { +public class SortedSetTest extends BaseCacheTestClass { // sortedSetPutElement @Test diff --git a/momento-sdk/src/intTest/java/momento/sdk/TopicClientTest.java b/momento-sdk/src/intTest/java/momento/sdk/TopicClientTest.java index 60ae090c..9aabdabb 100644 --- a/momento-sdk/src/intTest/java/momento/sdk/TopicClientTest.java +++ b/momento-sdk/src/intTest/java/momento/sdk/TopicClientTest.java @@ -18,7 +18,7 @@ import org.slf4j.Logger; import org.slf4j.LoggerFactory; -public class TopicClientTest extends BaseTestClass { +public class TopicClientTest extends BaseCacheTestClass { private static TopicClient topicClient; private final String topicName = "test-topic"; diff --git a/momento-sdk/src/intTest/java/momento/sdk/storage/ControlTests.java b/momento-sdk/src/intTest/java/momento/sdk/storage/ControlTests.java index 1cc31383..abdf134c 100644 --- a/momento-sdk/src/intTest/java/momento/sdk/storage/ControlTests.java +++ b/momento-sdk/src/intTest/java/momento/sdk/storage/ControlTests.java @@ -3,8 +3,7 @@ import static momento.sdk.TestUtils.randomString; import static org.assertj.core.api.Assertions.assertThat; -import java.time.Duration; -import momento.sdk.BaseTestClass; +import momento.sdk.BaseStorageTestClass; import momento.sdk.PreviewStorageClient; import momento.sdk.auth.CredentialProvider; import momento.sdk.config.StorageConfigurations; @@ -16,45 +15,21 @@ import momento.sdk.responses.storage.DeleteStoreResponse; import momento.sdk.responses.storage.ListStoresResponse; import org.assertj.core.api.InstanceOfAssertFactories; -import org.junit.jupiter.api.AfterAll; -import org.junit.jupiter.api.BeforeAll; import org.junit.jupiter.api.Test; -public class ControlTests extends BaseTestClass { - private static PreviewStorageClient client; - - public static final Duration TEN_SECONDS = Duration.ofSeconds(10); - - @BeforeAll - static void setup() { - client = - new PreviewStorageClient( - CredentialProvider.fromEnvVar("MOMENTO_API_KEY"), - StorageConfigurations.Laptop.latest()); - - // TODO re-using this name - client.createStore(System.getenv("TEST_CACHE_NAME")).join(); - } - - @AfterAll - static void tearDown() { - client.close(); - } - +public class ControlTests extends BaseStorageTestClass { @Test public void returnsAlreadyExistsWhenCreatingExistingStore() { - // TODO externalize this - // TODO rename env var to something broader like TEST_RESOURCE_NAME - final String existingStore = System.getenv("TEST_CACHE_NAME"); + final String existingStore = storeName; - assertThat(client.createStore(existingStore)) + assertThat(storageClient.createStore(existingStore)) .succeedsWithin(TEN_SECONDS) .asInstanceOf(InstanceOfAssertFactories.type(CreateStoreResponse.AlreadyExists.class)); } @Test public void returnsNotFoundWhenDeletingUnknownStore() { - assertThat(client.deleteStore(randomString("name"))) + assertThat(storageClient.deleteStore(randomString("name"))) .succeedsWithin(TEN_SECONDS) .asInstanceOf(InstanceOfAssertFactories.type(DeleteStoreResponse.Error.class)) .satisfies(error -> assertThat(error).hasCauseInstanceOf(StoreNotFoundException.class)); @@ -64,12 +39,12 @@ public void returnsNotFoundWhenDeletingUnknownStore() { public void listsStoresHappyPath() { final String storeName = randomString("name"); - assertThat(client.createStore(storeName)) + assertThat(storageClient.createStore(storeName)) .succeedsWithin(TEN_SECONDS) .isInstanceOf(CreateStoreResponse.Success.class); try { - assertThat(client.listStores()) + assertThat(storageClient.listStores()) .succeedsWithin(TEN_SECONDS) .asInstanceOf(InstanceOfAssertFactories.type(ListStoresResponse.Success.class)) .satisfies( @@ -77,11 +52,11 @@ public void listsStoresHappyPath() { assertThat(success.getStores()) .anyMatch(storeInfo -> storeInfo.getName().equals(storeName))); - final ListStoresResponse response = client.listStores().join(); + final ListStoresResponse response = storageClient.listStores().join(); assertThat(response).isInstanceOf(ListStoresResponse.Success.class); } finally { // cleanup - assertThat(client.deleteStore(storeName)) + assertThat(storageClient.deleteStore(storeName)) .succeedsWithin(TEN_SECONDS) .isInstanceOf(DeleteStoreResponse.Success.class); } @@ -89,7 +64,7 @@ public void listsStoresHappyPath() { @Test public void returnsBadRequestForEmptyStoreName() { - assertThat(client.createStore(" ")) + assertThat(storageClient.createStore(" ")) .succeedsWithin(TEN_SECONDS) .asInstanceOf(InstanceOfAssertFactories.type(CreateStoreResponse.Error.class)) .satisfies(error -> assertThat(error).hasCauseInstanceOf(BadRequestException.class)); @@ -97,12 +72,12 @@ public void returnsBadRequestForEmptyStoreName() { @Test public void throwsValidationExceptionForNullStoreName() { - assertThat(client.createStore(null)) + assertThat(storageClient.createStore(null)) .succeedsWithin(TEN_SECONDS) .asInstanceOf(InstanceOfAssertFactories.type(CreateStoreResponse.Error.class)) .satisfies(error -> assertThat(error).hasCauseInstanceOf(InvalidArgumentException.class)); - assertThat(client.deleteStore(null)) + assertThat(storageClient.deleteStore(null)) .succeedsWithin(TEN_SECONDS) .asInstanceOf(InstanceOfAssertFactories.type(DeleteStoreResponse.Error.class)) .satisfies(error -> assertThat(error).hasCauseInstanceOf(InvalidArgumentException.class)); @@ -113,25 +88,25 @@ public void deleteSucceeds() { final String storeName = randomString("name"); try { - assertThat(client.createStore(storeName)) + assertThat(storageClient.createStore(storeName)) .succeedsWithin(TEN_SECONDS) .isInstanceOf(CreateStoreResponse.Success.class); - assertThat(client.createStore(storeName)) + assertThat(storageClient.createStore(storeName)) .succeedsWithin(TEN_SECONDS) .asInstanceOf(InstanceOfAssertFactories.type(CreateStoreResponse.AlreadyExists.class)); - assertThat(client.deleteStore(storeName)) + assertThat(storageClient.deleteStore(storeName)) .succeedsWithin(TEN_SECONDS) .isInstanceOf(DeleteStoreResponse.Success.class); - assertThat(client.deleteStore(storeName)) + assertThat(storageClient.deleteStore(storeName)) .succeedsWithin(TEN_SECONDS) .asInstanceOf(InstanceOfAssertFactories.type(DeleteStoreResponse.Error.class)) .satisfies(error -> assertThat(error).hasCauseInstanceOf(StoreNotFoundException.class)); } finally { // Just in case the second create or delete fails - client.deleteStore(storeName).join(); + storageClient.deleteStore(storeName).join(); } } @@ -144,23 +119,23 @@ public void returnsErrorForBadToken() { + "s76573jnajhjjjhjdhnndy"; final CredentialProvider badTokenProvider = CredentialProvider.fromString(badToken); - try (final PreviewStorageClient client = + try (final PreviewStorageClient storageClient = new PreviewStorageClient( CredentialProvider.fromString(badToken), - StorageConfigurations.Laptop.latest()) /*CacheClient.builder( + StorageConfigurations.Laptop.latest()) /*CacheStorageClient.builder( badTokenProvider, Configurations.Laptop.latest(), Duration.ofSeconds(10)) .build()*/) { - assertThat(client.createStore(storeName)) + assertThat(storageClient.createStore(storeName)) .succeedsWithin(TEN_SECONDS) .asInstanceOf(InstanceOfAssertFactories.type(CreateStoreResponse.Error.class)) .satisfies(error -> assertThat(error).hasCauseInstanceOf(AuthenticationException.class)); - assertThat(client.deleteStore(storeName)) + assertThat(storageClient.deleteStore(storeName)) .succeedsWithin(TEN_SECONDS) .asInstanceOf(InstanceOfAssertFactories.type(DeleteStoreResponse.Error.class)) .satisfies(error -> assertThat(error).hasCauseInstanceOf(AuthenticationException.class)); - assertThat(client.listStores()) + assertThat(storageClient.listStores()) .succeedsWithin(TEN_SECONDS) .asInstanceOf(InstanceOfAssertFactories.type(ListStoresResponse.Error.class)) .satisfies(error -> assertThat(error).hasCauseInstanceOf(AuthenticationException.class)); diff --git a/momento-sdk/src/intTest/java/momento/sdk/storage/DataTests.java b/momento-sdk/src/intTest/java/momento/sdk/storage/DataTests.java index 3db2e4cc..289cb66f 100644 --- a/momento-sdk/src/intTest/java/momento/sdk/storage/DataTests.java +++ b/momento-sdk/src/intTest/java/momento/sdk/storage/DataTests.java @@ -4,52 +4,26 @@ import static org.assertj.core.api.AssertionsForClassTypes.assertThat; import static org.junit.jupiter.api.Assertions.assertThrows; -import momento.sdk.BaseTestClass; -import momento.sdk.PreviewStorageClient; -import momento.sdk.auth.CredentialProvider; -import momento.sdk.config.StorageConfigurations; +import momento.sdk.BaseStorageTestClass; import momento.sdk.exceptions.ClientSdkException; import momento.sdk.exceptions.StoreNotFoundException; import momento.sdk.responses.storage.DeleteResponse; import momento.sdk.responses.storage.GetResponse; import momento.sdk.responses.storage.PutResponse; -import org.junit.jupiter.api.AfterAll; -import org.junit.jupiter.api.BeforeAll; import org.junit.jupiter.api.Test; -public class DataTests extends BaseTestClass { - private static PreviewStorageClient client; - - // TODO can set to the same value as the cache tests - // TODO rename env var for clarity to TEST_RESOURCE_NAME or similar - private final String storeName = System.getenv("TEST_CACHE_NAME"); - - @BeforeAll - static void setup() { - client = - new PreviewStorageClient( - CredentialProvider.fromEnvVar("MOMENTO_API_KEY"), - StorageConfigurations.Laptop.latest()); - - client.createStore(System.getenv("TEST_CACHE_NAME")).join(); - } - - @AfterAll - static void tearDown() { - client.close(); - } - +public class DataTests extends BaseStorageTestClass { @Test void getReturnsValueAsStringAfterPut() { final String key = randomString("key"); final String value = randomString("value"); // Successful Set - final PutResponse putResponse = client.put(storeName, key, value).join(); + final PutResponse putResponse = storageClient.put(storeName, key, value).join(); assertThat(putResponse).isInstanceOf(PutResponse.Success.class); // Successful Get - final GetResponse getResponse = client.get(storeName, key).join(); + final GetResponse getResponse = storageClient.get(storeName, key).join(); assertThat(getResponse).isInstanceOf(GetResponse.Found.class); assertThat(getResponse.valueWhenFound().get().getString().get()).isEqualTo(value); } @@ -60,11 +34,11 @@ void getReturnsValueAsByteArrayAfterPut() { final String value = randomString("value"); // Successful Set - final PutResponse putResponse = client.put(storeName, key, value.getBytes()).join(); + final PutResponse putResponse = storageClient.put(storeName, key, value.getBytes()).join(); assertThat(putResponse).isInstanceOf(PutResponse.Success.class); // Successful Get - final GetResponse getResponse = client.get(storeName, key).join(); + final GetResponse getResponse = storageClient.get(storeName, key).join(); assertThat(getResponse).isInstanceOf(GetResponse.Found.class); assertThat(getResponse.valueWhenFound().get().getByteArray().get()).isEqualTo(value.getBytes()); } @@ -75,11 +49,11 @@ void getReturnsValueAsLongAfterPut() { final long value = 42L; // Successful Set - final PutResponse putResponse = client.put(storeName, key, value).join(); + final PutResponse putResponse = storageClient.put(storeName, key, value).join(); assertThat(putResponse).isInstanceOf(PutResponse.Success.class); // Successful Get - final GetResponse getResponse = client.get(storeName, key).join(); + final GetResponse getResponse = storageClient.get(storeName, key).join(); assertThat(getResponse).isInstanceOf(GetResponse.Found.class); assertThat(getResponse.valueWhenFound().get().getLong().get()).isEqualTo(value); } @@ -90,11 +64,11 @@ void getReturnsValueAsDoubleAfterPut() { final double value = 3.14; // Successful Set - final PutResponse putResponse = client.put(storeName, key, value).join(); + final PutResponse putResponse = storageClient.put(storeName, key, value).join(); assertThat(putResponse).isInstanceOf(PutResponse.Success.class); // Successful Get - final GetResponse getResponse = client.get(storeName, key).join(); + final GetResponse getResponse = storageClient.get(storeName, key).join(); assertThat(getResponse).isInstanceOf(GetResponse.Found.class); assertThat(getResponse.valueWhenFound().get().getDouble().get()).isEqualTo(value); } @@ -102,7 +76,7 @@ void getReturnsValueAsDoubleAfterPut() { @Test void storeKeyNotFound() { // Get key that was not set - final GetResponse response = client.get(storeName, randomString("key")).join(); + final GetResponse response = storageClient.get(storeName, randomString("key")).join(); assertThat(response).isInstanceOf(GetResponse.NotFound.class); assert response.valueWhenFound().isEmpty(); assert response instanceof GetResponse.NotFound; @@ -113,11 +87,11 @@ void storeKeyNotFound() { public void badStoreNameReturnsError() { final String storeName = randomString("name"); - final GetResponse getResponse = client.get(storeName, "").join(); + final GetResponse getResponse = storageClient.get(storeName, "").join(); assertThat(getResponse).isInstanceOf(GetResponse.Error.class); assertThat(((GetResponse.Error) getResponse)).hasCauseInstanceOf(StoreNotFoundException.class); - final PutResponse putResponse = client.put(storeName, "", "").join(); + final PutResponse putResponse = storageClient.put(storeName, "", "").join(); assertThat(putResponse).isInstanceOf(PutResponse.Error.class); assertThat(((PutResponse.Error) putResponse)).hasCauseInstanceOf(StoreNotFoundException.class); } @@ -126,8 +100,8 @@ public void badStoreNameReturnsError() { public void allowEmptyKeyValuesOnGet() throws Exception { final String emptyKey = ""; final String emptyValue = ""; - client.put(storeName, emptyKey, emptyValue).get(); - final GetResponse response = client.get(storeName, emptyKey).get(); + storageClient.put(storeName, emptyKey, emptyValue).get(); + final GetResponse response = storageClient.get(storeName, emptyKey).get(); assertThat(response).isInstanceOf(GetResponse.Found.class); assert response.valueWhenFound().get().getString().get().isEmpty(); } @@ -137,15 +111,15 @@ public void deleteHappyPath() throws Exception { final String key = "key"; final String value = "value"; - client.put(storeName, key, value).get(); - final GetResponse getResponse = client.get(storeName, key).get(); + storageClient.put(storeName, key, value).get(); + final GetResponse getResponse = storageClient.get(storeName, key).get(); assertThat(getResponse).isInstanceOf(GetResponse.Found.class); assertThat(getResponse.valueWhenFound().get().getString().get()).isEqualTo(value); - final DeleteResponse deleteResponse = client.delete(storeName, key).get(); + final DeleteResponse deleteResponse = storageClient.delete(storeName, key).get(); assertThat(deleteResponse).isInstanceOf(DeleteResponse.Success.class); - final GetResponse getAfterDeleteResponse = client.get(storeName, key).get(); + final GetResponse getAfterDeleteResponse = storageClient.get(storeName, key).get(); assert getAfterDeleteResponse.valueWhenFound().isEmpty(); assert getAfterDeleteResponse instanceof GetResponse.NotFound; } @@ -154,7 +128,7 @@ public void deleteHappyPath() throws Exception { public void deleteNonExistentKey() throws Exception { final String key = randomString("key"); - final DeleteResponse deleteResponse = client.delete(storeName, key).get(); + final DeleteResponse deleteResponse = storageClient.delete(storeName, key).get(); assertThat(deleteResponse).isInstanceOf(DeleteResponse.Success.class); } } diff --git a/momento-sdk/src/main/java/momento/sdk/PreviewStorageClientBuilder.java b/momento-sdk/src/main/java/momento/sdk/PreviewStorageClientBuilder.java index cf61dfba..f5b601d6 100644 --- a/momento-sdk/src/main/java/momento/sdk/PreviewStorageClientBuilder.java +++ b/momento-sdk/src/main/java/momento/sdk/PreviewStorageClientBuilder.java @@ -49,7 +49,7 @@ public PreviewStorageClientBuilder withConfiguration( * * @return the client. */ - public IPreviewStorageClient build() { + public PreviewStorageClient build() { if (credentialProvider == null) { credentialProvider = new EnvVarCredentialProvider("MOMENTO_API_KEY"); }