wip

CMSgov · Mar 7, 2024 · 5d2aa14 · 5d2aa14
1 parent d7ff459
commit 5d2aa14
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 5 deletions.
diff --git a/optout/src/main/java/gov/cms/ab2d/optout/OptOutS3.java b/optout/src/main/java/gov/cms/ab2d/optout/OptOutS3.java
@@ -62,9 +62,7 @@ public BufferedReader openFileS3() {
 
     public String createResponseOptOutFile(String responseContent) {
         try {
-            var key = CONF_FILE_NAME
-                    + new SimpleDateFormat(CONF_FILE_NAME_PATTERN).format(new Date())
-                    + CONF_FILE_FORMAT;
+            var key = getOutFileName();
 
             var objectRequest = PutObjectRequest.builder()
                     .bucket(bfdBucket)
@@ -93,5 +91,16 @@ public void deleteFileFromS3() {
         }
     }
 
+    public String getOutFileName() {
+        //bfdeft01/ab2d/in/testing.txt
+        var name = CONF_FILE_NAME
+                + new SimpleDateFormat(CONF_FILE_NAME_PATTERN).format(new Date())
+                + CONF_FILE_FORMAT;
+
+        String[] path = fileName.split("in");
+
+      return path[0] + "out/" + name;
+    }
+
 
 }
diff --git a/optout/src/test/java/gov/cms/ab2d/optout/OptOutS3Test.java b/optout/src/test/java/gov/cms/ab2d/optout/OptOutS3Test.java
@@ -12,8 +12,7 @@
 import java.nio.file.Files;
 import java.nio.file.Paths;
 
-import static gov.cms.ab2d.optout.OptOutConstants.TEST_BFD_BUCKET_NAME;
-import static gov.cms.ab2d.optout.OptOutConstants.TEST_FILE_NAME;
+import static gov.cms.ab2d.optout.OptOutConstants.*;
 import static gov.cms.ab2d.optout.S3MockAPIExtension.S3_CLIENT;
 import static org.junit.jupiter.api.Assertions.assertThrows;
 import static org.junit.jupiter.api.Assertions.assertTrue;
@@ -58,4 +57,13 @@ void deleteFileFromS3Test() {
         OPT_OUT_S3.deleteFileFromS3();
         Assertions.assertFalse(S3MockAPIExtension.isObjectExists(TEST_FILE_NAME));
     }
+
+    @Test
+    void getOutFileName() {
+        var subFolders = "bfdeft01/ab2d";
+        OPT_OUT_S3 = new OptOutS3(S3_CLIENT, subFolders + "/in/" + TEST_FILE_NAME, TEST_BFD_BUCKET_NAME, mock(LambdaLogger.class));
+        var outFileName = OPT_OUT_S3.getOutFileName();
+        Assertions.assertTrue(outFileName.startsWith(subFolders + "/out/" + CONF_FILE_NAME));
+        Assertions.assertTrue(outFileName.endsWith(CONF_FILE_FORMAT));
+    }
 }