Skip to content

Commit 920eeb3

Browse files
Unit tests added, configuration.java updated
1 parent 07d6e14 commit 920eeb3

File tree

3 files changed

+27
-4
lines changed

3 files changed

+27
-4
lines changed

zebedee-cms/src/main/java/com/github/onsdigital/zebedee/configuration/Configuration.java

+7-1
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,8 @@ public class Configuration {
4444
private static final String KEYRING_SECRET_KEY = "KEYRING_SECRET_KEY";
4545
private static final String KEYRING_INIT_VECTOR = "KEYRING_INIT_VECTOR";
4646
private static final String DATASET_WHITELIST = "drsi,mm23,mm22,ppi,dataset1,pusf,a01,x09,cla01,pn2,mgdp,diop,ios1,mret,mq10,rtisa";
47-
private static final String NON_TS_DATASET_WHITELIST = "dataset1, a01, x09, cla01, rtisa";
47+
private static final String NON_TS_DATASET_WHITELIST = "dataset1,a01,x09,cla01,rtisa";
48+
private static final String EXPECTED_DATASET1_PATH = "economy/inflationandpriceindices/datasets/growthratesofoutputandinputproducerpriceinflation";
4849

4950
private static final int VERIFY_RETRY_DELAY = 5000; // milliseconds
5051
private static final int VERIFY_RETRY_COUNT = 10;
@@ -268,6 +269,11 @@ public static String getNonTsDatasetWhitelist() {
268269
return StringUtils.defaultIfBlank(getValue("NON_TS_DATASET_WHITELIST"), NON_TS_DATASET_WHITELIST);
269270
}
270271

272+
// Get dataset1 path from dp-configs
273+
public static String getDataset1ExpectedPath() {
274+
return StringUtils.defaultIfBlank(getValue("EXPECTED_DATASET1_PATH"), EXPECTED_DATASET1_PATH);
275+
}
276+
271277
// Changeable upload-new endpoint parameters
272278
public static String getResumableType() {
273279
return StringUtils.defaultIfBlank(getValue("RESUMABLE_TYPE"), RESUMABLE_TYPE);

zebedee-cms/src/main/java/com/github/onsdigital/zebedee/model/approval/ApproveTask.java

+2-1
Original file line numberDiff line numberDiff line change
@@ -403,7 +403,8 @@ protected void uploadFile(Resource myFile, String fileName, String collectionId)
403403
String nonTsDatasetWhitelist = Configuration.getNonTsDatasetWhitelist();
404404
Set<String> nonTsDatasetWhitelistSet = Arrays.stream(nonTsDatasetWhitelist.split(","))
405405
.collect(Collectors.toSet());
406-
String expectedDataset1Path = "economy/inflationandpriceindices/datasets/growthratesofoutputandinputproducerpriceinflation";
406+
407+
String expectedDataset1Path = Configuration.getDataset1ExpectedPath();
407408
if (nonTsDatasetWhitelistSet.contains(baseFilename)) {
408409
if (baseFilename.contains("dataset1")) {
409410
// identify if its the PPI dataset

zebedee-cms/src/test/java/com/github/onsdigital/zebedee/model/approval/ApproveTaskTest.java

+18-2
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,6 @@
1717
import com.github.onsdigital.zebedee.session.model.Session;
1818
import com.github.onsdigital.zebedee.util.slack.Notifier;
1919
import org.apache.hc.core5.http.NameValuePair;
20-
import org.joda.time.LocalDate;
2120
import org.junit.After;
2221
import org.junit.Assert;
2322
import org.junit.Before;
@@ -29,7 +28,6 @@
2928
import java.nio.file.Path;
3029
import java.text.ParseException;
3130
import java.text.SimpleDateFormat;
32-
import java.time.LocalDateTime;
3331
import java.util.ArrayList;
3432
import java.util.Date;
3533
import java.util.List;
@@ -520,4 +518,22 @@ public void testfilePathGenerator(){
520518
assertEquals(task.filePathGenerator("rtisa", publishDate, "v123"), "ts-datasets/other/2024-07-18");
521519
assertEquals(task.filePathGenerator("cla01", publishDate, "v123"), "ts-datasets/other/2024-07-18");
522520
}
521+
522+
@Test
523+
public void testfilePathGenerator_check_default_value_dataset1() {
524+
String actual = Configuration.getDataset1ExpectedPath();
525+
String expected = "economy/inflationandpriceindices/datasets/growthratesofoutputandinputproducerpriceinflation";
526+
527+
assertEquals(expected, actual);
528+
}
529+
530+
@Test
531+
public void testfilePathGenerator_check_configured_value_dataset1() {
532+
System.setProperty("EXPECTED_DATASET1_PATH", "economy/inflationandpriceindices/datasets/");
533+
String actual = Configuration.getDataset1ExpectedPath();
534+
String expected = "economy/inflationandpriceindices/datasets/";
535+
536+
assertEquals(expected, actual);
537+
}
538+
523539
}

0 commit comments

Comments
 (0)