Skip to content

Commit e2544e0

Browse files
ctmay4ctmay4
committed
Starting to improve code coverage
1 parent 9f879bb commit e2544e0

1 file changed

Lines changed: 105 additions & 62 deletions

File tree

‎src/test/java/com/imsweb/seerapi/client/staging/StagingTest.java‎

Lines changed: 105 additions & 62 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,50 @@
1515
import com.imsweb.seerapi.client.glossary.Glossary.Category;
1616
import com.imsweb.seerapi.client.shared.KeywordMatch;
1717
import com.imsweb.seerapi.client.staging.cs.CsSchemaLookup;
18-
18+
import com.imsweb.seerapi.client.staging.cs.CsStagingData;
19+
import com.imsweb.seerapi.client.staging.cs.CsStagingData.CsInput;
20+
import com.imsweb.seerapi.client.staging.cs.CsStagingData.CsStagingInputBuilder;
21+
22+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.AJCC6_M;
23+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.AJCC6_MDESCRIPTOR;
24+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.AJCC6_N;
25+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.AJCC6_NDESCRIPTOR;
26+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.AJCC6_STAGE;
27+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.AJCC6_T;
28+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.AJCC6_TDESCRIPTOR;
29+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.AJCC7_M;
30+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.AJCC7_MDESCRIPTOR;
31+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.AJCC7_N;
32+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.AJCC7_NDESCRIPTOR;
33+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.AJCC7_STAGE;
34+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.AJCC7_T;
35+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.AJCC7_TDESCRIPTOR;
36+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.CSVER_DERIVED;
37+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.SCHEMA_NUMBER;
38+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.SS1977_M;
39+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.SS1977_N;
40+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.SS1977_STAGE;
41+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.SS1977_T;
42+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.SS2000_M;
43+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.SS2000_N;
44+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.SS2000_STAGE;
45+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.SS2000_T;
46+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.STOR_AJCC6_M;
47+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.STOR_AJCC6_MDESCRIPTOR;
48+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.STOR_AJCC6_N;
49+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.STOR_AJCC6_NDESCRIPTOR;
50+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.STOR_AJCC6_STAGE;
51+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.STOR_AJCC6_T;
52+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.STOR_AJCC6_TDESCRIPTOR;
53+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.STOR_AJCC7_M;
54+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.STOR_AJCC7_MDESCRIPTOR;
55+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.STOR_AJCC7_N;
56+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.STOR_AJCC7_NDESCRIPTOR;
57+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.STOR_AJCC7_STAGE;
58+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.STOR_AJCC7_T;
59+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.STOR_AJCC7_TDESCRIPTOR;
60+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.STOR_SS1977_STAGE;
61+
import static com.imsweb.seerapi.client.staging.cs.CsStagingData.CsOutput.STOR_SS2000_STAGE;
1962
import static org.assertj.core.api.Assertions.assertThat;
2063

2164
public class StagingTest {
@@ -143,27 +186,27 @@ public void testTableInvoledSchemas() throws IOException {
143186
}
144187

145188
@Test
146-
public void testStaging() throws IOException {
147-
// test this case: http://seer.cancer.gov/seertools/cstest/?mets=10&lnexam=99&diagnosis_year=2013&grade=9&exteval=9&age=060&site=C680&metseval=9&hist=8000&ext=100&version=020550&nodeseval=9&behav=3&lnpos=99&nodes=100&csver_original=020440&lvi=9&ssf1=020&size=075
148-
StagingData data = new StagingData();
149-
data.setInput("site", "C680");
150-
data.setInput("hist", "8000");
151-
data.setInput("behavior", "3");
152-
data.setInput("grade", "9");
153-
data.setInput("year_dx", "2013");
154-
data.setInput("cs_input_version_original", "020550");
155-
data.setInput("size", "075");
156-
data.setInput("extension", "100");
157-
data.setInput("extension_eval", "9");
158-
data.setInput("nodes", "100");
159-
data.setInput("nodes_eval", "9");
160-
data.setInput("nodes_pos", "99");
161-
data.setInput("nodes_exam", "99");
162-
data.setInput("mets", "10");
163-
data.setInput("mets_eval", "9");
164-
data.setInput("lvi", "9");
165-
data.setInput("age_dx", "060");
166-
data.setInput("ssf1", "020");
189+
public void testCsStaging() throws IOException {
190+
CsStagingData data = new CsStagingInputBuilder()
191+
.withInput(CsInput.PRIMARY_SITE, "C680")
192+
.withInput(CsInput.HISTOLOGY, "8000")
193+
.withInput(CsInput.BEHAVIOR, "3")
194+
.withInput(CsInput.GRADE, "9")
195+
.withInput(CsInput.DX_YEAR, "2013")
196+
.withInput(CsInput.CS_VERSION_ORIGINAL, "020550")
197+
.withInput(CsInput.TUMOR_SIZE, "075")
198+
.withInput(CsInput.EXTENSION, "100")
199+
.withInput(CsInput.EXTENSION_EVAL, "9")
200+
.withInput(CsInput.LYMPH_NODES, "100")
201+
.withInput(CsInput.LYMPH_NODES_EVAL, "9")
202+
.withInput(CsInput.REGIONAL_NODES_POSITIVE, "99")
203+
.withInput(CsInput.REGIONAL_NODES_EXAMINED, "99")
204+
.withInput(CsInput.METS_AT_DX, "10")
205+
.withInput(CsInput.METS_EVAL, "9")
206+
.withInput(CsInput.LVI, "9")
207+
.withInput(CsInput.AGE_AT_DX, "060")
208+
.withSsf(1, "020")
209+
.build();
167210

168211
// perform the staging
169212
StagingData output = _STAGING.stage(_ALGORITHM, _VERSION, data.getInput()).execute().body();
@@ -174,53 +217,53 @@ public void testStaging() throws IOException {
174217
assertThat(output.getPath()).hasSize(37);
175218

176219
// check output
177-
assertThat(output.getOutput("schema_number")).isEqualTo("129");
220+
assertThat(output.getOutput(SCHEMA_NUMBER.toString())).isEqualTo("129");
178221
assertThat(output.getSchemaId()).isEqualTo("urethra");
179-
assertThat(output.getOutput("csver_derived")).isEqualTo("020550");
222+
assertThat(output.getOutput(CSVER_DERIVED.toString())).isEqualTo("020550");
180223

181224
// AJCC 6
182-
assertThat(output.getOutput("ajcc6_t")).isEqualTo("T1");
183-
assertThat(output.getOutput("stor_ajcc6_t")).isEqualTo("10");
184-
assertThat(output.getOutput("ajcc6_tdescriptor")).isEqualTo("c");
185-
assertThat(output.getOutput("stor_ajcc6_tdescriptor")).isEqualTo("c");
186-
assertThat(output.getOutput("ajcc6_n")).isEqualTo("N1");
187-
assertThat(output.getOutput("stor_ajcc6_n")).isEqualTo("10");
188-
assertThat(output.getOutput("ajcc6_ndescriptor")).isEqualTo("c");
189-
assertThat(output.getOutput("stor_ajcc6_ndescriptor")).isEqualTo("c");
190-
assertThat(output.getOutput("ajcc6_m")).isEqualTo("M1");
191-
assertThat(output.getOutput("stor_ajcc6_m")).isEqualTo("10");
192-
assertThat(output.getOutput("ajcc6_mdescriptor")).isEqualTo("c");
193-
assertThat(output.getOutput("stor_ajcc6_mdescriptor")).isEqualTo("c");
194-
assertThat(output.getOutput("ajcc6_stage")).isEqualTo("IV");
195-
assertThat(output.getOutput("stor_ajcc6_stage")).isEqualTo("70");
225+
assertThat(output.getOutput(AJCC6_T.toString())).isEqualTo("T1");
226+
assertThat(output.getOutput(STOR_AJCC6_T.toString())).isEqualTo("10");
227+
assertThat(output.getOutput(AJCC6_TDESCRIPTOR.toString())).isEqualTo("c");
228+
assertThat(output.getOutput(STOR_AJCC6_TDESCRIPTOR.toString())).isEqualTo("c");
229+
assertThat(output.getOutput(AJCC6_N.toString())).isEqualTo("N1");
230+
assertThat(output.getOutput(STOR_AJCC6_N.toString())).isEqualTo("10");
231+
assertThat(output.getOutput(AJCC6_NDESCRIPTOR.toString())).isEqualTo("c");
232+
assertThat(output.getOutput(STOR_AJCC6_NDESCRIPTOR.toString())).isEqualTo("c");
233+
assertThat(output.getOutput(AJCC6_M.toString())).isEqualTo("M1");
234+
assertThat(output.getOutput(STOR_AJCC6_M.toString())).isEqualTo("10");
235+
assertThat(output.getOutput(AJCC6_MDESCRIPTOR.toString())).isEqualTo("c");
236+
assertThat(output.getOutput(STOR_AJCC6_MDESCRIPTOR.toString())).isEqualTo("c");
237+
assertThat(output.getOutput(AJCC6_STAGE.toString())).isEqualTo("IV");
238+
assertThat(output.getOutput(STOR_AJCC6_STAGE.toString())).isEqualTo("70");
196239

197240
// AJCC 7
198-
assertThat(output.getOutput("ajcc7_t")).isEqualTo("T1");
199-
assertThat(output.getOutput("stor_ajcc7_t")).isEqualTo("100");
200-
assertThat(output.getOutput("ajcc7_tdescriptor")).isEqualTo("c");
201-
assertThat(output.getOutput("stor_ajcc7_tdescriptor")).isEqualTo("c");
202-
assertThat(output.getOutput("ajcc7_n")).isEqualTo("N1");
203-
assertThat(output.getOutput("stor_ajcc7_n")).isEqualTo("100");
204-
assertThat(output.getOutput("ajcc7_ndescriptor")).isEqualTo("c");
205-
assertThat(output.getOutput("stor_ajcc7_ndescriptor")).isEqualTo("c");
206-
assertThat(output.getOutput("ajcc7_m")).isEqualTo("M1");
207-
assertThat(output.getOutput("stor_ajcc7_m")).isEqualTo("100");
208-
assertThat(output.getOutput("ajcc7_mdescriptor")).isEqualTo("c");
209-
assertThat(output.getOutput("stor_ajcc7_mdescriptor")).isEqualTo("c");
210-
assertThat(output.getOutput("ajcc7_stage")).isEqualTo("IV");
211-
assertThat(output.getOutput("stor_ajcc7_stage")).isEqualTo("700");
241+
assertThat(output.getOutput(AJCC7_T.toString())).isEqualTo("T1");
242+
assertThat(output.getOutput(STOR_AJCC7_T.toString())).isEqualTo("100");
243+
assertThat(output.getOutput(AJCC7_TDESCRIPTOR.toString())).isEqualTo("c");
244+
assertThat(output.getOutput(STOR_AJCC7_TDESCRIPTOR.toString())).isEqualTo("c");
245+
assertThat(output.getOutput(AJCC7_N.toString())).isEqualTo("N1");
246+
assertThat(output.getOutput(STOR_AJCC7_N.toString())).isEqualTo("100");
247+
assertThat(output.getOutput(AJCC7_NDESCRIPTOR.toString())).isEqualTo("c");
248+
assertThat(output.getOutput(STOR_AJCC7_NDESCRIPTOR.toString())).isEqualTo("c");
249+
assertThat(output.getOutput(AJCC7_M.toString())).isEqualTo("M1");
250+
assertThat(output.getOutput(STOR_AJCC7_M.toString())).isEqualTo("100");
251+
assertThat(output.getOutput(AJCC7_MDESCRIPTOR.toString())).isEqualTo("c");
252+
assertThat(output.getOutput(STOR_AJCC7_MDESCRIPTOR.toString())).isEqualTo("c");
253+
assertThat(output.getOutput(AJCC7_STAGE.toString())).isEqualTo("IV");
254+
assertThat(output.getOutput(STOR_AJCC7_STAGE.toString())).isEqualTo("700");
212255

213256
// Summary Stage
214-
assertThat(output.getOutput("t77")).isEqualTo("L");
215-
assertThat(output.getOutput("n77")).isEqualTo("RN");
216-
assertThat(output.getOutput("m77")).isEqualTo("D");
217-
assertThat(output.getOutput("ss77")).isEqualTo("D");
218-
assertThat(output.getOutput("stor_ss77")).isEqualTo("7");
219-
assertThat(output.getOutput("t2000")).isEqualTo("L");
220-
assertThat(output.getOutput("n2000")).isEqualTo("RN");
221-
assertThat(output.getOutput("m2000")).isEqualTo("D");
222-
assertThat(output.getOutput("ss2000")).isEqualTo("D");
223-
assertThat(output.getOutput("stor_ss2000")).isEqualTo("7");
257+
assertThat(output.getOutput(SS1977_T.toString())).isEqualTo("L");
258+
assertThat(output.getOutput(SS1977_N.toString())).isEqualTo("RN");
259+
assertThat(output.getOutput(SS1977_M.toString())).isEqualTo("D");
260+
assertThat(output.getOutput(SS1977_STAGE.toString())).isEqualTo("D");
261+
assertThat(output.getOutput(STOR_SS1977_STAGE.toString())).isEqualTo("7");
262+
assertThat(output.getOutput(SS2000_T.toString())).isEqualTo("L");
263+
assertThat(output.getOutput(SS2000_N.toString())).isEqualTo("RN");
264+
assertThat(output.getOutput(SS2000_M.toString())).isEqualTo("D");
265+
assertThat(output.getOutput(SS2000_STAGE.toString())).isEqualTo("D");
266+
assertThat(output.getOutput(STOR_SS2000_STAGE.toString())).isEqualTo("7");
224267
}
225268

226269
@Test

0 commit comments

Comments
 (0)