Add GetObjectApiTest, PutObjectsApiTest, ListKeyVersionsApiTest

app/build.gradle

@@ -4,6 +4,7 @@ buildscript {
  ext.protobufVersion = '3.21.7'
  ext.jerseyVersion = '3.1.0'
  ext.junitVersion = '5.9.0'
+ ext.mockitoVersion = '5.2.0'
  ext.postgresVersion = '42.5.1'
  ext.jooqVersion = '3.17.7'
  ext.guiceVersion = '5.1.0'
@@ -33,6 +34,8 @@ version '1.0'
 dependencies {
  implementation "com.google.protobuf:protobuf-java:$protobufVersion"
 
+ implementation('org.glassfish.jersey.containers:jersey-container-servlet:3.1.0')
+
  //jOOQ & Postgres impl deps
  implementation "org.jooq:jooq:$jooqVersion"
  implementation "org.jooq:jooq-meta:$jooqVersion"
@@ -42,7 +45,12 @@ dependencies {
 
  implementation "com.google.inject:guice:$guiceVersion"
 
+ compileOnly 'org.projectlombok:lombok:1.18.24'
+ annotationProcessor 'org.projectlombok:lombok:1.18.24'
+
  testImplementation "org.junit.jupiter:junit-jupiter-api:$junitVersion"
+ testImplementation "org.junit.jupiter:junit-jupiter-params:$junitVersion"
+ testImplementation "org.mockito:mockito-core:$mockitoVersion"
  testRuntimeOnly "org.junit.jupiter:junit-jupiter-engine:$junitVersion"
  testImplementation "org.hamcrest:hamcrest-library:2.2"
  testImplementation "org.testcontainers:junit-jupiter:1.17.6"

app/src/main/java/org/vss/VSSApplication.java

@@ -0,0 +1,10 @@
+package org.vss;
+
+import jakarta.ws.rs.ApplicationPath;
+import org.glassfish.jersey.server.ResourceConfig;
+
+@ApplicationPath("/")
+public class VSSApplication extends ResourceConfig {
+ public VSSApplication() {
+ }
+}

app/src/main/java/org/vss/api/AbstractVssApi.java

@@ -0,0 +1,48 @@
+package org.vss.api;
+
+import com.google.protobuf.GeneratedMessageV3;
+import com.google.protobuf.InvalidProtocolBufferException;
+import jakarta.inject.Inject;
+import jakarta.ws.rs.core.Response;
+import org.vss.ErrorCode;
+import org.vss.ErrorResponse;
+import org.vss.KVStore;
+import org.vss.exception.ConflictException;
+
+public abstract class AbstractVssApi {
+ final KVStore kvStore;
+
+ @Inject
+ public AbstractVssApi(KVStore kvStore) {
+ this.kvStore = kvStore;
+ }
+
+ Response toResponse(GeneratedMessageV3 protoResponse) {
+
+ return Response
+ .status(Response.Status.OK)
+ .entity(protoResponse.toByteArray())
+ .build();
+ }
+
+ Response toErrorResponse(Exception e) {
+ ErrorCode errorCode;
+ if (e instanceof ConflictException) {
+ errorCode = ErrorCode.CONFLICT_EXCEPTION;
+ } else if (e instanceof IllegalArgumentException
+ || e instanceof InvalidProtocolBufferException) {
+ errorCode = ErrorCode.INVALID_REQUEST_EXCEPTION;
+ } else {
+ errorCode = ErrorCode.INTERNAL_SERVER_EXCEPTION;
+ }
+
+ ErrorResponse errorResponse = ErrorResponse.newBuilder()
+ .setErrorCode(errorCode)
+ .setMessage(e.getMessage())
+ .build();
+
+ return Response.status(errorCode.getNumber())
+ .entity(errorResponse.toByteArray())
+ .build();
+ }
+}

app/src/main/java/org/vss/api/GetObjectApi.java

@@ -0,0 +1,35 @@
+package org.vss.api;
+
+import jakarta.inject.Inject;
+import jakarta.ws.rs.POST;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.Produces;
+import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
+import lombok.extern.slf4j.Slf4j;
+import org.vss.GetObjectRequest;
+import org.vss.GetObjectResponse;
+import org.vss.KVStore;
+
+@Path(VssApiEndpoint.GET_OBJECT)
+@Slf4j
+public class GetObjectApi extends AbstractVssApi {
+
+ @Inject
+ public GetObjectApi(KVStore kvstore) {
+ super(kvstore);
+ }
+
+ @POST
+ @Produces(MediaType.APPLICATION_OCTET_STREAM)
+ public Response execute(byte[] payload) {
+ try {
+ GetObjectRequest request = GetObjectRequest.parseFrom(payload);
+ GetObjectResponse response = kvStore.get(request);
+ return toResponse(response);
+ } catch (Exception e) {
+ log.error("Exception in GetObjectApi: ", e);
+ return toErrorResponse(e);
+ }
+ }
+}

app/src/main/java/org/vss/api/ListKeyVersionsApi.java

@@ -0,0 +1,35 @@
+package org.vss.api;
+
+import jakarta.inject.Inject;
+import jakarta.ws.rs.POST;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.Produces;
+import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
+import lombok.extern.slf4j.Slf4j;
+import org.vss.KVStore;
+import org.vss.ListKeyVersionsRequest;
+import org.vss.ListKeyVersionsResponse;
+
+@Path(VssApiEndpoint.LIST_KEY_VERSIONS)
+@Slf4j
+public class ListKeyVersionsApi extends AbstractVssApi {
+
+ @Inject
+ public ListKeyVersionsApi(KVStore kvStore) {
+ super(kvStore);
+ }
+
+ @POST
+ @Produces(MediaType.APPLICATION_OCTET_STREAM)
+ public Response execute(byte[] payload) {
+ try {
+ ListKeyVersionsRequest request = ListKeyVersionsRequest.parseFrom(payload);
+ ListKeyVersionsResponse response = kvStore.listKeyVersions(request);
+ return toResponse(response);
+ } catch (Exception e) {
+ log.error("Exception in ListKeyVersionsApi: ", e);
+ return toErrorResponse(e);
+ }
+ }
+}

app/src/main/java/org/vss/api/PutObjectsApi.java

@@ -0,0 +1,35 @@
+package org.vss.api;
+
+import jakarta.inject.Inject;
+import jakarta.ws.rs.POST;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.Produces;
+import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
+import lombok.extern.slf4j.Slf4j;
+import org.vss.KVStore;
+import org.vss.PutObjectRequest;
+import org.vss.PutObjectResponse;
+
+@Path(VssApiEndpoint.PUT_OBJECTS)
+@Slf4j
+public class PutObjectsApi extends AbstractVssApi {
+
+ @Inject
+ public PutObjectsApi(KVStore kvStore) {
+ super(kvStore);
+ }
+
+ @POST
+ @Produces(MediaType.APPLICATION_OCTET_STREAM)
+ public Response execute(byte[] payload) {
+ try {
+ PutObjectRequest putObjectRequest = PutObjectRequest.parseFrom(payload);
+ PutObjectResponse response = kvStore.put(putObjectRequest);
+ return toResponse(response);
+ } catch (Exception e) {
+ log.error("Exception in PutObjectsApi: ", e);
+ return toErrorResponse(e);
+ }
+ }
+}

app/src/main/java/org/vss/api/VssApiEndpoint.java

@@ -0,0 +1,7 @@
+package org.vss.api;
+
+public class VssApiEndpoint {
+ public static final String GET_OBJECT = "/getObject";
+ public static final String PUT_OBJECTS = "/putObjects";
+ public static final String LIST_KEY_VERSIONS = "/listKeyVersions";
+}

app/src/test/java/org/vss/api/GetObjectApiTest.java

@@ -0,0 +1,86 @@
+package org.vss.api;
+
+import com.google.protobuf.ByteString;
+import jakarta.ws.rs.core.Response;
+import java.nio.charset.StandardCharsets;
+import java.util.stream.Stream;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.Arguments;
+import org.junit.jupiter.params.provider.MethodSource;
+import org.vss.ErrorCode;
+import org.vss.ErrorResponse;
+import org.vss.GetObjectRequest;
+import org.vss.GetObjectResponse;
+import org.vss.KVStore;
+import org.vss.KeyValue;
+import org.vss.exception.ConflictException;
+
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.is;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+class GetObjectApiTest {
+ private GetObjectApi getObjectApi;
+ private KVStore mockKVStore;
+
+ private static String TEST_STORE_ID = "storeId";
+ private static String TEST_KEY = "key";
+ private static KeyValue TEST_KV = KeyValue.newBuilder().setKey(TEST_KEY).setValue(
+ ByteString.copyFrom("test_value", StandardCharsets.UTF_8)).build();
+
+ @BeforeEach
+ void setUp() {
+ mockKVStore = mock(KVStore.class);
+ getObjectApi = new GetObjectApi(mockKVStore);
+ }
+
+ @Test
+ void execute_ValidPayload_ReturnsResponse() {
+ GetObjectRequest expectedRequest =
+ GetObjectRequest.newBuilder().setStoreId(TEST_STORE_ID).setKey(TEST_KEY).build();
+ byte[] payload = expectedRequest.toByteArray();
+ GetObjectResponse mockResponse = GetObjectResponse.newBuilder().setValue(TEST_KV).build();
+ when(mockKVStore.get(expectedRequest)).thenReturn(mockResponse);
+
+ Response actualResponse = getObjectApi.execute(payload);
+
+ assertThat(actualResponse.getStatus(), is(Response.Status.OK.getStatusCode()));
+ assertThat(actualResponse.getEntity(), is(mockResponse.toByteArray()));
+ verify(mockKVStore).get(expectedRequest);
+ }
+
+ @ParameterizedTest
+ @MethodSource("provideErrorTestCases")
+ void execute_InvalidPayload_ReturnsErrorResponse(Exception exception,
+ ErrorCode errorCode) {
+ GetObjectRequest expectedRequest = GetObjectRequest.newBuilder()
+ .setStoreId(TEST_STORE_ID)
+ .setKey(TEST_KEY)
+ .build();
+ byte[] payload = expectedRequest.toByteArray();
+ when(mockKVStore.get(any())).thenThrow(exception);
+
+ Response response = getObjectApi.execute(payload);
+
+ ErrorResponse expectedErrorResponse = ErrorResponse.newBuilder()
+ .setErrorCode(errorCode)
+ .setMessage("")
+ .build();
+ assertThat(response.getEntity(), is(expectedErrorResponse.toByteArray()));
+ assertThat(response.getStatus(), is(expectedErrorResponse.getErrorCode().getNumber()));
+ verify(mockKVStore).get(expectedRequest);
+ }
+
+ private static Stream<Arguments> provideErrorTestCases() {
+ return Stream.of(
+ Arguments.of(new ConflictException(""), ErrorCode.CONFLICT_EXCEPTION),
+ Arguments.of(new IllegalArgumentException(""), ErrorCode.INVALID_REQUEST_EXCEPTION),
+ Arguments.of(new RuntimeException(""), ErrorCode.INTERNAL_SERVER_EXCEPTION)
+ );
+ }
+}

app/src/test/java/org/vss/api/ListKeyVersionsApiTest.java

@@ -0,0 +1,92 @@
+package org.vss.api;
+
+import com.google.protobuf.ByteString;
+import jakarta.ws.rs.core.Response;
+import java.nio.charset.StandardCharsets;
+import java.util.List;
+import java.util.stream.Stream;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.Arguments;
+import org.junit.jupiter.params.provider.MethodSource;
+import org.vss.ErrorCode;
+import org.vss.ErrorResponse;
+import org.vss.KVStore;
+import org.vss.KeyValue;
+import org.vss.ListKeyVersionsRequest;
+import org.vss.ListKeyVersionsResponse;
+import org.vss.exception.ConflictException;
+
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.is;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+public class ListKeyVersionsApiTest {
+ private ListKeyVersionsApi listKeyVersionsApi;
+ private KVStore mockKVStore;
+
+ private static String TEST_STORE_ID = "storeId";
+ private static String TEST_KEY = "key";
+ private static KeyValue TEST_KV = KeyValue.newBuilder().setKey(TEST_KEY).setValue(
+ ByteString.copyFrom("test_value", StandardCharsets.UTF_8)).build();
+
+ @BeforeEach
+ void setUp() {
+ mockKVStore = mock(KVStore.class);
+ listKeyVersionsApi = new ListKeyVersionsApi(mockKVStore);
+ }
+
+ @Test
+ void execute_ValidPayload_ReturnsResponse() {
+ ListKeyVersionsRequest expectedRequest =
+ ListKeyVersionsRequest.newBuilder()
+ .setStoreId(TEST_STORE_ID)
+ .setKeyPrefix(TEST_KEY)
+ .build();
+ byte[] payload = expectedRequest.toByteArray();
+ ListKeyVersionsResponse mockResponse = ListKeyVersionsResponse.newBuilder().addAllKeyVersions(
+ List.of(TEST_KV)).build();
+ when(mockKVStore.listKeyVersions(expectedRequest)).thenReturn(mockResponse);
+
+ Response actualResponse = listKeyVersionsApi.execute(payload);
+
+ assertThat(actualResponse.getStatus(), is(Response.Status.OK.getStatusCode()));
+ assertThat(actualResponse.getEntity(), is(mockResponse.toByteArray()));
+ verify(mockKVStore).listKeyVersions(expectedRequest);
+ }
+
+ @ParameterizedTest
+ @MethodSource("provideErrorTestCases")
+ void execute_InvalidPayload_ReturnsErrorResponse(Exception exception,
+ ErrorCode errorCode) {
+ ListKeyVersionsRequest expectedRequest =
+ ListKeyVersionsRequest.newBuilder()
+ .setStoreId(TEST_STORE_ID)
+ .setKeyPrefix(TEST_KEY)
+ .build();
+ byte[] payload = expectedRequest.toByteArray();
+ when(mockKVStore.listKeyVersions(any())).thenThrow(exception);
+
+ Response response = listKeyVersionsApi.execute(payload);
+
+ ErrorResponse expectedErrorResponse = ErrorResponse.newBuilder()
+ .setErrorCode(errorCode)
+ .setMessage("")
+ .build();
+ assertThat(response.getEntity(), is(expectedErrorResponse.toByteArray()));
+ assertThat(response.getStatus(), is(expectedErrorResponse.getErrorCode().getNumber()));
+ verify(mockKVStore).listKeyVersions(expectedRequest);
+ }
+
+ private static Stream<Arguments> provideErrorTestCases() {
+ return Stream.of(
+ Arguments.of(new ConflictException(""), ErrorCode.CONFLICT_EXCEPTION),
+ Arguments.of(new IllegalArgumentException(""), ErrorCode.INVALID_REQUEST_EXCEPTION),
+ Arguments.of(new RuntimeException(""), ErrorCode.INTERNAL_SERVER_EXCEPTION)
+ );
+ }
+}

app/build.gradle

@@ -4,6 +4,7 @@ buildscript {
  ext.protobufVersion = '3.21.7'
  ext.jerseyVersion = '3.1.0'
  ext.junitVersion = '5.9.0'
+ ext.mockitoVersion = '5.2.0'
  ext.postgresVersion = '42.5.1'
  ext.jooqVersion = '3.17.7'
  ext.guiceVersion = '5.1.0'
@@ -33,6 +34,8 @@ version '1.0'
 dependencies {
  implementation "com.google.protobuf:protobuf-java:$protobufVersion"
 
+ implementation('org.glassfish.jersey.containers:jersey-container-servlet:3.1.0')
+
  //jOOQ & Postgres impl deps
  implementation "org.jooq:jooq:$jooqVersion"
  implementation "org.jooq:jooq-meta:$jooqVersion"
@@ -42,7 +45,12 @@ dependencies {
 
  implementation "com.google.inject:guice:$guiceVersion"
 
+ compileOnly 'org.projectlombok:lombok:1.18.24'
+ annotationProcessor 'org.projectlombok:lombok:1.18.24'
+
  testImplementation "org.junit.jupiter:junit-jupiter-api:$junitVersion"
+ testImplementation "org.junit.jupiter:junit-jupiter-params:$junitVersion"
+ testImplementation "org.mockito:mockito-core:$mockitoVersion"
  testRuntimeOnly "org.junit.jupiter:junit-jupiter-engine:$junitVersion"
  testImplementation "org.hamcrest:hamcrest-library:2.2"
  testImplementation "org.testcontainers:junit-jupiter:1.17.6"

app/src/main/java/org/vss/VSSApplication.java

@@ -0,0 +1,10 @@
+package org.vss;
+
+import jakarta.ws.rs.ApplicationPath;
+import org.glassfish.jersey.server.ResourceConfig;
+
+@ApplicationPath("/")
+public class VSSApplication extends ResourceConfig {
+ public VSSApplication() {
+ }
+}

app/src/main/java/org/vss/api/AbstractVssApi.java

@@ -0,0 +1,48 @@
+package org.vss.api;
+
+import com.google.protobuf.GeneratedMessageV3;
+import com.google.protobuf.InvalidProtocolBufferException;
+import jakarta.inject.Inject;
+import jakarta.ws.rs.core.Response;
+import org.vss.ErrorCode;
+import org.vss.ErrorResponse;
+import org.vss.KVStore;
+import org.vss.exception.ConflictException;
+
+public abstract class AbstractVssApi {
+ final KVStore kvStore;
+
+ @Inject
+ public AbstractVssApi(KVStore kvStore) {
+ this.kvStore = kvStore;
+ }
+
+ Response toResponse(GeneratedMessageV3 protoResponse) {
+
+ return Response
+ .status(Response.Status.OK)
+ .entity(protoResponse.toByteArray())
+ .build();
+ }
+
+ Response toErrorResponse(Exception e) {
+ ErrorCode errorCode;
+ if (e instanceof ConflictException) {
+ errorCode = ErrorCode.CONFLICT_EXCEPTION;
+ } else if (e instanceof IllegalArgumentException
+ || e instanceof InvalidProtocolBufferException) {
+ errorCode = ErrorCode.INVALID_REQUEST_EXCEPTION;
+ } else {
+ errorCode = ErrorCode.INTERNAL_SERVER_EXCEPTION;
+ }
+
+ ErrorResponse errorResponse = ErrorResponse.newBuilder()
+ .setErrorCode(errorCode)
+ .setMessage(e.getMessage())
+ .build();
+
+ return Response.status(errorCode.getNumber())
+ .entity(errorResponse.toByteArray())
+ .build();
+ }
+}

app/src/main/java/org/vss/api/GetObjectApi.java

@@ -0,0 +1,35 @@
+package org.vss.api;
+
+import jakarta.inject.Inject;
+import jakarta.ws.rs.POST;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.Produces;
+import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
+import lombok.extern.slf4j.Slf4j;
+import org.vss.GetObjectRequest;
+import org.vss.GetObjectResponse;
+import org.vss.KVStore;
+
+@Path(VssApiEndpoint.GET_OBJECT)
+@Slf4j
+public class GetObjectApi extends AbstractVssApi {
+
+ @Inject
+ public GetObjectApi(KVStore kvstore) {
+ super(kvstore);
+ }
+
+ @POST
+ @Produces(MediaType.APPLICATION_OCTET_STREAM)
+ public Response execute(byte[] payload) {
+ try {
+ GetObjectRequest request = GetObjectRequest.parseFrom(payload);
+ GetObjectResponse response = kvStore.get(request);
+ return toResponse(response);
+ } catch (Exception e) {
+ log.error("Exception in GetObjectApi: ", e);
+ return toErrorResponse(e);
+ }
+ }
+}

app/src/main/java/org/vss/api/ListKeyVersionsApi.java

@@ -0,0 +1,35 @@
+package org.vss.api;
+
+import jakarta.inject.Inject;
+import jakarta.ws.rs.POST;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.Produces;
+import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
+import lombok.extern.slf4j.Slf4j;
+import org.vss.KVStore;
+import org.vss.ListKeyVersionsRequest;
+import org.vss.ListKeyVersionsResponse;
+
+@Path(VssApiEndpoint.LIST_KEY_VERSIONS)
+@Slf4j
+public class ListKeyVersionsApi extends AbstractVssApi {
+
+ @Inject
+ public ListKeyVersionsApi(KVStore kvStore) {
+ super(kvStore);
+ }
+
+ @POST
+ @Produces(MediaType.APPLICATION_OCTET_STREAM)
+ public Response execute(byte[] payload) {
+ try {
+ ListKeyVersionsRequest request = ListKeyVersionsRequest.parseFrom(payload);
+ ListKeyVersionsResponse response = kvStore.listKeyVersions(request);
+ return toResponse(response);
+ } catch (Exception e) {
+ log.error("Exception in ListKeyVersionsApi: ", e);
+ return toErrorResponse(e);
+ }
+ }
+}

app/src/main/java/org/vss/api/PutObjectsApi.java

@@ -0,0 +1,35 @@
+package org.vss.api;
+
+import jakarta.inject.Inject;
+import jakarta.ws.rs.POST;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.Produces;
+import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
+import lombok.extern.slf4j.Slf4j;
+import org.vss.KVStore;
+import org.vss.PutObjectRequest;
+import org.vss.PutObjectResponse;
+
+@Path(VssApiEndpoint.PUT_OBJECTS)
+@Slf4j
+public class PutObjectsApi extends AbstractVssApi {
+
+ @Inject
+ public PutObjectsApi(KVStore kvStore) {
+ super(kvStore);
+ }
+
+ @POST
+ @Produces(MediaType.APPLICATION_OCTET_STREAM)
+ public Response execute(byte[] payload) {
+ try {
+ PutObjectRequest putObjectRequest = PutObjectRequest.parseFrom(payload);
+ PutObjectResponse response = kvStore.put(putObjectRequest);
+ return toResponse(response);
+ } catch (Exception e) {
+ log.error("Exception in PutObjectsApi: ", e);
+ return toErrorResponse(e);
+ }
+ }
+}

app/src/main/java/org/vss/api/VssApiEndpoint.java

@@ -0,0 +1,7 @@
+package org.vss.api;
+
+public class VssApiEndpoint {
+ public static final String GET_OBJECT = "/getObject";
+ public static final String PUT_OBJECTS = "/putObjects";
+ public static final String LIST_KEY_VERSIONS = "/listKeyVersions";
+}

app/src/test/java/org/vss/api/GetObjectApiTest.java

@@ -0,0 +1,86 @@
+package org.vss.api;
+
+import com.google.protobuf.ByteString;
+import jakarta.ws.rs.core.Response;
+import java.nio.charset.StandardCharsets;
+import java.util.stream.Stream;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.Arguments;
+import org.junit.jupiter.params.provider.MethodSource;
+import org.vss.ErrorCode;
+import org.vss.ErrorResponse;
+import org.vss.GetObjectRequest;
+import org.vss.GetObjectResponse;
+import org.vss.KVStore;
+import org.vss.KeyValue;
+import org.vss.exception.ConflictException;
+
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.is;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+class GetObjectApiTest {
+ private GetObjectApi getObjectApi;
+ private KVStore mockKVStore;
+
+ private static String TEST_STORE_ID = "storeId";
+ private static String TEST_KEY = "key";
+ private static KeyValue TEST_KV = KeyValue.newBuilder().setKey(TEST_KEY).setValue(
+ ByteString.copyFrom("test_value", StandardCharsets.UTF_8)).build();
+
+ @BeforeEach
+ void setUp() {
+ mockKVStore = mock(KVStore.class);
+ getObjectApi = new GetObjectApi(mockKVStore);
+ }
+
+ @Test
+ void execute_ValidPayload_ReturnsResponse() {
+ GetObjectRequest expectedRequest =
+ GetObjectRequest.newBuilder().setStoreId(TEST_STORE_ID).setKey(TEST_KEY).build();
+ byte[] payload = expectedRequest.toByteArray();
+ GetObjectResponse mockResponse = GetObjectResponse.newBuilder().setValue(TEST_KV).build();
+ when(mockKVStore.get(expectedRequest)).thenReturn(mockResponse);
+
+ Response actualResponse = getObjectApi.execute(payload);
+
+ assertThat(actualResponse.getStatus(), is(Response.Status.OK.getStatusCode()));
+ assertThat(actualResponse.getEntity(), is(mockResponse.toByteArray()));
+ verify(mockKVStore).get(expectedRequest);
+ }
+
+ @ParameterizedTest
+ @MethodSource("provideErrorTestCases")
+ void execute_InvalidPayload_ReturnsErrorResponse(Exception exception,
+ ErrorCode errorCode) {
+ GetObjectRequest expectedRequest = GetObjectRequest.newBuilder()
+ .setStoreId(TEST_STORE_ID)
+ .setKey(TEST_KEY)
+ .build();
+ byte[] payload = expectedRequest.toByteArray();
+ when(mockKVStore.get(any())).thenThrow(exception);
+
+ Response response = getObjectApi.execute(payload);
+
+ ErrorResponse expectedErrorResponse = ErrorResponse.newBuilder()
+ .setErrorCode(errorCode)
+ .setMessage("")
+ .build();
+ assertThat(response.getEntity(), is(expectedErrorResponse.toByteArray()));
+ assertThat(response.getStatus(), is(expectedErrorResponse.getErrorCode().getNumber()));
+ verify(mockKVStore).get(expectedRequest);
+ }
+
+ private static Stream<Arguments> provideErrorTestCases() {
+ return Stream.of(
+ Arguments.of(new ConflictException(""), ErrorCode.CONFLICT_EXCEPTION),
+ Arguments.of(new IllegalArgumentException(""), ErrorCode.INVALID_REQUEST_EXCEPTION),
+ Arguments.of(new RuntimeException(""), ErrorCode.INTERNAL_SERVER_EXCEPTION)
+ );
+ }
+}

app/src/test/java/org/vss/api/ListKeyVersionsApiTest.java

@@ -0,0 +1,92 @@
+package org.vss.api;
+
+import com.google.protobuf.ByteString;
+import jakarta.ws.rs.core.Response;
+import java.nio.charset.StandardCharsets;
+import java.util.List;
+import java.util.stream.Stream;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.Arguments;
+import org.junit.jupiter.params.provider.MethodSource;
+import org.vss.ErrorCode;
+import org.vss.ErrorResponse;
+import org.vss.KVStore;
+import org.vss.KeyValue;
+import org.vss.ListKeyVersionsRequest;
+import org.vss.ListKeyVersionsResponse;
+import org.vss.exception.ConflictException;
+
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.is;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+public class ListKeyVersionsApiTest {
+ private ListKeyVersionsApi listKeyVersionsApi;
+ private KVStore mockKVStore;
+
+ private static String TEST_STORE_ID = "storeId";
+ private static String TEST_KEY = "key";
+ private static KeyValue TEST_KV = KeyValue.newBuilder().setKey(TEST_KEY).setValue(
+ ByteString.copyFrom("test_value", StandardCharsets.UTF_8)).build();
+
+ @BeforeEach
+ void setUp() {
+ mockKVStore = mock(KVStore.class);
+ listKeyVersionsApi = new ListKeyVersionsApi(mockKVStore);
+ }
+
+ @Test
+ void execute_ValidPayload_ReturnsResponse() {
+ ListKeyVersionsRequest expectedRequest =
+ ListKeyVersionsRequest.newBuilder()
+ .setStoreId(TEST_STORE_ID)
+ .setKeyPrefix(TEST_KEY)
+ .build();
+ byte[] payload = expectedRequest.toByteArray();
+ ListKeyVersionsResponse mockResponse = ListKeyVersionsResponse.newBuilder().addAllKeyVersions(
+ List.of(TEST_KV)).build();
+ when(mockKVStore.listKeyVersions(expectedRequest)).thenReturn(mockResponse);
+
+ Response actualResponse = listKeyVersionsApi.execute(payload);
+
+ assertThat(actualResponse.getStatus(), is(Response.Status.OK.getStatusCode()));
+ assertThat(actualResponse.getEntity(), is(mockResponse.toByteArray()));
+ verify(mockKVStore).listKeyVersions(expectedRequest);
+ }
+
+ @ParameterizedTest
+ @MethodSource("provideErrorTestCases")
+ void execute_InvalidPayload_ReturnsErrorResponse(Exception exception,
+ ErrorCode errorCode) {
+ ListKeyVersionsRequest expectedRequest =
+ ListKeyVersionsRequest.newBuilder()
+ .setStoreId(TEST_STORE_ID)
+ .setKeyPrefix(TEST_KEY)
+ .build();
+ byte[] payload = expectedRequest.toByteArray();
+ when(mockKVStore.listKeyVersions(any())).thenThrow(exception);
+
+ Response response = listKeyVersionsApi.execute(payload);
+
+ ErrorResponse expectedErrorResponse = ErrorResponse.newBuilder()
+ .setErrorCode(errorCode)
+ .setMessage("")
+ .build();
+ assertThat(response.getEntity(), is(expectedErrorResponse.toByteArray()));
+ assertThat(response.getStatus(), is(expectedErrorResponse.getErrorCode().getNumber()));
+ verify(mockKVStore).listKeyVersions(expectedRequest);
+ }
+
+ private static Stream<Arguments> provideErrorTestCases() {
+ return Stream.of(
+ Arguments.of(new ConflictException(""), ErrorCode.CONFLICT_EXCEPTION),
+ Arguments.of(new IllegalArgumentException(""), ErrorCode.INVALID_REQUEST_EXCEPTION),
+ Arguments.of(new RuntimeException(""), ErrorCode.INTERNAL_SERVER_EXCEPTION)
+ );
+ }
+}

app/build.gradle

@@ -4,6 +4,7 @@ buildscript {
  ext.protobufVersion = '3.21.7'
  ext.jerseyVersion = '3.1.0'
  ext.junitVersion = '5.9.0'
+ ext.mockitoVersion = '5.2.0'
  ext.postgresVersion = '42.5.1'
  ext.jooqVersion = '3.17.7'
  ext.guiceVersion = '5.1.0'
@@ -33,6 +34,8 @@ version '1.0'
 dependencies {
  implementation "com.google.protobuf:protobuf-java:$protobufVersion"
 
+ implementation('org.glassfish.jersey.containers:jersey-container-servlet:3.1.0')
+
  //jOOQ & Postgres impl deps
  implementation "org.jooq:jooq:$jooqVersion"
  implementation "org.jooq:jooq-meta:$jooqVersion"
@@ -42,7 +45,12 @@ dependencies {
 
  implementation "com.google.inject:guice:$guiceVersion"
 
+ compileOnly 'org.projectlombok:lombok:1.18.24'
+ annotationProcessor 'org.projectlombok:lombok:1.18.24'
+
  testImplementation "org.junit.jupiter:junit-jupiter-api:$junitVersion"
+ testImplementation "org.junit.jupiter:junit-jupiter-params:$junitVersion"
+ testImplementation "org.mockito:mockito-core:$mockitoVersion"
  testRuntimeOnly "org.junit.jupiter:junit-jupiter-engine:$junitVersion"
  testImplementation "org.hamcrest:hamcrest-library:2.2"
  testImplementation "org.testcontainers:junit-jupiter:1.17.6"

app/src/main/java/org/vss/VSSApplication.java

@@ -0,0 +1,10 @@
+package org.vss;
+
+import jakarta.ws.rs.ApplicationPath;
+import org.glassfish.jersey.server.ResourceConfig;
+
+@ApplicationPath("/")
+public class VSSApplication extends ResourceConfig {
+ public VSSApplication() {
+ }
+}

app/src/main/java/org/vss/api/AbstractVssApi.java

@@ -0,0 +1,48 @@
+package org.vss.api;
+
+import com.google.protobuf.GeneratedMessageV3;
+import com.google.protobuf.InvalidProtocolBufferException;
+import jakarta.inject.Inject;
+import jakarta.ws.rs.core.Response;
+import org.vss.ErrorCode;
+import org.vss.ErrorResponse;
+import org.vss.KVStore;
+import org.vss.exception.ConflictException;
+
+public abstract class AbstractVssApi {
+ final KVStore kvStore;
+
+ @Inject
+ public AbstractVssApi(KVStore kvStore) {
+ this.kvStore = kvStore;
+ }
+
+ Response toResponse(GeneratedMessageV3 protoResponse) {
+
+ return Response
+ .status(Response.Status.OK)
+ .entity(protoResponse.toByteArray())
+ .build();
+ }
+
+ Response toErrorResponse(Exception e) {
+ ErrorCode errorCode;
+ if (e instanceof ConflictException) {
+ errorCode = ErrorCode.CONFLICT_EXCEPTION;
+ } else if (e instanceof IllegalArgumentException
+ || e instanceof InvalidProtocolBufferException) {
+ errorCode = ErrorCode.INVALID_REQUEST_EXCEPTION;
+ } else {
+ errorCode = ErrorCode.INTERNAL_SERVER_EXCEPTION;
+ }
+
+ ErrorResponse errorResponse = ErrorResponse.newBuilder()
+ .setErrorCode(errorCode)
+ .setMessage(e.getMessage())
+ .build();
+
+ return Response.status(errorCode.getNumber())
+ .entity(errorResponse.toByteArray())
+ .build();
+ }
+}

app/src/main/java/org/vss/api/GetObjectApi.java

@@ -0,0 +1,35 @@
+package org.vss.api;
+
+import jakarta.inject.Inject;
+import jakarta.ws.rs.POST;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.Produces;
+import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
+import lombok.extern.slf4j.Slf4j;
+import org.vss.GetObjectRequest;
+import org.vss.GetObjectResponse;
+import org.vss.KVStore;
+
+@Path(VssApiEndpoint.GET_OBJECT)
+@Slf4j
+public class GetObjectApi extends AbstractVssApi {
+
+ @Inject
+ public GetObjectApi(KVStore kvstore) {
+ super(kvstore);
+ }
+
+ @POST
+ @Produces(MediaType.APPLICATION_OCTET_STREAM)
+ public Response execute(byte[] payload) {
+ try {
+ GetObjectRequest request = GetObjectRequest.parseFrom(payload);
+ GetObjectResponse response = kvStore.get(request);
+ return toResponse(response);
+ } catch (Exception e) {
+ log.error("Exception in GetObjectApi: ", e);
+ return toErrorResponse(e);
+ }
+ }
+}

app/src/main/java/org/vss/api/ListKeyVersionsApi.java

@@ -0,0 +1,35 @@
+package org.vss.api;
+
+import jakarta.inject.Inject;
+import jakarta.ws.rs.POST;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.Produces;
+import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
+import lombok.extern.slf4j.Slf4j;
+import org.vss.KVStore;
+import org.vss.ListKeyVersionsRequest;
+import org.vss.ListKeyVersionsResponse;
+
+@Path(VssApiEndpoint.LIST_KEY_VERSIONS)
+@Slf4j
+public class ListKeyVersionsApi extends AbstractVssApi {
+
+ @Inject
+ public ListKeyVersionsApi(KVStore kvStore) {
+ super(kvStore);
+ }
+
+ @POST
+ @Produces(MediaType.APPLICATION_OCTET_STREAM)
+ public Response execute(byte[] payload) {
+ try {
+ ListKeyVersionsRequest request = ListKeyVersionsRequest.parseFrom(payload);
+ ListKeyVersionsResponse response = kvStore.listKeyVersions(request);
+ return toResponse(response);
+ } catch (Exception e) {
+ log.error("Exception in ListKeyVersionsApi: ", e);
+ return toErrorResponse(e);
+ }
+ }
+}

app/src/main/java/org/vss/api/PutObjectsApi.java

@@ -0,0 +1,35 @@
+package org.vss.api;
+
+import jakarta.inject.Inject;
+import jakarta.ws.rs.POST;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.Produces;
+import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
+import lombok.extern.slf4j.Slf4j;
+import org.vss.KVStore;
+import org.vss.PutObjectRequest;
+import org.vss.PutObjectResponse;
+
+@Path(VssApiEndpoint.PUT_OBJECTS)
+@Slf4j
+public class PutObjectsApi extends AbstractVssApi {
+
+ @Inject
+ public PutObjectsApi(KVStore kvStore) {
+ super(kvStore);
+ }
+
+ @POST
+ @Produces(MediaType.APPLICATION_OCTET_STREAM)
+ public Response execute(byte[] payload) {
+ try {
+ PutObjectRequest putObjectRequest = PutObjectRequest.parseFrom(payload);
+ PutObjectResponse response = kvStore.put(putObjectRequest);
+ return toResponse(response);
+ } catch (Exception e) {
+ log.error("Exception in PutObjectsApi: ", e);
+ return toErrorResponse(e);
+ }
+ }
+}

app/src/main/java/org/vss/api/VssApiEndpoint.java

@@ -0,0 +1,7 @@
+package org.vss.api;
+
+public class VssApiEndpoint {
+ public static final String GET_OBJECT = "/getObject";
+ public static final String PUT_OBJECTS = "/putObjects";
+ public static final String LIST_KEY_VERSIONS = "/listKeyVersions";
+}

app/src/test/java/org/vss/api/GetObjectApiTest.java

@@ -0,0 +1,86 @@
+package org.vss.api;
+
+import com.google.protobuf.ByteString;
+import jakarta.ws.rs.core.Response;
+import java.nio.charset.StandardCharsets;
+import java.util.stream.Stream;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.Arguments;
+import org.junit.jupiter.params.provider.MethodSource;
+import org.vss.ErrorCode;
+import org.vss.ErrorResponse;
+import org.vss.GetObjectRequest;
+import org.vss.GetObjectResponse;
+import org.vss.KVStore;
+import org.vss.KeyValue;
+import org.vss.exception.ConflictException;
+
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.is;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+class GetObjectApiTest {
+ private GetObjectApi getObjectApi;
+ private KVStore mockKVStore;
+
+ private static String TEST_STORE_ID = "storeId";
+ private static String TEST_KEY = "key";
+ private static KeyValue TEST_KV = KeyValue.newBuilder().setKey(TEST_KEY).setValue(
+ ByteString.copyFrom("test_value", StandardCharsets.UTF_8)).build();
+
+ @BeforeEach
+ void setUp() {
+ mockKVStore = mock(KVStore.class);
+ getObjectApi = new GetObjectApi(mockKVStore);
+ }
+
+ @Test
+ void execute_ValidPayload_ReturnsResponse() {
+ GetObjectRequest expectedRequest =
+ GetObjectRequest.newBuilder().setStoreId(TEST_STORE_ID).setKey(TEST_KEY).build();
+ byte[] payload = expectedRequest.toByteArray();
+ GetObjectResponse mockResponse = GetObjectResponse.newBuilder().setValue(TEST_KV).build();
+ when(mockKVStore.get(expectedRequest)).thenReturn(mockResponse);
+
+ Response actualResponse = getObjectApi.execute(payload);
+
+ assertThat(actualResponse.getStatus(), is(Response.Status.OK.getStatusCode()));
+ assertThat(actualResponse.getEntity(), is(mockResponse.toByteArray()));
+ verify(mockKVStore).get(expectedRequest);
+ }
+
+ @ParameterizedTest
+ @MethodSource("provideErrorTestCases")
+ void execute_InvalidPayload_ReturnsErrorResponse(Exception exception,
+ ErrorCode errorCode) {
+ GetObjectRequest expectedRequest = GetObjectRequest.newBuilder()
+ .setStoreId(TEST_STORE_ID)
+ .setKey(TEST_KEY)
+ .build();
+ byte[] payload = expectedRequest.toByteArray();
+ when(mockKVStore.get(any())).thenThrow(exception);
+
+ Response response = getObjectApi.execute(payload);
+
+ ErrorResponse expectedErrorResponse = ErrorResponse.newBuilder()
+ .setErrorCode(errorCode)
+ .setMessage("")
+ .build();
+ assertThat(response.getEntity(), is(expectedErrorResponse.toByteArray()));
+ assertThat(response.getStatus(), is(expectedErrorResponse.getErrorCode().getNumber()));
+ verify(mockKVStore).get(expectedRequest);
+ }
+
+ private static Stream<Arguments> provideErrorTestCases() {
+ return Stream.of(
+ Arguments.of(new ConflictException(""), ErrorCode.CONFLICT_EXCEPTION),
+ Arguments.of(new IllegalArgumentException(""), ErrorCode.INVALID_REQUEST_EXCEPTION),
+ Arguments.of(new RuntimeException(""), ErrorCode.INTERNAL_SERVER_EXCEPTION)
+ );
+ }
+}

app/src/test/java/org/vss/api/ListKeyVersionsApiTest.java

@@ -0,0 +1,92 @@
+package org.vss.api;
+
+import com.google.protobuf.ByteString;
+import jakarta.ws.rs.core.Response;
+import java.nio.charset.StandardCharsets;
+import java.util.List;
+import java.util.stream.Stream;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.Arguments;
+import org.junit.jupiter.params.provider.MethodSource;
+import org.vss.ErrorCode;
+import org.vss.ErrorResponse;
+import org.vss.KVStore;
+import org.vss.KeyValue;
+import org.vss.ListKeyVersionsRequest;
+import org.vss.ListKeyVersionsResponse;
+import org.vss.exception.ConflictException;
+
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.is;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+public class ListKeyVersionsApiTest {
+ private ListKeyVersionsApi listKeyVersionsApi;
+ private KVStore mockKVStore;
+
+ private static String TEST_STORE_ID = "storeId";
+ private static String TEST_KEY = "key";
+ private static KeyValue TEST_KV = KeyValue.newBuilder().setKey(TEST_KEY).setValue(
+ ByteString.copyFrom("test_value", StandardCharsets.UTF_8)).build();
+
+ @BeforeEach
+ void setUp() {
+ mockKVStore = mock(KVStore.class);
+ listKeyVersionsApi = new ListKeyVersionsApi(mockKVStore);
+ }
+
+ @Test
+ void execute_ValidPayload_ReturnsResponse() {
+ ListKeyVersionsRequest expectedRequest =
+ ListKeyVersionsRequest.newBuilder()
+ .setStoreId(TEST_STORE_ID)
+ .setKeyPrefix(TEST_KEY)
+ .build();
+ byte[] payload = expectedRequest.toByteArray();
+ ListKeyVersionsResponse mockResponse = ListKeyVersionsResponse.newBuilder().addAllKeyVersions(
+ List.of(TEST_KV)).build();
+ when(mockKVStore.listKeyVersions(expectedRequest)).thenReturn(mockResponse);
+
+ Response actualResponse = listKeyVersionsApi.execute(payload);
+
+ assertThat(actualResponse.getStatus(), is(Response.Status.OK.getStatusCode()));
+ assertThat(actualResponse.getEntity(), is(mockResponse.toByteArray()));
+ verify(mockKVStore).listKeyVersions(expectedRequest);
+ }
+
+ @ParameterizedTest
+ @MethodSource("provideErrorTestCases")
+ void execute_InvalidPayload_ReturnsErrorResponse(Exception exception,
+ ErrorCode errorCode) {
+ ListKeyVersionsRequest expectedRequest =
+ ListKeyVersionsRequest.newBuilder()
+ .setStoreId(TEST_STORE_ID)
+ .setKeyPrefix(TEST_KEY)
+ .build();
+ byte[] payload = expectedRequest.toByteArray();
+ when(mockKVStore.listKeyVersions(any())).thenThrow(exception);
+
+ Response response = listKeyVersionsApi.execute(payload);
+
+ ErrorResponse expectedErrorResponse = ErrorResponse.newBuilder()
+ .setErrorCode(errorCode)
+ .setMessage("")
+ .build();
+ assertThat(response.getEntity(), is(expectedErrorResponse.toByteArray()));
+ assertThat(response.getStatus(), is(expectedErrorResponse.getErrorCode().getNumber()));
+ verify(mockKVStore).listKeyVersions(expectedRequest);
+ }
+
+ private static Stream<Arguments> provideErrorTestCases() {
+ return Stream.of(
+ Arguments.of(new ConflictException(""), ErrorCode.CONFLICT_EXCEPTION),
+ Arguments.of(new IllegalArgumentException(""), ErrorCode.INVALID_REQUEST_EXCEPTION),
+ Arguments.of(new RuntimeException(""), ErrorCode.INTERNAL_SERVER_EXCEPTION)
+ );
+ }
+}

app/build.gradle

@@ -4,6 +4,7 @@ buildscript {
  ext.protobufVersion = '3.21.7'
  ext.jerseyVersion = '3.1.0'
  ext.junitVersion = '5.9.0'
+ ext.mockitoVersion = '5.2.0'
  ext.postgresVersion = '42.5.1'
  ext.jooqVersion = '3.17.7'
  ext.guiceVersion = '5.1.0'
@@ -33,6 +34,8 @@ version '1.0'
 dependencies {
  implementation "com.google.protobuf:protobuf-java:$protobufVersion"
 
+ implementation('org.glassfish.jersey.containers:jersey-container-servlet:3.1.0')
+
  //jOOQ & Postgres impl deps
  implementation "org.jooq:jooq:$jooqVersion"
  implementation "org.jooq:jooq-meta:$jooqVersion"
@@ -42,7 +45,12 @@ dependencies {
 
  implementation "com.google.inject:guice:$guiceVersion"
 
+ compileOnly 'org.projectlombok:lombok:1.18.24'
+ annotationProcessor 'org.projectlombok:lombok:1.18.24'
+
  testImplementation "org.junit.jupiter:junit-jupiter-api:$junitVersion"
+ testImplementation "org.junit.jupiter:junit-jupiter-params:$junitVersion"
+ testImplementation "org.mockito:mockito-core:$mockitoVersion"
  testRuntimeOnly "org.junit.jupiter:junit-jupiter-engine:$junitVersion"
  testImplementation "org.hamcrest:hamcrest-library:2.2"
  testImplementation "org.testcontainers:junit-jupiter:1.17.6"

app/src/main/java/org/vss/VSSApplication.java

@@ -0,0 +1,10 @@
+package org.vss;
+
+import jakarta.ws.rs.ApplicationPath;
+import org.glassfish.jersey.server.ResourceConfig;
+
+@ApplicationPath("/")
+public class VSSApplication extends ResourceConfig {
+ public VSSApplication() {
+ }
+}

app/src/main/java/org/vss/api/AbstractVssApi.java

@@ -0,0 +1,48 @@
+package org.vss.api;
+
+import com.google.protobuf.GeneratedMessageV3;
+import com.google.protobuf.InvalidProtocolBufferException;
+import jakarta.inject.Inject;
+import jakarta.ws.rs.core.Response;
+import org.vss.ErrorCode;
+import org.vss.ErrorResponse;
+import org.vss.KVStore;
+import org.vss.exception.ConflictException;
+
+public abstract class AbstractVssApi {
+ final KVStore kvStore;
+
+ @Inject
+ public AbstractVssApi(KVStore kvStore) {
+ this.kvStore = kvStore;
+ }
+
+ Response toResponse(GeneratedMessageV3 protoResponse) {
+
+ return Response
+ .status(Response.Status.OK)
+ .entity(protoResponse.toByteArray())
+ .build();
+ }
+
+ Response toErrorResponse(Exception e) {
+ ErrorCode errorCode;
+ if (e instanceof ConflictException) {
+ errorCode = ErrorCode.CONFLICT_EXCEPTION;
+ } else if (e instanceof IllegalArgumentException
+ || e instanceof InvalidProtocolBufferException) {
+ errorCode = ErrorCode.INVALID_REQUEST_EXCEPTION;
+ } else {
+ errorCode = ErrorCode.INTERNAL_SERVER_EXCEPTION;
+ }
+
+ ErrorResponse errorResponse = ErrorResponse.newBuilder()
+ .setErrorCode(errorCode)
+ .setMessage(e.getMessage())
+ .build();
+
+ return Response.status(errorCode.getNumber())
+ .entity(errorResponse.toByteArray())
+ .build();
+ }
+}

app/src/main/java/org/vss/api/GetObjectApi.java

@@ -0,0 +1,35 @@
+package org.vss.api;
+
+import jakarta.inject.Inject;
+import jakarta.ws.rs.POST;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.Produces;
+import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
+import lombok.extern.slf4j.Slf4j;
+import org.vss.GetObjectRequest;
+import org.vss.GetObjectResponse;
+import org.vss.KVStore;
+
+@Path(VssApiEndpoint.GET_OBJECT)
+@Slf4j
+public class GetObjectApi extends AbstractVssApi {
+
+ @Inject
+ public GetObjectApi(KVStore kvstore) {
+ super(kvstore);
+ }
+
+ @POST
+ @Produces(MediaType.APPLICATION_OCTET_STREAM)
+ public Response execute(byte[] payload) {
+ try {
+ GetObjectRequest request = GetObjectRequest.parseFrom(payload);
+ GetObjectResponse response = kvStore.get(request);
+ return toResponse(response);
+ } catch (Exception e) {
+ log.error("Exception in GetObjectApi: ", e);
+ return toErrorResponse(e);
+ }
+ }
+}

app/src/main/java/org/vss/api/ListKeyVersionsApi.java

@@ -0,0 +1,35 @@
+package org.vss.api;
+
+import jakarta.inject.Inject;
+import jakarta.ws.rs.POST;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.Produces;
+import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
+import lombok.extern.slf4j.Slf4j;
+import org.vss.KVStore;
+import org.vss.ListKeyVersionsRequest;
+import org.vss.ListKeyVersionsResponse;
+
+@Path(VssApiEndpoint.LIST_KEY_VERSIONS)
+@Slf4j
+public class ListKeyVersionsApi extends AbstractVssApi {
+
+ @Inject
+ public ListKeyVersionsApi(KVStore kvStore) {
+ super(kvStore);
+ }
+
+ @POST
+ @Produces(MediaType.APPLICATION_OCTET_STREAM)
+ public Response execute(byte[] payload) {
+ try {
+ ListKeyVersionsRequest request = ListKeyVersionsRequest.parseFrom(payload);
+ ListKeyVersionsResponse response = kvStore.listKeyVersions(request);
+ return toResponse(response);
+ } catch (Exception e) {
+ log.error("Exception in ListKeyVersionsApi: ", e);
+ return toErrorResponse(e);
+ }
+ }
+}

app/src/main/java/org/vss/api/PutObjectsApi.java

@@ -0,0 +1,35 @@
+package org.vss.api;
+
+import jakarta.inject.Inject;
+import jakarta.ws.rs.POST;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.Produces;
+import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
+import lombok.extern.slf4j.Slf4j;
+import org.vss.KVStore;
+import org.vss.PutObjectRequest;
+import org.vss.PutObjectResponse;
+
+@Path(VssApiEndpoint.PUT_OBJECTS)
+@Slf4j
+public class PutObjectsApi extends AbstractVssApi {
+
+ @Inject
+ public PutObjectsApi(KVStore kvStore) {
+ super(kvStore);
+ }
+
+ @POST
+ @Produces(MediaType.APPLICATION_OCTET_STREAM)
+ public Response execute(byte[] payload) {
+ try {
+ PutObjectRequest putObjectRequest = PutObjectRequest.parseFrom(payload);
+ PutObjectResponse response = kvStore.put(putObjectRequest);
+ return toResponse(response);
+ } catch (Exception e) {
+ log.error("Exception in PutObjectsApi: ", e);
+ return toErrorResponse(e);
+ }
+ }
+}

app/src/main/java/org/vss/api/VssApiEndpoint.java

@@ -0,0 +1,7 @@
+package org.vss.api;
+
+public class VssApiEndpoint {
+ public static final String GET_OBJECT = "/getObject";
+ public static final String PUT_OBJECTS = "/putObjects";
+ public static final String LIST_KEY_VERSIONS = "/listKeyVersions";
+}

app/src/test/java/org/vss/api/GetObjectApiTest.java

@@ -0,0 +1,86 @@
+package org.vss.api;
+
+import com.google.protobuf.ByteString;
+import jakarta.ws.rs.core.Response;
+import java.nio.charset.StandardCharsets;
+import java.util.stream.Stream;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.Arguments;
+import org.junit.jupiter.params.provider.MethodSource;
+import org.vss.ErrorCode;
+import org.vss.ErrorResponse;
+import org.vss.GetObjectRequest;
+import org.vss.GetObjectResponse;
+import org.vss.KVStore;
+import org.vss.KeyValue;
+import org.vss.exception.ConflictException;
+
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.is;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+class GetObjectApiTest {
+ private GetObjectApi getObjectApi;
+ private KVStore mockKVStore;
+
+ private static String TEST_STORE_ID = "storeId";
+ private static String TEST_KEY = "key";
+ private static KeyValue TEST_KV = KeyValue.newBuilder().setKey(TEST_KEY).setValue(
+ ByteString.copyFrom("test_value", StandardCharsets.UTF_8)).build();
+
+ @BeforeEach
+ void setUp() {
+ mockKVStore = mock(KVStore.class);
+ getObjectApi = new GetObjectApi(mockKVStore);
+ }
+
+ @Test
+ void execute_ValidPayload_ReturnsResponse() {
+ GetObjectRequest expectedRequest =
+ GetObjectRequest.newBuilder().setStoreId(TEST_STORE_ID).setKey(TEST_KEY).build();
+ byte[] payload = expectedRequest.toByteArray();
+ GetObjectResponse mockResponse = GetObjectResponse.newBuilder().setValue(TEST_KV).build();
+ when(mockKVStore.get(expectedRequest)).thenReturn(mockResponse);
+
+ Response actualResponse = getObjectApi.execute(payload);
+
+ assertThat(actualResponse.getStatus(), is(Response.Status.OK.getStatusCode()));
+ assertThat(actualResponse.getEntity(), is(mockResponse.toByteArray()));
+ verify(mockKVStore).get(expectedRequest);
+ }
+
+ @ParameterizedTest
+ @MethodSource("provideErrorTestCases")
+ void execute_InvalidPayload_ReturnsErrorResponse(Exception exception,
+ ErrorCode errorCode) {
+ GetObjectRequest expectedRequest = GetObjectRequest.newBuilder()
+ .setStoreId(TEST_STORE_ID)
+ .setKey(TEST_KEY)
+ .build();
+ byte[] payload = expectedRequest.toByteArray();
+ when(mockKVStore.get(any())).thenThrow(exception);
+
+ Response response = getObjectApi.execute(payload);
+
+ ErrorResponse expectedErrorResponse = ErrorResponse.newBuilder()
+ .setErrorCode(errorCode)
+ .setMessage("")
+ .build();
+ assertThat(response.getEntity(), is(expectedErrorResponse.toByteArray()));
+ assertThat(response.getStatus(), is(expectedErrorResponse.getErrorCode().getNumber()));
+ verify(mockKVStore).get(expectedRequest);
+ }
+
+ private static Stream<Arguments> provideErrorTestCases() {
+ return Stream.of(
+ Arguments.of(new ConflictException(""), ErrorCode.CONFLICT_EXCEPTION),
+ Arguments.of(new IllegalArgumentException(""), ErrorCode.INVALID_REQUEST_EXCEPTION),
+ Arguments.of(new RuntimeException(""), ErrorCode.INTERNAL_SERVER_EXCEPTION)
+ );
+ }
+}

app/src/test/java/org/vss/api/ListKeyVersionsApiTest.java

@@ -0,0 +1,92 @@
+package org.vss.api;
+
+import com.google.protobuf.ByteString;
+import jakarta.ws.rs.core.Response;
+import java.nio.charset.StandardCharsets;
+import java.util.List;
+import java.util.stream.Stream;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.Arguments;
+import org.junit.jupiter.params.provider.MethodSource;
+import org.vss.ErrorCode;
+import org.vss.ErrorResponse;
+import org.vss.KVStore;
+import org.vss.KeyValue;
+import org.vss.ListKeyVersionsRequest;
+import org.vss.ListKeyVersionsResponse;
+import org.vss.exception.ConflictException;
+
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.is;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+public class ListKeyVersionsApiTest {
+ private ListKeyVersionsApi listKeyVersionsApi;
+ private KVStore mockKVStore;
+
+ private static String TEST_STORE_ID = "storeId";
+ private static String TEST_KEY = "key";
+ private static KeyValue TEST_KV = KeyValue.newBuilder().setKey(TEST_KEY).setValue(
+ ByteString.copyFrom("test_value", StandardCharsets.UTF_8)).build();
+
+ @BeforeEach
+ void setUp() {
+ mockKVStore = mock(KVStore.class);
+ listKeyVersionsApi = new ListKeyVersionsApi(mockKVStore);
+ }
+
+ @Test
+ void execute_ValidPayload_ReturnsResponse() {
+ ListKeyVersionsRequest expectedRequest =
+ ListKeyVersionsRequest.newBuilder()
+ .setStoreId(TEST_STORE_ID)
+ .setKeyPrefix(TEST_KEY)
+ .build();
+ byte[] payload = expectedRequest.toByteArray();
+ ListKeyVersionsResponse mockResponse = ListKeyVersionsResponse.newBuilder().addAllKeyVersions(
+ List.of(TEST_KV)).build();
+ when(mockKVStore.listKeyVersions(expectedRequest)).thenReturn(mockResponse);
+
+ Response actualResponse = listKeyVersionsApi.execute(payload);
+
+ assertThat(actualResponse.getStatus(), is(Response.Status.OK.getStatusCode()));
+ assertThat(actualResponse.getEntity(), is(mockResponse.toByteArray()));
+ verify(mockKVStore).listKeyVersions(expectedRequest);
+ }
+
+ @ParameterizedTest
+ @MethodSource("provideErrorTestCases")
+ void execute_InvalidPayload_ReturnsErrorResponse(Exception exception,
+ ErrorCode errorCode) {
+ ListKeyVersionsRequest expectedRequest =
+ ListKeyVersionsRequest.newBuilder()
+ .setStoreId(TEST_STORE_ID)
+ .setKeyPrefix(TEST_KEY)
+ .build();
+ byte[] payload = expectedRequest.toByteArray();
+ when(mockKVStore.listKeyVersions(any())).thenThrow(exception);
+
+ Response response = listKeyVersionsApi.execute(payload);
+
+ ErrorResponse expectedErrorResponse = ErrorResponse.newBuilder()
+ .setErrorCode(errorCode)
+ .setMessage("")
+ .build();
+ assertThat(response.getEntity(), is(expectedErrorResponse.toByteArray()));
+ assertThat(response.getStatus(), is(expectedErrorResponse.getErrorCode().getNumber()));
+ verify(mockKVStore).listKeyVersions(expectedRequest);
+ }
+
+ private static Stream<Arguments> provideErrorTestCases() {
+ return Stream.of(
+ Arguments.of(new ConflictException(""), ErrorCode.CONFLICT_EXCEPTION),
+ Arguments.of(new IllegalArgumentException(""), ErrorCode.INVALID_REQUEST_EXCEPTION),
+ Arguments.of(new RuntimeException(""), ErrorCode.INTERNAL_SERVER_EXCEPTION)
+ );
+ }
+}

app/build.gradle

@@ -4,6 +4,7 @@ buildscript {
  ext.protobufVersion = '3.21.7'
  ext.jerseyVersion = '3.1.0'
  ext.junitVersion = '5.9.0'
+ ext.mockitoVersion = '5.2.0'
  ext.postgresVersion = '42.5.1'
  ext.jooqVersion = '3.17.7'
  ext.guiceVersion = '5.1.0'
@@ -33,6 +34,8 @@ version '1.0'
 dependencies {
  implementation "com.google.protobuf:protobuf-java:$protobufVersion"
 
+ implementation('org.glassfish.jersey.containers:jersey-container-servlet:3.1.0')
+
  //jOOQ & Postgres impl deps
  implementation "org.jooq:jooq:$jooqVersion"
  implementation "org.jooq:jooq-meta:$jooqVersion"
@@ -42,7 +45,12 @@ dependencies {
 
  implementation "com.google.inject:guice:$guiceVersion"
 
+ compileOnly 'org.projectlombok:lombok:1.18.24'
+ annotationProcessor 'org.projectlombok:lombok:1.18.24'
+
  testImplementation "org.junit.jupiter:junit-jupiter-api:$junitVersion"
+ testImplementation "org.junit.jupiter:junit-jupiter-params:$junitVersion"
+ testImplementation "org.mockito:mockito-core:$mockitoVersion"
  testRuntimeOnly "org.junit.jupiter:junit-jupiter-engine:$junitVersion"
  testImplementation "org.hamcrest:hamcrest-library:2.2"
  testImplementation "org.testcontainers:junit-jupiter:1.17.6"

app/src/main/java/org/vss/VSSApplication.java

@@ -0,0 +1,10 @@
+package org.vss;
+
+import jakarta.ws.rs.ApplicationPath;
+import org.glassfish.jersey.server.ResourceConfig;
+
+@ApplicationPath("/")
+public class VSSApplication extends ResourceConfig {
+ public VSSApplication() {
+ }
+}

app/src/main/java/org/vss/api/AbstractVssApi.java

@@ -0,0 +1,48 @@
+package org.vss.api;
+
+import com.google.protobuf.GeneratedMessageV3;
+import com.google.protobuf.InvalidProtocolBufferException;
+import jakarta.inject.Inject;
+import jakarta.ws.rs.core.Response;
+import org.vss.ErrorCode;
+import org.vss.ErrorResponse;
+import org.vss.KVStore;
+import org.vss.exception.ConflictException;
+
+public abstract class AbstractVssApi {
+ final KVStore kvStore;
+
+ @Inject
+ public AbstractVssApi(KVStore kvStore) {
+ this.kvStore = kvStore;
+ }
+
+ Response toResponse(GeneratedMessageV3 protoResponse) {
+
+ return Response
+ .status(Response.Status.OK)
+ .entity(protoResponse.toByteArray())
+ .build();
+ }
+
+ Response toErrorResponse(Exception e) {
+ ErrorCode errorCode;
+ if (e instanceof ConflictException) {
+ errorCode = ErrorCode.CONFLICT_EXCEPTION;
+ } else if (e instanceof IllegalArgumentException
+ || e instanceof InvalidProtocolBufferException) {
+ errorCode = ErrorCode.INVALID_REQUEST_EXCEPTION;
+ } else {
+ errorCode = ErrorCode.INTERNAL_SERVER_EXCEPTION;
+ }
+
+ ErrorResponse errorResponse = ErrorResponse.newBuilder()
+ .setErrorCode(errorCode)
+ .setMessage(e.getMessage())
+ .build();
+
+ return Response.status(errorCode.getNumber())
+ .entity(errorResponse.toByteArray())
+ .build();
+ }
+}

app/src/main/java/org/vss/api/GetObjectApi.java

@@ -0,0 +1,35 @@
+package org.vss.api;
+
+import jakarta.inject.Inject;
+import jakarta.ws.rs.POST;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.Produces;
+import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
+import lombok.extern.slf4j.Slf4j;
+import org.vss.GetObjectRequest;
+import org.vss.GetObjectResponse;
+import org.vss.KVStore;
+
+@Path(VssApiEndpoint.GET_OBJECT)
+@Slf4j
+public class GetObjectApi extends AbstractVssApi {
+
+ @Inject
+ public GetObjectApi(KVStore kvstore) {
+ super(kvstore);
+ }
+
+ @POST
+ @Produces(MediaType.APPLICATION_OCTET_STREAM)
+ public Response execute(byte[] payload) {
+ try {
+ GetObjectRequest request = GetObjectRequest.parseFrom(payload);
+ GetObjectResponse response = kvStore.get(request);
+ return toResponse(response);
+ } catch (Exception e) {
+ log.error("Exception in GetObjectApi: ", e);
+ return toErrorResponse(e);
+ }
+ }
+}

app/src/main/java/org/vss/api/ListKeyVersionsApi.java

@@ -0,0 +1,35 @@
+package org.vss.api;
+
+import jakarta.inject.Inject;
+import jakarta.ws.rs.POST;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.Produces;
+import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
+import lombok.extern.slf4j.Slf4j;
+import org.vss.KVStore;
+import org.vss.ListKeyVersionsRequest;
+import org.vss.ListKeyVersionsResponse;
+
+@Path(VssApiEndpoint.LIST_KEY_VERSIONS)
+@Slf4j
+public class ListKeyVersionsApi extends AbstractVssApi {
+
+ @Inject
+ public ListKeyVersionsApi(KVStore kvStore) {
+ super(kvStore);
+ }
+
+ @POST
+ @Produces(MediaType.APPLICATION_OCTET_STREAM)
+ public Response execute(byte[] payload) {
+ try {
+ ListKeyVersionsRequest request = ListKeyVersionsRequest.parseFrom(payload);
+ ListKeyVersionsResponse response = kvStore.listKeyVersions(request);
+ return toResponse(response);
+ } catch (Exception e) {
+ log.error("Exception in ListKeyVersionsApi: ", e);
+ return toErrorResponse(e);
+ }
+ }
+}

app/src/main/java/org/vss/api/PutObjectsApi.java

@@ -0,0 +1,35 @@
+package org.vss.api;
+
+import jakarta.inject.Inject;
+import jakarta.ws.rs.POST;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.Produces;
+import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
+import lombok.extern.slf4j.Slf4j;
+import org.vss.KVStore;
+import org.vss.PutObjectRequest;
+import org.vss.PutObjectResponse;
+
+@Path(VssApiEndpoint.PUT_OBJECTS)
+@Slf4j
+public class PutObjectsApi extends AbstractVssApi {
+
+ @Inject
+ public PutObjectsApi(KVStore kvStore) {
+ super(kvStore);
+ }
+
+ @POST
+ @Produces(MediaType.APPLICATION_OCTET_STREAM)
+ public Response execute(byte[] payload) {
+ try {
+ PutObjectRequest putObjectRequest = PutObjectRequest.parseFrom(payload);
+ PutObjectResponse response = kvStore.put(putObjectRequest);
+ return toResponse(response);
+ } catch (Exception e) {
+ log.error("Exception in PutObjectsApi: ", e);
+ return toErrorResponse(e);
+ }
+ }
+}

app/src/main/java/org/vss/api/VssApiEndpoint.java

@@ -0,0 +1,7 @@
+package org.vss.api;
+
+public class VssApiEndpoint {
+ public static final String GET_OBJECT = "/getObject";
+ public static final String PUT_OBJECTS = "/putObjects";
+ public static final String LIST_KEY_VERSIONS = "/listKeyVersions";
+}

app/src/test/java/org/vss/api/GetObjectApiTest.java

@@ -0,0 +1,86 @@
+package org.vss.api;
+
+import com.google.protobuf.ByteString;
+import jakarta.ws.rs.core.Response;
+import java.nio.charset.StandardCharsets;
+import java.util.stream.Stream;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.Arguments;
+import org.junit.jupiter.params.provider.MethodSource;
+import org.vss.ErrorCode;
+import org.vss.ErrorResponse;
+import org.vss.GetObjectRequest;
+import org.vss.GetObjectResponse;
+import org.vss.KVStore;
+import org.vss.KeyValue;
+import org.vss.exception.ConflictException;
+
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.is;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+class GetObjectApiTest {
+ private GetObjectApi getObjectApi;
+ private KVStore mockKVStore;
+
+ private static String TEST_STORE_ID = "storeId";
+ private static String TEST_KEY = "key";
+ private static KeyValue TEST_KV = KeyValue.newBuilder().setKey(TEST_KEY).setValue(
+ ByteString.copyFrom("test_value", StandardCharsets.UTF_8)).build();
+
+ @BeforeEach
+ void setUp() {
+ mockKVStore = mock(KVStore.class);
+ getObjectApi = new GetObjectApi(mockKVStore);
+ }
+
+ @Test
+ void execute_ValidPayload_ReturnsResponse() {
+ GetObjectRequest expectedRequest =
+ GetObjectRequest.newBuilder().setStoreId(TEST_STORE_ID).setKey(TEST_KEY).build();
+ byte[] payload = expectedRequest.toByteArray();
+ GetObjectResponse mockResponse = GetObjectResponse.newBuilder().setValue(TEST_KV).build();
+ when(mockKVStore.get(expectedRequest)).thenReturn(mockResponse);
+
+ Response actualResponse = getObjectApi.execute(payload);
+
+ assertThat(actualResponse.getStatus(), is(Response.Status.OK.getStatusCode()));
+ assertThat(actualResponse.getEntity(), is(mockResponse.toByteArray()));
+ verify(mockKVStore).get(expectedRequest);
+ }
+
+ @ParameterizedTest
+ @MethodSource("provideErrorTestCases")
+ void execute_InvalidPayload_ReturnsErrorResponse(Exception exception,
+ ErrorCode errorCode) {
+ GetObjectRequest expectedRequest = GetObjectRequest.newBuilder()
+ .setStoreId(TEST_STORE_ID)
+ .setKey(TEST_KEY)
+ .build();
+ byte[] payload = expectedRequest.toByteArray();
+ when(mockKVStore.get(any())).thenThrow(exception);
+
+ Response response = getObjectApi.execute(payload);
+
+ ErrorResponse expectedErrorResponse = ErrorResponse.newBuilder()
+ .setErrorCode(errorCode)
+ .setMessage("")
+ .build();
+ assertThat(response.getEntity(), is(expectedErrorResponse.toByteArray()));
+ assertThat(response.getStatus(), is(expectedErrorResponse.getErrorCode().getNumber()));
+ verify(mockKVStore).get(expectedRequest);
+ }
+
+ private static Stream<Arguments> provideErrorTestCases() {
+ return Stream.of(
+ Arguments.of(new ConflictException(""), ErrorCode.CONFLICT_EXCEPTION),
+ Arguments.of(new IllegalArgumentException(""), ErrorCode.INVALID_REQUEST_EXCEPTION),
+ Arguments.of(new RuntimeException(""), ErrorCode.INTERNAL_SERVER_EXCEPTION)
+ );
+ }
+}

app/src/test/java/org/vss/api/ListKeyVersionsApiTest.java

@@ -0,0 +1,92 @@
+package org.vss.api;
+
+import com.google.protobuf.ByteString;
+import jakarta.ws.rs.core.Response;
+import java.nio.charset.StandardCharsets;
+import java.util.List;
+import java.util.stream.Stream;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.Arguments;
+import org.junit.jupiter.params.provider.MethodSource;
+import org.vss.ErrorCode;
+import org.vss.ErrorResponse;
+import org.vss.KVStore;
+import org.vss.KeyValue;
+import org.vss.ListKeyVersionsRequest;
+import org.vss.ListKeyVersionsResponse;
+import org.vss.exception.ConflictException;
+
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.is;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+public class ListKeyVersionsApiTest {
+ private ListKeyVersionsApi listKeyVersionsApi;
+ private KVStore mockKVStore;
+
+ private static String TEST_STORE_ID = "storeId";
+ private static String TEST_KEY = "key";
+ private static KeyValue TEST_KV = KeyValue.newBuilder().setKey(TEST_KEY).setValue(
+ ByteString.copyFrom("test_value", StandardCharsets.UTF_8)).build();
+
+ @BeforeEach
+ void setUp() {
+ mockKVStore = mock(KVStore.class);
+ listKeyVersionsApi = new ListKeyVersionsApi(mockKVStore);
+ }
+
+ @Test
+ void execute_ValidPayload_ReturnsResponse() {
+ ListKeyVersionsRequest expectedRequest =
+ ListKeyVersionsRequest.newBuilder()
+ .setStoreId(TEST_STORE_ID)
+ .setKeyPrefix(TEST_KEY)
+ .build();
+ byte[] payload = expectedRequest.toByteArray();
+ ListKeyVersionsResponse mockResponse = ListKeyVersionsResponse.newBuilder().addAllKeyVersions(
+ List.of(TEST_KV)).build();
+ when(mockKVStore.listKeyVersions(expectedRequest)).thenReturn(mockResponse);
+
+ Response actualResponse = listKeyVersionsApi.execute(payload);
+
+ assertThat(actualResponse.getStatus(), is(Response.Status.OK.getStatusCode()));
+ assertThat(actualResponse.getEntity(), is(mockResponse.toByteArray()));
+ verify(mockKVStore).listKeyVersions(expectedRequest);
+ }
+
+ @ParameterizedTest
+ @MethodSource("provideErrorTestCases")
+ void execute_InvalidPayload_ReturnsErrorResponse(Exception exception,
+ ErrorCode errorCode) {
+ ListKeyVersionsRequest expectedRequest =
+ ListKeyVersionsRequest.newBuilder()
+ .setStoreId(TEST_STORE_ID)
+ .setKeyPrefix(TEST_KEY)
+ .build();
+ byte[] payload = expectedRequest.toByteArray();
+ when(mockKVStore.listKeyVersions(any())).thenThrow(exception);
+
+ Response response = listKeyVersionsApi.execute(payload);
+
+ ErrorResponse expectedErrorResponse = ErrorResponse.newBuilder()
+ .setErrorCode(errorCode)
+ .setMessage("")
+ .build();
+ assertThat(response.getEntity(), is(expectedErrorResponse.toByteArray()));
+ assertThat(response.getStatus(), is(expectedErrorResponse.getErrorCode().getNumber()));
+ verify(mockKVStore).listKeyVersions(expectedRequest);
+ }
+
+ private static Stream<Arguments> provideErrorTestCases() {
+ return Stream.of(
+ Arguments.of(new ConflictException(""), ErrorCode.CONFLICT_EXCEPTION),
+ Arguments.of(new IllegalArgumentException(""), ErrorCode.INVALID_REQUEST_EXCEPTION),
+ Arguments.of(new RuntimeException(""), ErrorCode.INTERNAL_SERVER_EXCEPTION)
+ );
+ }
+}

app/build.gradle

@@ -4,6 +4,7 @@ buildscript {
  ext.protobufVersion = '3.21.7'
  ext.jerseyVersion = '3.1.0'
  ext.junitVersion = '5.9.0'
+ ext.mockitoVersion = '5.2.0'
  ext.postgresVersion = '42.5.1'
  ext.jooqVersion = '3.17.7'
  ext.guiceVersion = '5.1.0'
@@ -33,6 +34,8 @@ version '1.0'
 dependencies {
  implementation "com.google.protobuf:protobuf-java:$protobufVersion"
 
+ implementation('org.glassfish.jersey.containers:jersey-container-servlet:3.1.0')
+
  //jOOQ & Postgres impl deps
  implementation "org.jooq:jooq:$jooqVersion"
  implementation "org.jooq:jooq-meta:$jooqVersion"
@@ -42,7 +45,12 @@ dependencies {
 
  implementation "com.google.inject:guice:$guiceVersion"
 
+ compileOnly 'org.projectlombok:lombok:1.18.24'
+ annotationProcessor 'org.projectlombok:lombok:1.18.24'
+
  testImplementation "org.junit.jupiter:junit-jupiter-api:$junitVersion"
+ testImplementation "org.junit.jupiter:junit-jupiter-params:$junitVersion"
+ testImplementation "org.mockito:mockito-core:$mockitoVersion"
  testRuntimeOnly "org.junit.jupiter:junit-jupiter-engine:$junitVersion"
  testImplementation "org.hamcrest:hamcrest-library:2.2"
  testImplementation "org.testcontainers:junit-jupiter:1.17.6"

app/src/main/java/org/vss/VSSApplication.java

@@ -0,0 +1,10 @@
+package org.vss;
+
+import jakarta.ws.rs.ApplicationPath;
+import org.glassfish.jersey.server.ResourceConfig;
+
+@ApplicationPath("/")
+public class VSSApplication extends ResourceConfig {
+ public VSSApplication() {
+ }
+}

app/src/main/java/org/vss/api/AbstractVssApi.java

@@ -0,0 +1,48 @@
+package org.vss.api;
+
+import com.google.protobuf.GeneratedMessageV3;
+import com.google.protobuf.InvalidProtocolBufferException;
+import jakarta.inject.Inject;
+import jakarta.ws.rs.core.Response;
+import org.vss.ErrorCode;
+import org.vss.ErrorResponse;
+import org.vss.KVStore;
+import org.vss.exception.ConflictException;
+
+public abstract class AbstractVssApi {
+ final KVStore kvStore;
+
+ @Inject
+ public AbstractVssApi(KVStore kvStore) {
+ this.kvStore = kvStore;
+ }
+
+ Response toResponse(GeneratedMessageV3 protoResponse) {
+
+ return Response
+ .status(Response.Status.OK)
+ .entity(protoResponse.toByteArray())
+ .build();
+ }
+
+ Response toErrorResponse(Exception e) {
+ ErrorCode errorCode;
+ if (e instanceof ConflictException) {
+ errorCode = ErrorCode.CONFLICT_EXCEPTION;
+ } else if (e instanceof IllegalArgumentException
+ || e instanceof InvalidProtocolBufferException) {
+ errorCode = ErrorCode.INVALID_REQUEST_EXCEPTION;
+ } else {
+ errorCode = ErrorCode.INTERNAL_SERVER_EXCEPTION;
+ }
+
+ ErrorResponse errorResponse = ErrorResponse.newBuilder()
+ .setErrorCode(errorCode)
+ .setMessage(e.getMessage())
+ .build();
+
+ return Response.status(errorCode.getNumber())
+ .entity(errorResponse.toByteArray())
+ .build();
+ }
+}

app/src/main/java/org/vss/api/GetObjectApi.java

@@ -0,0 +1,35 @@
+package org.vss.api;
+
+import jakarta.inject.Inject;
+import jakarta.ws.rs.POST;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.Produces;
+import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
+import lombok.extern.slf4j.Slf4j;
+import org.vss.GetObjectRequest;
+import org.vss.GetObjectResponse;
+import org.vss.KVStore;
+
+@Path(VssApiEndpoint.GET_OBJECT)
+@Slf4j
+public class GetObjectApi extends AbstractVssApi {
+
+ @Inject
+ public GetObjectApi(KVStore kvstore) {
+ super(kvstore);
+ }
+
+ @POST
+ @Produces(MediaType.APPLICATION_OCTET_STREAM)
+ public Response execute(byte[] payload) {
+ try {
+ GetObjectRequest request = GetObjectRequest.parseFrom(payload);
+ GetObjectResponse response = kvStore.get(request);
+ return toResponse(response);
+ } catch (Exception e) {
+ log.error("Exception in GetObjectApi: ", e);
+ return toErrorResponse(e);
+ }
+ }
+}

app/src/main/java/org/vss/api/ListKeyVersionsApi.java

@@ -0,0 +1,35 @@
+package org.vss.api;
+
+import jakarta.inject.Inject;
+import jakarta.ws.rs.POST;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.Produces;
+import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
+import lombok.extern.slf4j.Slf4j;
+import org.vss.KVStore;
+import org.vss.ListKeyVersionsRequest;
+import org.vss.ListKeyVersionsResponse;
+
+@Path(VssApiEndpoint.LIST_KEY_VERSIONS)
+@Slf4j
+public class ListKeyVersionsApi extends AbstractVssApi {
+
+ @Inject
+ public ListKeyVersionsApi(KVStore kvStore) {
+ super(kvStore);
+ }
+
+ @POST
+ @Produces(MediaType.APPLICATION_OCTET_STREAM)
+ public Response execute(byte[] payload) {
+ try {
+ ListKeyVersionsRequest request = ListKeyVersionsRequest.parseFrom(payload);
+ ListKeyVersionsResponse response = kvStore.listKeyVersions(request);
+ return toResponse(response);
+ } catch (Exception e) {
+ log.error("Exception in ListKeyVersionsApi: ", e);
+ return toErrorResponse(e);
+ }
+ }
+}

app/src/main/java/org/vss/api/PutObjectsApi.java

@@ -0,0 +1,35 @@
+package org.vss.api;
+
+import jakarta.inject.Inject;
+import jakarta.ws.rs.POST;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.Produces;
+import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
+import lombok.extern.slf4j.Slf4j;
+import org.vss.KVStore;
+import org.vss.PutObjectRequest;
+import org.vss.PutObjectResponse;
+
+@Path(VssApiEndpoint.PUT_OBJECTS)
+@Slf4j
+public class PutObjectsApi extends AbstractVssApi {
+
+ @Inject
+ public PutObjectsApi(KVStore kvStore) {
+ super(kvStore);
+ }
+
+ @POST
+ @Produces(MediaType.APPLICATION_OCTET_STREAM)
+ public Response execute(byte[] payload) {
+ try {
+ PutObjectRequest putObjectRequest = PutObjectRequest.parseFrom(payload);
+ PutObjectResponse response = kvStore.put(putObjectRequest);
+ return toResponse(response);
+ } catch (Exception e) {
+ log.error("Exception in PutObjectsApi: ", e);
+ return toErrorResponse(e);
+ }
+ }
+}

app/src/main/java/org/vss/api/VssApiEndpoint.java

@@ -0,0 +1,7 @@
+package org.vss.api;
+
+public class VssApiEndpoint {
+ public static final String GET_OBJECT = "/getObject";
+ public static final String PUT_OBJECTS = "/putObjects";
+ public static final String LIST_KEY_VERSIONS = "/listKeyVersions";
+}

app/src/test/java/org/vss/api/GetObjectApiTest.java

@@ -0,0 +1,86 @@
+package org.vss.api;
+
+import com.google.protobuf.ByteString;
+import jakarta.ws.rs.core.Response;
+import java.nio.charset.StandardCharsets;
+import java.util.stream.Stream;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.Arguments;
+import org.junit.jupiter.params.provider.MethodSource;
+import org.vss.ErrorCode;
+import org.vss.ErrorResponse;
+import org.vss.GetObjectRequest;
+import org.vss.GetObjectResponse;
+import org.vss.KVStore;
+import org.vss.KeyValue;
+import org.vss.exception.ConflictException;
+
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.is;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+class GetObjectApiTest {
+ private GetObjectApi getObjectApi;
+ private KVStore mockKVStore;
+
+ private static String TEST_STORE_ID = "storeId";
+ private static String TEST_KEY = "key";
+ private static KeyValue TEST_KV = KeyValue.newBuilder().setKey(TEST_KEY).setValue(
+ ByteString.copyFrom("test_value", StandardCharsets.UTF_8)).build();
+
+ @BeforeEach
+ void setUp() {
+ mockKVStore = mock(KVStore.class);
+ getObjectApi = new GetObjectApi(mockKVStore);
+ }
+
+ @Test
+ void execute_ValidPayload_ReturnsResponse() {
+ GetObjectRequest expectedRequest =
+ GetObjectRequest.newBuilder().setStoreId(TEST_STORE_ID).setKey(TEST_KEY).build();
+ byte[] payload = expectedRequest.toByteArray();
+ GetObjectResponse mockResponse = GetObjectResponse.newBuilder().setValue(TEST_KV).build();
+ when(mockKVStore.get(expectedRequest)).thenReturn(mockResponse);
+
+ Response actualResponse = getObjectApi.execute(payload);
+
+ assertThat(actualResponse.getStatus(), is(Response.Status.OK.getStatusCode()));
+ assertThat(actualResponse.getEntity(), is(mockResponse.toByteArray()));
+ verify(mockKVStore).get(expectedRequest);
+ }
+
+ @ParameterizedTest
+ @MethodSource("provideErrorTestCases")
+ void execute_InvalidPayload_ReturnsErrorResponse(Exception exception,
+ ErrorCode errorCode) {
+ GetObjectRequest expectedRequest = GetObjectRequest.newBuilder()
+ .setStoreId(TEST_STORE_ID)
+ .setKey(TEST_KEY)
+ .build();
+ byte[] payload = expectedRequest.toByteArray();
+ when(mockKVStore.get(any())).thenThrow(exception);
+
+ Response response = getObjectApi.execute(payload);
+
+ ErrorResponse expectedErrorResponse = ErrorResponse.newBuilder()
+ .setErrorCode(errorCode)
+ .setMessage("")
+ .build();
+ assertThat(response.getEntity(), is(expectedErrorResponse.toByteArray()));
+ assertThat(response.getStatus(), is(expectedErrorResponse.getErrorCode().getNumber()));
+ verify(mockKVStore).get(expectedRequest);
+ }
+
+ private static Stream<Arguments> provideErrorTestCases() {
+ return Stream.of(
+ Arguments.of(new ConflictException(""), ErrorCode.CONFLICT_EXCEPTION),
+ Arguments.of(new IllegalArgumentException(""), ErrorCode.INVALID_REQUEST_EXCEPTION),
+ Arguments.of(new RuntimeException(""), ErrorCode.INTERNAL_SERVER_EXCEPTION)
+ );
+ }
+}

app/src/test/java/org/vss/api/ListKeyVersionsApiTest.java

@@ -0,0 +1,92 @@
+package org.vss.api;
+
+import com.google.protobuf.ByteString;
+import jakarta.ws.rs.core.Response;
+import java.nio.charset.StandardCharsets;
+import java.util.List;
+import java.util.stream.Stream;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.Arguments;
+import org.junit.jupiter.params.provider.MethodSource;
+import org.vss.ErrorCode;
+import org.vss.ErrorResponse;
+import org.vss.KVStore;
+import org.vss.KeyValue;
+import org.vss.ListKeyVersionsRequest;
+import org.vss.ListKeyVersionsResponse;
+import org.vss.exception.ConflictException;
+
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.is;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+public class ListKeyVersionsApiTest {
+ private ListKeyVersionsApi listKeyVersionsApi;
+ private KVStore mockKVStore;
+
+ private static String TEST_STORE_ID = "storeId";
+ private static String TEST_KEY = "key";
+ private static KeyValue TEST_KV = KeyValue.newBuilder().setKey(TEST_KEY).setValue(
+ ByteString.copyFrom("test_value", StandardCharsets.UTF_8)).build();
+
+ @BeforeEach
+ void setUp() {
+ mockKVStore = mock(KVStore.class);
+ listKeyVersionsApi = new ListKeyVersionsApi(mockKVStore);
+ }
+
+ @Test
+ void execute_ValidPayload_ReturnsResponse() {
+ ListKeyVersionsRequest expectedRequest =
+ ListKeyVersionsRequest.newBuilder()
+ .setStoreId(TEST_STORE_ID)
+ .setKeyPrefix(TEST_KEY)
+ .build();
+ byte[] payload = expectedRequest.toByteArray();
+ ListKeyVersionsResponse mockResponse = ListKeyVersionsResponse.newBuilder().addAllKeyVersions(
+ List.of(TEST_KV)).build();
+ when(mockKVStore.listKeyVersions(expectedRequest)).thenReturn(mockResponse);
+
+ Response actualResponse = listKeyVersionsApi.execute(payload);
+
+ assertThat(actualResponse.getStatus(), is(Response.Status.OK.getStatusCode()));
+ assertThat(actualResponse.getEntity(), is(mockResponse.toByteArray()));
+ verify(mockKVStore).listKeyVersions(expectedRequest);
+ }
+
+ @ParameterizedTest
+ @MethodSource("provideErrorTestCases")
+ void execute_InvalidPayload_ReturnsErrorResponse(Exception exception,
+ ErrorCode errorCode) {
+ ListKeyVersionsRequest expectedRequest =
+ ListKeyVersionsRequest.newBuilder()
+ .setStoreId(TEST_STORE_ID)
+ .setKeyPrefix(TEST_KEY)
+ .build();
+ byte[] payload = expectedRequest.toByteArray();
+ when(mockKVStore.listKeyVersions(any())).thenThrow(exception);
+
+ Response response = listKeyVersionsApi.execute(payload);
+
+ ErrorResponse expectedErrorResponse = ErrorResponse.newBuilder()
+ .setErrorCode(errorCode)
+ .setMessage("")
+ .build();
+ assertThat(response.getEntity(), is(expectedErrorResponse.toByteArray()));
+ assertThat(response.getStatus(), is(expectedErrorResponse.getErrorCode().getNumber()));
+ verify(mockKVStore).listKeyVersions(expectedRequest);
+ }
+
+ private static Stream<Arguments> provideErrorTestCases() {
+ return Stream.of(
+ Arguments.of(new ConflictException(""), ErrorCode.CONFLICT_EXCEPTION),
+ Arguments.of(new IllegalArgumentException(""), ErrorCode.INVALID_REQUEST_EXCEPTION),
+ Arguments.of(new RuntimeException(""), ErrorCode.INTERNAL_SERVER_EXCEPTION)
+ );
+ }
+}

app/build.gradle

@@ -4,6 +4,7 @@ buildscript {
  ext.protobufVersion = '3.21.7'
  ext.jerseyVersion = '3.1.0'
  ext.junitVersion = '5.9.0'
+ ext.mockitoVersion = '5.2.0'
  ext.postgresVersion = '42.5.1'
  ext.jooqVersion = '3.17.7'
  ext.guiceVersion = '5.1.0'
@@ -33,6 +34,8 @@ version '1.0'
 dependencies {
  implementation "com.google.protobuf:protobuf-java:$protobufVersion"
 
+ implementation('org.glassfish.jersey.containers:jersey-container-servlet:3.1.0')
+
  //jOOQ & Postgres impl deps
  implementation "org.jooq:jooq:$jooqVersion"
  implementation "org.jooq:jooq-meta:$jooqVersion"
@@ -42,7 +45,12 @@ dependencies {
 
  implementation "com.google.inject:guice:$guiceVersion"
 
+ compileOnly 'org.projectlombok:lombok:1.18.24'
+ annotationProcessor 'org.projectlombok:lombok:1.18.24'
+
  testImplementation "org.junit.jupiter:junit-jupiter-api:$junitVersion"
+ testImplementation "org.junit.jupiter:junit-jupiter-params:$junitVersion"
+ testImplementation "org.mockito:mockito-core:$mockitoVersion"
  testRuntimeOnly "org.junit.jupiter:junit-jupiter-engine:$junitVersion"
  testImplementation "org.hamcrest:hamcrest-library:2.2"
  testImplementation "org.testcontainers:junit-jupiter:1.17.6"

app/src/main/java/org/vss/VSSApplication.java

@@ -0,0 +1,10 @@
+package org.vss;
+
+import jakarta.ws.rs.ApplicationPath;
+import org.glassfish.jersey.server.ResourceConfig;
+
+@ApplicationPath("/")
+public class VSSApplication extends ResourceConfig {
+ public VSSApplication() {
+ }
+}

app/src/main/java/org/vss/api/AbstractVssApi.java

@@ -0,0 +1,48 @@
+package org.vss.api;
+
+import com.google.protobuf.GeneratedMessageV3;
+import com.google.protobuf.InvalidProtocolBufferException;
+import jakarta.inject.Inject;
+import jakarta.ws.rs.core.Response;
+import org.vss.ErrorCode;
+import org.vss.ErrorResponse;
+import org.vss.KVStore;
+import org.vss.exception.ConflictException;
+
+public abstract class AbstractVssApi {
+ final KVStore kvStore;
+
+ @Inject
+ public AbstractVssApi(KVStore kvStore) {
+ this.kvStore = kvStore;
+ }
+
+ Response toResponse(GeneratedMessageV3 protoResponse) {
+
+ return Response
+ .status(Response.Status.OK)
+ .entity(protoResponse.toByteArray())
+ .build();
+ }
+
+ Response toErrorResponse(Exception e) {
+ ErrorCode errorCode;
+ if (e instanceof ConflictException) {
+ errorCode = ErrorCode.CONFLICT_EXCEPTION;
+ } else if (e instanceof IllegalArgumentException
+ || e instanceof InvalidProtocolBufferException) {
+ errorCode = ErrorCode.INVALID_REQUEST_EXCEPTION;
+ } else {
+ errorCode = ErrorCode.INTERNAL_SERVER_EXCEPTION;
+ }
+
+ ErrorResponse errorResponse = ErrorResponse.newBuilder()
+ .setErrorCode(errorCode)
+ .setMessage(e.getMessage())
+ .build();
+
+ return Response.status(errorCode.getNumber())
+ .entity(errorResponse.toByteArray())
+ .build();
+ }
+}

app/src/main/java/org/vss/api/GetObjectApi.java

@@ -0,0 +1,35 @@
+package org.vss.api;
+
+import jakarta.inject.Inject;
+import jakarta.ws.rs.POST;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.Produces;
+import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
+import lombok.extern.slf4j.Slf4j;
+import org.vss.GetObjectRequest;
+import org.vss.GetObjectResponse;
+import org.vss.KVStore;
+
+@Path(VssApiEndpoint.GET_OBJECT)
+@Slf4j
+public class GetObjectApi extends AbstractVssApi {
+
+ @Inject
+ public GetObjectApi(KVStore kvstore) {
+ super(kvstore);
+ }
+
+ @POST
+ @Produces(MediaType.APPLICATION_OCTET_STREAM)
+ public Response execute(byte[] payload) {
+ try {
+ GetObjectRequest request = GetObjectRequest.parseFrom(payload);
+ GetObjectResponse response = kvStore.get(request);
+ return toResponse(response);
+ } catch (Exception e) {
+ log.error("Exception in GetObjectApi: ", e);
+ return toErrorResponse(e);
+ }
+ }
+}

app/src/main/java/org/vss/api/ListKeyVersionsApi.java

@@ -0,0 +1,35 @@
+package org.vss.api;
+
+import jakarta.inject.Inject;
+import jakarta.ws.rs.POST;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.Produces;
+import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
+import lombok.extern.slf4j.Slf4j;
+import org.vss.KVStore;
+import org.vss.ListKeyVersionsRequest;
+import org.vss.ListKeyVersionsResponse;
+
+@Path(VssApiEndpoint.LIST_KEY_VERSIONS)
+@Slf4j
+public class ListKeyVersionsApi extends AbstractVssApi {
+
+ @Inject
+ public ListKeyVersionsApi(KVStore kvStore) {
+ super(kvStore);
+ }
+
+ @POST
+ @Produces(MediaType.APPLICATION_OCTET_STREAM)
+ public Response execute(byte[] payload) {
+ try {
+ ListKeyVersionsRequest request = ListKeyVersionsRequest.parseFrom(payload);
+ ListKeyVersionsResponse response = kvStore.listKeyVersions(request);
+ return toResponse(response);
+ } catch (Exception e) {
+ log.error("Exception in ListKeyVersionsApi: ", e);
+ return toErrorResponse(e);
+ }
+ }
+}

app/src/main/java/org/vss/api/PutObjectsApi.java

@@ -0,0 +1,35 @@
+package org.vss.api;
+
+import jakarta.inject.Inject;
+import jakarta.ws.rs.POST;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.Produces;
+import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
+import lombok.extern.slf4j.Slf4j;
+import org.vss.KVStore;
+import org.vss.PutObjectRequest;
+import org.vss.PutObjectResponse;
+
+@Path(VssApiEndpoint.PUT_OBJECTS)
+@Slf4j
+public class PutObjectsApi extends AbstractVssApi {
+
+ @Inject
+ public PutObjectsApi(KVStore kvStore) {
+ super(kvStore);
+ }
+
+ @POST
+ @Produces(MediaType.APPLICATION_OCTET_STREAM)
+ public Response execute(byte[] payload) {
+ try {
+ PutObjectRequest putObjectRequest = PutObjectRequest.parseFrom(payload);
+ PutObjectResponse response = kvStore.put(putObjectRequest);
+ return toResponse(response);
+ } catch (Exception e) {
+ log.error("Exception in PutObjectsApi: ", e);
+ return toErrorResponse(e);
+ }
+ }
+}

app/src/main/java/org/vss/api/VssApiEndpoint.java

@@ -0,0 +1,7 @@
+package org.vss.api;
+
+public class VssApiEndpoint {
+ public static final String GET_OBJECT = "/getObject";
+ public static final String PUT_OBJECTS = "/putObjects";
+ public static final String LIST_KEY_VERSIONS = "/listKeyVersions";
+}

app/src/test/java/org/vss/api/GetObjectApiTest.java

@@ -0,0 +1,86 @@
+package org.vss.api;
+
+import com.google.protobuf.ByteString;
+import jakarta.ws.rs.core.Response;
+import java.nio.charset.StandardCharsets;
+import java.util.stream.Stream;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.Arguments;
+import org.junit.jupiter.params.provider.MethodSource;
+import org.vss.ErrorCode;
+import org.vss.ErrorResponse;
+import org.vss.GetObjectRequest;
+import org.vss.GetObjectResponse;
+import org.vss.KVStore;
+import org.vss.KeyValue;
+import org.vss.exception.ConflictException;
+
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.is;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+class GetObjectApiTest {
+ private GetObjectApi getObjectApi;
+ private KVStore mockKVStore;
+
+ private static String TEST_STORE_ID = "storeId";
+ private static String TEST_KEY = "key";
+ private static KeyValue TEST_KV = KeyValue.newBuilder().setKey(TEST_KEY).setValue(
+ ByteString.copyFrom("test_value", StandardCharsets.UTF_8)).build();
+
+ @BeforeEach
+ void setUp() {
+ mockKVStore = mock(KVStore.class);
+ getObjectApi = new GetObjectApi(mockKVStore);
+ }
+
+ @Test
+ void execute_ValidPayload_ReturnsResponse() {
+ GetObjectRequest expectedRequest =
+ GetObjectRequest.newBuilder().setStoreId(TEST_STORE_ID).setKey(TEST_KEY).build();
+ byte[] payload = expectedRequest.toByteArray();
+ GetObjectResponse mockResponse = GetObjectResponse.newBuilder().setValue(TEST_KV).build();
+ when(mockKVStore.get(expectedRequest)).thenReturn(mockResponse);
+
+ Response actualResponse = getObjectApi.execute(payload);
+
+ assertThat(actualResponse.getStatus(), is(Response.Status.OK.getStatusCode()));
+ assertThat(actualResponse.getEntity(), is(mockResponse.toByteArray()));
+ verify(mockKVStore).get(expectedRequest);
+ }
+
+ @ParameterizedTest
+ @MethodSource("provideErrorTestCases")
+ void execute_InvalidPayload_ReturnsErrorResponse(Exception exception,
+ ErrorCode errorCode) {
+ GetObjectRequest expectedRequest = GetObjectRequest.newBuilder()
+ .setStoreId(TEST_STORE_ID)
+ .setKey(TEST_KEY)
+ .build();
+ byte[] payload = expectedRequest.toByteArray();
+ when(mockKVStore.get(any())).thenThrow(exception);
+
+ Response response = getObjectApi.execute(payload);
+
+ ErrorResponse expectedErrorResponse = ErrorResponse.newBuilder()
+ .setErrorCode(errorCode)
+ .setMessage("")
+ .build();
+ assertThat(response.getEntity(), is(expectedErrorResponse.toByteArray()));
+ assertThat(response.getStatus(), is(expectedErrorResponse.getErrorCode().getNumber()));
+ verify(mockKVStore).get(expectedRequest);
+ }
+
+ private static Stream<Arguments> provideErrorTestCases() {
+ return Stream.of(
+ Arguments.of(new ConflictException(""), ErrorCode.CONFLICT_EXCEPTION),
+ Arguments.of(new IllegalArgumentException(""), ErrorCode.INVALID_REQUEST_EXCEPTION),
+ Arguments.of(new RuntimeException(""), ErrorCode.INTERNAL_SERVER_EXCEPTION)
+ );
+ }
+}

app/src/test/java/org/vss/api/ListKeyVersionsApiTest.java

@@ -0,0 +1,92 @@
+package org.vss.api;
+
+import com.google.protobuf.ByteString;
+import jakarta.ws.rs.core.Response;
+import java.nio.charset.StandardCharsets;
+import java.util.List;
+import java.util.stream.Stream;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.Arguments;
+import org.junit.jupiter.params.provider.MethodSource;
+import org.vss.ErrorCode;
+import org.vss.ErrorResponse;
+import org.vss.KVStore;
+import org.vss.KeyValue;
+import org.vss.ListKeyVersionsRequest;
+import org.vss.ListKeyVersionsResponse;
+import org.vss.exception.ConflictException;
+
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.is;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+public class ListKeyVersionsApiTest {
+ private ListKeyVersionsApi listKeyVersionsApi;
+ private KVStore mockKVStore;
+
+ private static String TEST_STORE_ID = "storeId";
+ private static String TEST_KEY = "key";
+ private static KeyValue TEST_KV = KeyValue.newBuilder().setKey(TEST_KEY).setValue(
+ ByteString.copyFrom("test_value", StandardCharsets.UTF_8)).build();
+
+ @BeforeEach
+ void setUp() {
+ mockKVStore = mock(KVStore.class);
+ listKeyVersionsApi = new ListKeyVersionsApi(mockKVStore);
+ }
+
+ @Test
+ void execute_ValidPayload_ReturnsResponse() {
+ ListKeyVersionsRequest expectedRequest =
+ ListKeyVersionsRequest.newBuilder()
+ .setStoreId(TEST_STORE_ID)
+ .setKeyPrefix(TEST_KEY)
+ .build();
+ byte[] payload = expectedRequest.toByteArray();
+ ListKeyVersionsResponse mockResponse = ListKeyVersionsResponse.newBuilder().addAllKeyVersions(
+ List.of(TEST_KV)).build();
+ when(mockKVStore.listKeyVersions(expectedRequest)).thenReturn(mockResponse);
+
+ Response actualResponse = listKeyVersionsApi.execute(payload);
+
+ assertThat(actualResponse.getStatus(), is(Response.Status.OK.getStatusCode()));
+ assertThat(actualResponse.getEntity(), is(mockResponse.toByteArray()));
+ verify(mockKVStore).listKeyVersions(expectedRequest);
+ }
+
+ @ParameterizedTest
+ @MethodSource("provideErrorTestCases")
+ void execute_InvalidPayload_ReturnsErrorResponse(Exception exception,
+ ErrorCode errorCode) {
+ ListKeyVersionsRequest expectedRequest =
+ ListKeyVersionsRequest.newBuilder()
+ .setStoreId(TEST_STORE_ID)
+ .setKeyPrefix(TEST_KEY)
+ .build();
+ byte[] payload = expectedRequest.toByteArray();
+ when(mockKVStore.listKeyVersions(any())).thenThrow(exception);
+
+ Response response = listKeyVersionsApi.execute(payload);
+
+ ErrorResponse expectedErrorResponse = ErrorResponse.newBuilder()
+ .setErrorCode(errorCode)
+ .setMessage("")
+ .build();
+ assertThat(response.getEntity(), is(expectedErrorResponse.toByteArray()));
+ assertThat(response.getStatus(), is(expectedErrorResponse.getErrorCode().getNumber()));
+ verify(mockKVStore).listKeyVersions(expectedRequest);
+ }
+
+ private static Stream<Arguments> provideErrorTestCases() {
+ return Stream.of(
+ Arguments.of(new ConflictException(""), ErrorCode.CONFLICT_EXCEPTION),
+ Arguments.of(new IllegalArgumentException(""), ErrorCode.INVALID_REQUEST_EXCEPTION),
+ Arguments.of(new RuntimeException(""), ErrorCode.INTERNAL_SERVER_EXCEPTION)
+ );
+ }
+}

app/build.gradle

@@ -4,6 +4,7 @@ buildscript {
  ext.protobufVersion = '3.21.7'
  ext.jerseyVersion = '3.1.0'
  ext.junitVersion = '5.9.0'
+ ext.mockitoVersion = '5.2.0'
  ext.postgresVersion = '42.5.1'
  ext.jooqVersion = '3.17.7'
  ext.guiceVersion = '5.1.0'
@@ -33,6 +34,8 @@ version '1.0'
 dependencies {
  implementation "com.google.protobuf:protobuf-java:$protobufVersion"
 
+ implementation('org.glassfish.jersey.containers:jersey-container-servlet:3.1.0')
+
  //jOOQ & Postgres impl deps
  implementation "org.jooq:jooq:$jooqVersion"
  implementation "org.jooq:jooq-meta:$jooqVersion"
@@ -42,7 +45,12 @@ dependencies {
 
  implementation "com.google.inject:guice:$guiceVersion"
 
+ compileOnly 'org.projectlombok:lombok:1.18.24'
+ annotationProcessor 'org.projectlombok:lombok:1.18.24'
+
  testImplementation "org.junit.jupiter:junit-jupiter-api:$junitVersion"
+ testImplementation "org.junit.jupiter:junit-jupiter-params:$junitVersion"
+ testImplementation "org.mockito:mockito-core:$mockitoVersion"
  testRuntimeOnly "org.junit.jupiter:junit-jupiter-engine:$junitVersion"
  testImplementation "org.hamcrest:hamcrest-library:2.2"
  testImplementation "org.testcontainers:junit-jupiter:1.17.6"

app/src/main/java/org/vss/VSSApplication.java

@@ -0,0 +1,10 @@
+package org.vss;
+
+import jakarta.ws.rs.ApplicationPath;
+import org.glassfish.jersey.server.ResourceConfig;
+
+@ApplicationPath("/")
+public class VSSApplication extends ResourceConfig {
+ public VSSApplication() {
+ }
+}

app/src/main/java/org/vss/api/AbstractVssApi.java

@@ -0,0 +1,48 @@
+package org.vss.api;
+
+import com.google.protobuf.GeneratedMessageV3;
+import com.google.protobuf.InvalidProtocolBufferException;
+import jakarta.inject.Inject;
+import jakarta.ws.rs.core.Response;
+import org.vss.ErrorCode;
+import org.vss.ErrorResponse;
+import org.vss.KVStore;
+import org.vss.exception.ConflictException;
+
+public abstract class AbstractVssApi {
+ final KVStore kvStore;
+
+ @Inject
+ public AbstractVssApi(KVStore kvStore) {
+ this.kvStore = kvStore;
+ }
+
+ Response toResponse(GeneratedMessageV3 protoResponse) {
+
+ return Response
+ .status(Response.Status.OK)
+ .entity(protoResponse.toByteArray())
+ .build();
+ }
+
+ Response toErrorResponse(Exception e) {
+ ErrorCode errorCode;
+ if (e instanceof ConflictException) {
+ errorCode = ErrorCode.CONFLICT_EXCEPTION;
+ } else if (e instanceof IllegalArgumentException
+ || e instanceof InvalidProtocolBufferException) {
+ errorCode = ErrorCode.INVALID_REQUEST_EXCEPTION;
+ } else {
+ errorCode = ErrorCode.INTERNAL_SERVER_EXCEPTION;
+ }
+
+ ErrorResponse errorResponse = ErrorResponse.newBuilder()
+ .setErrorCode(errorCode)
+ .setMessage(e.getMessage())
+ .build();
+
+ return Response.status(errorCode.getNumber())
+ .entity(errorResponse.toByteArray())
+ .build();
+ }
+}

app/src/main/java/org/vss/api/GetObjectApi.java

@@ -0,0 +1,35 @@
+package org.vss.api;
+
+import jakarta.inject.Inject;
+import jakarta.ws.rs.POST;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.Produces;
+import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
+import lombok.extern.slf4j.Slf4j;
+import org.vss.GetObjectRequest;
+import org.vss.GetObjectResponse;
+import org.vss.KVStore;
+
+@Path(VssApiEndpoint.GET_OBJECT)
+@Slf4j
+public class GetObjectApi extends AbstractVssApi {
+
+ @Inject
+ public GetObjectApi(KVStore kvstore) {
+ super(kvstore);
+ }
+
+ @POST
+ @Produces(MediaType.APPLICATION_OCTET_STREAM)
+ public Response execute(byte[] payload) {
+ try {
+ GetObjectRequest request = GetObjectRequest.parseFrom(payload);
+ GetObjectResponse response = kvStore.get(request);
+ return toResponse(response);
+ } catch (Exception e) {
+ log.error("Exception in GetObjectApi: ", e);
+ return toErrorResponse(e);
+ }
+ }
+}

app/src/main/java/org/vss/api/ListKeyVersionsApi.java

@@ -0,0 +1,35 @@
+package org.vss.api;
+
+import jakarta.inject.Inject;
+import jakarta.ws.rs.POST;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.Produces;
+import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
+import lombok.extern.slf4j.Slf4j;
+import org.vss.KVStore;
+import org.vss.ListKeyVersionsRequest;
+import org.vss.ListKeyVersionsResponse;
+
+@Path(VssApiEndpoint.LIST_KEY_VERSIONS)
+@Slf4j
+public class ListKeyVersionsApi extends AbstractVssApi {
+
+ @Inject
+ public ListKeyVersionsApi(KVStore kvStore) {
+ super(kvStore);
+ }
+
+ @POST
+ @Produces(MediaType.APPLICATION_OCTET_STREAM)
+ public Response execute(byte[] payload) {
+ try {
+ ListKeyVersionsRequest request = ListKeyVersionsRequest.parseFrom(payload);
+ ListKeyVersionsResponse response = kvStore.listKeyVersions(request);
+ return toResponse(response);
+ } catch (Exception e) {
+ log.error("Exception in ListKeyVersionsApi: ", e);
+ return toErrorResponse(e);
+ }
+ }
+}

app/src/main/java/org/vss/api/PutObjectsApi.java

@@ -0,0 +1,35 @@
+package org.vss.api;
+
+import jakarta.inject.Inject;
+import jakarta.ws.rs.POST;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.Produces;
+import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
+import lombok.extern.slf4j.Slf4j;
+import org.vss.KVStore;
+import org.vss.PutObjectRequest;
+import org.vss.PutObjectResponse;
+
+@Path(VssApiEndpoint.PUT_OBJECTS)
+@Slf4j
+public class PutObjectsApi extends AbstractVssApi {
+
+ @Inject
+ public PutObjectsApi(KVStore kvStore) {
+ super(kvStore);
+ }
+
+ @POST
+ @Produces(MediaType.APPLICATION_OCTET_STREAM)
+ public Response execute(byte[] payload) {
+ try {
+ PutObjectRequest putObjectRequest = PutObjectRequest.parseFrom(payload);
+ PutObjectResponse response = kvStore.put(putObjectRequest);
+ return toResponse(response);
+ } catch (Exception e) {
+ log.error("Exception in PutObjectsApi: ", e);
+ return toErrorResponse(e);
+ }
+ }
+}

app/src/main/java/org/vss/api/VssApiEndpoint.java

@@ -0,0 +1,7 @@
+package org.vss.api;
+
+public class VssApiEndpoint {
+ public static final String GET_OBJECT = "/getObject";
+ public static final String PUT_OBJECTS = "/putObjects";
+ public static final String LIST_KEY_VERSIONS = "/listKeyVersions";
+}

app/src/test/java/org/vss/api/GetObjectApiTest.java

@@ -0,0 +1,86 @@
+package org.vss.api;
+
+import com.google.protobuf.ByteString;
+import jakarta.ws.rs.core.Response;
+import java.nio.charset.StandardCharsets;
+import java.util.stream.Stream;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.Arguments;
+import org.junit.jupiter.params.provider.MethodSource;
+import org.vss.ErrorCode;
+import org.vss.ErrorResponse;
+import org.vss.GetObjectRequest;
+import org.vss.GetObjectResponse;
+import org.vss.KVStore;
+import org.vss.KeyValue;
+import org.vss.exception.ConflictException;
+
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.is;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+class GetObjectApiTest {
+ private GetObjectApi getObjectApi;
+ private KVStore mockKVStore;
+
+ private static String TEST_STORE_ID = "storeId";
+ private static String TEST_KEY = "key";
+ private static KeyValue TEST_KV = KeyValue.newBuilder().setKey(TEST_KEY).setValue(
+ ByteString.copyFrom("test_value", StandardCharsets.UTF_8)).build();
+
+ @BeforeEach
+ void setUp() {
+ mockKVStore = mock(KVStore.class);
+ getObjectApi = new GetObjectApi(mockKVStore);
+ }
+
+ @Test
+ void execute_ValidPayload_ReturnsResponse() {
+ GetObjectRequest expectedRequest =
+ GetObjectRequest.newBuilder().setStoreId(TEST_STORE_ID).setKey(TEST_KEY).build();
+ byte[] payload = expectedRequest.toByteArray();
+ GetObjectResponse mockResponse = GetObjectResponse.newBuilder().setValue(TEST_KV).build();
+ when(mockKVStore.get(expectedRequest)).thenReturn(mockResponse);
+
+ Response actualResponse = getObjectApi.execute(payload);
+
+ assertThat(actualResponse.getStatus(), is(Response.Status.OK.getStatusCode()));
+ assertThat(actualResponse.getEntity(), is(mockResponse.toByteArray()));
+ verify(mockKVStore).get(expectedRequest);
+ }
+
+ @ParameterizedTest
+ @MethodSource("provideErrorTestCases")
+ void execute_InvalidPayload_ReturnsErrorResponse(Exception exception,
+ ErrorCode errorCode) {
+ GetObjectRequest expectedRequest = GetObjectRequest.newBuilder()
+ .setStoreId(TEST_STORE_ID)
+ .setKey(TEST_KEY)
+ .build();
+ byte[] payload = expectedRequest.toByteArray();
+ when(mockKVStore.get(any())).thenThrow(exception);
+
+ Response response = getObjectApi.execute(payload);
+
+ ErrorResponse expectedErrorResponse = ErrorResponse.newBuilder()
+ .setErrorCode(errorCode)
+ .setMessage("")
+ .build();
+ assertThat(response.getEntity(), is(expectedErrorResponse.toByteArray()));
+ assertThat(response.getStatus(), is(expectedErrorResponse.getErrorCode().getNumber()));
+ verify(mockKVStore).get(expectedRequest);
+ }
+
+ private static Stream<Arguments> provideErrorTestCases() {
+ return Stream.of(
+ Arguments.of(new ConflictException(""), ErrorCode.CONFLICT_EXCEPTION),
+ Arguments.of(new IllegalArgumentException(""), ErrorCode.INVALID_REQUEST_EXCEPTION),
+ Arguments.of(new RuntimeException(""), ErrorCode.INTERNAL_SERVER_EXCEPTION)
+ );
+ }
+}

app/src/test/java/org/vss/api/ListKeyVersionsApiTest.java

@@ -0,0 +1,92 @@
+package org.vss.api;
+
+import com.google.protobuf.ByteString;
+import jakarta.ws.rs.core.Response;
+import java.nio.charset.StandardCharsets;
+import java.util.List;
+import java.util.stream.Stream;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.Arguments;
+import org.junit.jupiter.params.provider.MethodSource;
+import org.vss.ErrorCode;
+import org.vss.ErrorResponse;
+import org.vss.KVStore;
+import org.vss.KeyValue;
+import org.vss.ListKeyVersionsRequest;
+import org.vss.ListKeyVersionsResponse;
+import org.vss.exception.ConflictException;
+
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.is;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+public class ListKeyVersionsApiTest {
+ private ListKeyVersionsApi listKeyVersionsApi;
+ private KVStore mockKVStore;
+
+ private static String TEST_STORE_ID = "storeId";
+ private static String TEST_KEY = "key";
+ private static KeyValue TEST_KV = KeyValue.newBuilder().setKey(TEST_KEY).setValue(
+ ByteString.copyFrom("test_value", StandardCharsets.UTF_8)).build();
+
+ @BeforeEach
+ void setUp() {
+ mockKVStore = mock(KVStore.class);
+ listKeyVersionsApi = new ListKeyVersionsApi(mockKVStore);
+ }
+
+ @Test
+ void execute_ValidPayload_ReturnsResponse() {
+ ListKeyVersionsRequest expectedRequest =
+ ListKeyVersionsRequest.newBuilder()
+ .setStoreId(TEST_STORE_ID)
+ .setKeyPrefix(TEST_KEY)
+ .build();
+ byte[] payload = expectedRequest.toByteArray();
+ ListKeyVersionsResponse mockResponse = ListKeyVersionsResponse.newBuilder().addAllKeyVersions(
+ List.of(TEST_KV)).build();
+ when(mockKVStore.listKeyVersions(expectedRequest)).thenReturn(mockResponse);
+
+ Response actualResponse = listKeyVersionsApi.execute(payload);
+
+ assertThat(actualResponse.getStatus(), is(Response.Status.OK.getStatusCode()));
+ assertThat(actualResponse.getEntity(), is(mockResponse.toByteArray()));
+ verify(mockKVStore).listKeyVersions(expectedRequest);
+ }
+
+ @ParameterizedTest
+ @MethodSource("provideErrorTestCases")
+ void execute_InvalidPayload_ReturnsErrorResponse(Exception exception,
+ ErrorCode errorCode) {
+ ListKeyVersionsRequest expectedRequest =
+ ListKeyVersionsRequest.newBuilder()
+ .setStoreId(TEST_STORE_ID)
+ .setKeyPrefix(TEST_KEY)
+ .build();
+ byte[] payload = expectedRequest.toByteArray();
+ when(mockKVStore.listKeyVersions(any())).thenThrow(exception);
+
+ Response response = listKeyVersionsApi.execute(payload);
+
+ ErrorResponse expectedErrorResponse = ErrorResponse.newBuilder()
+ .setErrorCode(errorCode)
+ .setMessage("")
+ .build();
+ assertThat(response.getEntity(), is(expectedErrorResponse.toByteArray()));
+ assertThat(response.getStatus(), is(expectedErrorResponse.getErrorCode().getNumber()));
+ verify(mockKVStore).listKeyVersions(expectedRequest);
+ }
+
+ private static Stream<Arguments> provideErrorTestCases() {
+ return Stream.of(
+ Arguments.of(new ConflictException(""), ErrorCode.CONFLICT_EXCEPTION),
+ Arguments.of(new IllegalArgumentException(""), ErrorCode.INVALID_REQUEST_EXCEPTION),
+ Arguments.of(new RuntimeException(""), ErrorCode.INTERNAL_SERVER_EXCEPTION)
+ );
+ }
+}