1414import com .uid2 .shared .secure .AttestationFailure ;
1515import com .uid2 .shared .secure .AttestationResult ;
1616import com .uid2 .shared .secure .ICoreAttestationService ;
17- import com .uid2 .shared .store .reader .RotatingS3KeyProvider ;
17+ import com .uid2 .shared .store .reader .RotatingCloudEncryptionKeyProvider ;
1818import io .vertx .core .*;
1919import io .vertx .core .buffer .Buffer ;
2020import io .vertx .core .json .JsonArray ;
2424import io .vertx .junit5 .VertxExtension ;
2525import io .vertx .junit5 .VertxTestContext ;
2626
27- import static com .uid2 .core .service .KeyMetadataProvider .KeysMetadataPathName ;
28- import static com .uid2 .shared .Const .Config .KeysetsMetadataPathProp ;
2927import static org .junit .jupiter .api .Assertions .*;
3028
3129import org .junit .jupiter .api .BeforeEach ;
4947import java .util .*;
5048import java .util .concurrent .Callable ;
5149
52- import com .uid2 .shared .model .S3Key ;
50+ import com .uid2 .shared .model .CloudEncryptionKey ;
5351import java .util .Arrays ;
5452
5553import static org .mockito .Mockito .*;
@@ -71,7 +69,7 @@ public class TestCoreVerticle {
7169 @ Mock
7270 private JwtService jwtService ;
7371 @ Mock
74- private RotatingS3KeyProvider s3KeyProvider ;
72+ private RotatingCloudEncryptionKeyProvider cloudEncryptionKeyProvider ;
7573
7674 private AttestationService attestationService ;
7775
@@ -118,7 +116,7 @@ void deployVerticle(TestInfo info, Vertx vertx, VertxTestContext testContext) th
118116 }
119117 });
120118
121- CoreVerticle verticle = new CoreVerticle (cloudStorage , authProvider , attestationService , attestationTokenService , enclaveIdentifierProvider , operatorJWTTokenProvider , jwtService , s3KeyProvider );
119+ CoreVerticle verticle = new CoreVerticle (cloudStorage , authProvider , attestationService , attestationTokenService , enclaveIdentifierProvider , operatorJWTTokenProvider , jwtService , cloudEncryptionKeyProvider );
122120 vertx .deployVerticle (verticle , testContext .succeeding (id -> testContext .completeNow ()));
123121
124122 }
@@ -641,7 +639,7 @@ void wrongMethodForEndpoint(Vertx vertx, VertxTestContext testContext) {
641639
642640 @ Test
643641 void wrongMethodForEndpointS3 (Vertx vertx , VertxTestContext testContext ) {
644- post (vertx , "/s3encryption_keys /retrieve" , makeAttestationRequestJson (null , null ), ar -> {
642+ post (vertx , "/cloud_encryption_keys /retrieve" , makeAttestationRequestJson (null , null ), ar -> {
645643 try {
646644 HttpResponse response = ar .result ();
647645 assertEquals (405 , response .statusCode ());
@@ -655,37 +653,37 @@ void wrongMethodForEndpointS3(Vertx vertx, VertxTestContext testContext) {
655653
656654 @ Tag ("dontForceJwt" )
657655 @ Test
658- void s3encryptionKeyRetrieveSuccess (Vertx vertx , VertxTestContext testContext ) {
656+ void cloudEncryptionKeyRetrieveSuccess (Vertx vertx , VertxTestContext testContext ) {
659657 fakeAuth (attestationProtocolPublic , Role .OPERATOR );
660658 addAttestationProvider (attestationProtocolPublic );
661659 onHandleAttestationRequest (() -> {
662660 byte [] resultPublicKey = null ;
663661 return Future .succeededFuture (new AttestationResult (resultPublicKey , "test" ));
664662 });
665663
666- S3Key key = new S3Key (1 , 88 , 1687635529 , 1687808329 , "newSecret" );
664+ CloudEncryptionKey key = new CloudEncryptionKey (1 , 88 , 1687635529 , 1687808329 , "newSecret" );
667665
668- List <S3Key > keys = Arrays .asList (key );
669- when (s3KeyProvider .getKeys (88 )).thenReturn (keys );
666+ List <CloudEncryptionKey > keys = Arrays .asList (key );
667+ when (cloudEncryptionKeyProvider .getKeys (88 )).thenReturn (keys );
670668
671- get (vertx , "s3encryption_keys /retrieve" , ar -> {
669+ get (vertx , "cloud_encryption_keys /retrieve" , ar -> {
672670 try {
673671 if (ar .succeeded ()) {
674672 HttpResponse <Buffer > response = ar .result ();
675673 assertEquals (200 , response .statusCode ());
676674
677675 JsonObject json = response .bodyAsJsonObject ();
678- JsonArray s3KeysArray = json .getJsonArray ("s3Keys " );
676+ JsonArray cloudEncryptionKeysArray = json .getJsonArray ("cloudEncryptionKeys " );
679677
680- assertNotNull (s3KeysArray );
681- assertEquals (1 , s3KeysArray .size ());
678+ assertNotNull ( cloudEncryptionKeysArray );
679+ assertEquals (1 , cloudEncryptionKeysArray .size ());
682680
683- JsonObject s3KeyJson = s3KeysArray .getJsonObject (0 );
684- assertEquals (1 , s3KeyJson .getInteger ("id" ));
685- assertEquals (88 , s3KeyJson .getInteger ("siteId" ));
686- assertEquals (1687635529 , s3KeyJson .getLong ("activates" ));
687- assertEquals (1687808329 , s3KeyJson .getLong ("created" ));
688- assertEquals ("newSecret" , s3KeyJson .getString ("secret" ));
681+ JsonObject cloudEncryptionKeyJson = cloudEncryptionKeysArray .getJsonObject (0 );
682+ assertEquals (1 , cloudEncryptionKeyJson .getInteger ("id" ));
683+ assertEquals (88 , cloudEncryptionKeyJson .getInteger ("siteId" ));
684+ assertEquals (1687635529 , cloudEncryptionKeyJson .getLong ("activates" ));
685+ assertEquals (1687808329 , cloudEncryptionKeyJson .getLong ("created" ));
686+ assertEquals ("newSecret" , cloudEncryptionKeyJson .getString ("secret" ));
689687
690688 testContext .completeNow ();
691689 } else {
@@ -700,41 +698,41 @@ void s3encryptionKeyRetrieveSuccess(Vertx vertx, VertxTestContext testContext) {
700698
701699 @ Tag ("dontForceJwt" )
702700 @ Test
703- void s3encryptionKeyRetrieveSuccessWithThreeKeys (Vertx vertx , VertxTestContext testContext ) {
701+ void cloudEncryptionencryptionKeyRetrieveSuccessWithThreeKeys (Vertx vertx , VertxTestContext testContext ) {
704702 fakeAuth (attestationProtocolPublic , Role .OPERATOR );
705703 addAttestationProvider (attestationProtocolPublic );
706704 onHandleAttestationRequest (() -> {
707705 byte [] resultPublicKey = null ;
708706 return Future .succeededFuture (new AttestationResult (resultPublicKey , "test" ));
709707 });
710708
711- // Create 3 S3Key objects
712- S3Key key1 = new S3Key (1 , 88 , 1687635529 , 1687808329 , "secret1" );
713- S3Key key2 = new S3Key (2 , 88 , 1687635530 , 1687808330 , "secret2" );
714- S3Key key3 = new S3Key (3 , 88 , 1687635531 , 1687808331 , "secret3" );
709+ // Create 3 CloudEncryptionKey objects
710+ CloudEncryptionKey key1 = new CloudEncryptionKey (1 , 88 , 1687635529 , 1687808329 , "secret1" );
711+ CloudEncryptionKey key2 = new CloudEncryptionKey (2 , 88 , 1687635530 , 1687808330 , "secret2" );
712+ CloudEncryptionKey key3 = new CloudEncryptionKey (3 , 88 , 1687635531 , 1687808331 , "secret3" );
715713
716- List <S3Key > keys = Arrays .asList (key1 , key2 , key3 );
717- when (s3KeyProvider .getKeys (88 )).thenReturn (keys );
714+ List <CloudEncryptionKey > keys = Arrays .asList (key1 , key2 , key3 );
715+ when (cloudEncryptionKeyProvider .getKeys (88 )).thenReturn (keys );
718716
719- get (vertx , "s3encryption_keys /retrieve" , ar -> {
717+ get (vertx , "cloud_encryption_keys /retrieve" , ar -> {
720718 try {
721719 if (ar .succeeded ()) {
722720 HttpResponse <Buffer > response = ar .result ();
723721 assertEquals (200 , response .statusCode ());
724722
725723 JsonObject json = response .bodyAsJsonObject ();
726- JsonArray s3KeysArray = json .getJsonArray ("s3Keys " );
724+ JsonArray cloudEncryptionKeysArray = json .getJsonArray ("cloudEncryptionKeys " );
727725
728- assertNotNull (s3KeysArray );
729- assertEquals (3 , s3KeysArray .size ());
726+ assertNotNull (cloudEncryptionKeysArray );
727+ assertEquals (3 , cloudEncryptionKeysArray .size ());
730728
731729 for (int i = 0 ; i < 3 ; i ++) {
732- JsonObject s3KeyJson = s3KeysArray .getJsonObject (i );
733- assertEquals (i + 1 , s3KeyJson .getInteger ("id" ));
734- assertEquals (88 , s3KeyJson .getInteger ("siteId" ));
735- assertEquals (1687635529 + i , s3KeyJson .getLong ("activates" ));
736- assertEquals (1687808329 + i , s3KeyJson .getLong ("created" ));
737- assertEquals ("secret" + (i + 1 ), s3KeyJson .getString ("secret" ));
730+ JsonObject cloudEncryptionKeyJson = cloudEncryptionKeysArray .getJsonObject (i );
731+ assertEquals (i + 1 , cloudEncryptionKeyJson .getInteger ("id" ));
732+ assertEquals (88 , cloudEncryptionKeyJson .getInteger ("siteId" ));
733+ assertEquals (1687635529 + i , cloudEncryptionKeyJson .getLong ("activates" ));
734+ assertEquals (1687808329 + i , cloudEncryptionKeyJson .getLong ("created" ));
735+ assertEquals ("secret" + (i + 1 ), cloudEncryptionKeyJson .getString ("secret" ));
738736 }
739737
740738 testContext .completeNow ();
@@ -749,7 +747,7 @@ void s3encryptionKeyRetrieveSuccessWithThreeKeys(Vertx vertx, VertxTestContext t
749747
750748 @ Tag ("dontForceJwt" )
751749 @ Test
752- void s3encryptionKeyRetrieveNoKeysOrError (Vertx vertx , VertxTestContext testContext ) {
750+ void cloudEncryptionKeyRetrieveNoKeysOrError (Vertx vertx , VertxTestContext testContext ) {
753751 fakeAuth (attestationProtocolPublic , Role .OPERATOR );
754752 addAttestationProvider (attestationProtocolPublic );
755753 onHandleAttestationRequest (() -> {
@@ -758,22 +756,22 @@ void s3encryptionKeyRetrieveNoKeysOrError(Vertx vertx, VertxTestContext testCont
758756 });
759757
760758 // Test case 1: No keys found
761- when (s3KeyProvider .getKeys (anyInt ())).thenReturn (Collections .emptyList ());
759+ when (cloudEncryptionKeyProvider .getKeys (anyInt ())).thenReturn (Collections .emptyList ());
762760
763- get (vertx , "s3encryption_keys /retrieve" , ar -> {
761+ get (vertx , "cloud_encryption_keys /retrieve" , ar -> {
764762 try {
765763 if (ar .succeeded ()) {
766764 HttpResponse <Buffer > response = ar .result ();
767765 assertEquals (500 , response .statusCode ());
768766
769767 JsonObject json = response .bodyAsJsonObject ();
770- assertEquals ("No S3 keys found" , json .getString ("status" ));
771- assertTrue (json .getString ("message" ).contains ("No S3 keys found for siteId:" ));
768+ assertEquals ("No Cloud Encryption keys found" , json .getString ("status" ));
769+ assertTrue (json .getString ("message" ).contains ("No Cloud Encryption keys found for siteId:" ));
772770
773771 // Test case 2: Exception thrown
774- when (s3KeyProvider .getKeys (anyInt ())).thenThrow (new RuntimeException ("Test exception" ));
772+ when (cloudEncryptionKeyProvider .getKeys (anyInt ())).thenThrow (new RuntimeException ("Test exception" ));
775773
776- get (vertx , "s3encryption_keys /retrieve" , ar2 -> {
774+ get (vertx , "cloud_encryption_keys /retrieve" , ar2 -> {
777775 if (ar2 .succeeded ()) {
778776 HttpResponse <Buffer > response2 = ar2 .result ();
779777 assertEquals (500 , response2 .statusCode ());
0 commit comments