Skip to content

Commit 9141565

Browse files
committed
Make validateDateRange non-default
1 parent 3229f47 commit 9141565

File tree

5 files changed

+23
-62
lines changed

5 files changed

+23
-62
lines changed

dumper/app/src/main/java/com/google/edwmigration/dumper/application/dumper/connector/Connector.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ default String getDescription() {
5454
@OverridingMethodsMustInvokeSuper
5555
default void validate(@Nonnull ConnectorArguments arguments) {}
5656

57-
default void validateDateRange(ConnectorArguments arguments) {
57+
static void validateDateRange(@Nonnull ConnectorArguments arguments) {
5858
ZonedDateTime startDate = arguments.getStartDate();
5959
ZonedDateTime endDate = arguments.getEndDate();
6060

dumper/app/src/main/java/com/google/edwmigration/dumper/application/dumper/connector/airflow/AirflowConnector.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -236,7 +236,7 @@ public void validate(@Nonnull ConnectorArguments arguments) {
236236
Preconditions.checkState(arguments.getSchema() != null, "--schema is required with --host");
237237
}
238238

239-
validateDateRange(arguments);
239+
Connector.validateDateRange(arguments);
240240
}
241241

242242
@Nonnull

dumper/app/src/main/java/com/google/edwmigration/dumper/application/dumper/connector/cloudera/manager/ClouderaManagerConnector.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,7 @@ public void validate(@Nonnull ConnectorArguments arguments) {
137137
Preconditions.checkNotNull(
138138
clouderaUser, "--user is required for Cloudera Manager API connector");
139139

140-
validateDateRange(arguments);
140+
Connector.validateDateRange(arguments);
141141
}
142142

143143
private void doClouderaManagerLogin(

dumper/app/src/main/java/com/google/edwmigration/dumper/application/dumper/connector/hadoop/oozie/OozieConnector.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ public String getDefaultFileName(boolean isAssessment, Clock clock) {
8282
public void validate(@Nonnull ConnectorArguments arguments) {
8383
Preconditions.checkState(arguments.isAssessment(), "--assessment flag is required");
8484

85-
validateDateRange(arguments);
85+
Connector.validateDateRange(arguments);
8686
}
8787

8888
@Override

dumper/app/src/test/java/com/google/edwmigration/dumper/application/dumper/connector/ConnectorTest.java

Lines changed: 19 additions & 58 deletions
Original file line numberDiff line numberDiff line change
@@ -16,101 +16,62 @@
1616
*/
1717
package com.google.edwmigration.dumper.application.dumper.connector;
1818

19+
import static com.google.edwmigration.dumper.application.dumper.connector.Connector.validateDateRange;
1920
import static org.junit.Assert.assertEquals;
2021
import static org.junit.Assert.assertThrows;
2122

2223
import com.google.edwmigration.dumper.application.dumper.ConnectorArguments;
23-
import com.google.edwmigration.dumper.application.dumper.handle.Handle;
24-
import com.google.edwmigration.dumper.application.dumper.task.Task;
25-
import java.time.Clock;
26-
import java.util.List;
27-
import javax.annotation.Nonnull;
2824
import org.junit.Test;
2925

3026
public class ConnectorTest {
31-
private final Connector connector = new EmptyConnector();
3227

3328
@Test
3429
public void validateDateRange_startDateAndEndDate_success() throws Exception {
35-
String argsStr = "--connector test --start-date=2001-02-20 --end-date=2001-02-25";
30+
ConnectorArguments args =
31+
new ConnectorArguments(
32+
"--connector", "test", "--start-date=2001-02-20", "--end-date=2001-02-25");
3633

3734
// Act
38-
connector.validateDateRange(toArgs(argsStr));
35+
validateDateRange(args);
3936
}
4037

4138
@Test
42-
public void validateDateRange_startDateAfterEndDate_throws() {
43-
String argsStr = "--connector test --start-date=2001-02-20 --end-date=2001-02-20";
39+
public void validateDateRange_startDateAfterEndDate_throws() throws Exception {
40+
ConnectorArguments args =
41+
new ConnectorArguments(
42+
"--connector", "test", "--start-date=2001-02-20", "--end-date=2001-02-20");
4443

4544
Exception exception =
46-
assertThrows(
47-
IllegalStateException.class, () -> connector.validateDateRange(toArgs(argsStr)));
45+
assertThrows(IllegalStateException.class, () -> Connector.validateDateRange(args));
4846
assertEquals(
4947
"Start date [2001-02-20T00:00Z] must be before end date [2001-02-20T00:00Z].",
5048
exception.getMessage());
5149
}
5250

5351
@Test
54-
public void validateDateRange_endDateAlone_throws() {
55-
String argsStr = "--connector test --end-date=2001-02-20";
52+
public void validateDateRange_endDateAlone_throws() throws Exception {
53+
ConnectorArguments args =
54+
new ConnectorArguments("--connector", "test", "--end-date=2001-02-20");
5655

57-
Exception exception =
58-
assertThrows(
59-
IllegalStateException.class, () -> connector.validateDateRange(toArgs(argsStr)));
56+
Exception exception = assertThrows(IllegalStateException.class, () -> validateDateRange(args));
6057
assertEquals(
6158
"End date can be specified only with start date, but start date was null.",
6259
exception.getMessage());
6360
}
6461

6562
@Test
66-
public void validateDateRange_startDateAlone_throws() {
67-
String argsStr = "--connector test --start-date=2001-02-20";
63+
public void validateDateRange_startDateAlone_throws() throws Exception {
64+
ConnectorArguments args =
65+
new ConnectorArguments("--connector", "test", "--start-date=2001-02-20");
6866

69-
Exception exception =
70-
assertThrows(RuntimeException.class, () -> connector.validateDateRange(toArgs(argsStr)));
67+
Exception exception = assertThrows(RuntimeException.class, () -> validateDateRange(args));
7168
assertEquals(
7269
"End date must be specified with start date, but was null.", exception.getMessage());
7370
}
7471

7572
@Test
7673
public void validateDateRange_requiredArgs_success() throws Exception {
7774
// Act
78-
connector.validateDateRange(toArgs("--connector test"));
79-
}
80-
81-
private static class EmptyConnector implements Connector {
82-
83-
@Nonnull
84-
@Override
85-
public String getName() {
86-
return null;
87-
}
88-
89-
@Nonnull
90-
@Override
91-
public String getDefaultFileName(boolean isAssessment, Clock clock) {
92-
return null;
93-
}
94-
95-
@Override
96-
public void addTasksTo(
97-
@Nonnull List<? super Task<?>> out, @Nonnull ConnectorArguments arguments)
98-
throws Exception {}
99-
100-
@Nonnull
101-
@Override
102-
public Handle open(@Nonnull ConnectorArguments arguments) throws Exception {
103-
return null;
104-
}
105-
106-
@Nonnull
107-
@Override
108-
public Iterable<ConnectorProperty> getPropertyConstants() {
109-
return null;
110-
}
111-
}
112-
113-
private static ConnectorArguments toArgs(String args) throws Exception {
114-
return new ConnectorArguments(args.split(" "));
75+
validateDateRange(new ConnectorArguments("--connector", "test"));
11576
}
11677
}

0 commit comments

Comments
 (0)