FilesExpand file tree

devops_test

/

testIAQE4Y.java

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

63 lines (48 loc) · 2.39 KB

devops_test

/

testIAQE4Y.java

Copy path

Top

File metadata and controls

• Code
• Blame

63 lines (48 loc) · 2.39 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

package dataflow;

import static org.junit.Assert.assertThat;

import java.io.File;

import java.io.IOException;

import java.util.Arrays;

import java.util.List;

import java.util.stream.Collectors;

import common.gspCommon;

import gudusoft.gsqlparser.EDbVendor;

import gudusoft.gsqlparser.dlineage.DataFlowAnalyzer;

import gudusoft.gsqlparser.dlineage.dataflow.model.json.Dataflow;

import gudusoft.gsqlparser.dlineage.dataflow.model.json.Relationship;

import gudusoft.gsqlparser.dlineage.dataflow.model.json.Transform;

import gudusoft.gsqlparser.dlineage.dataflow.model.xml.dataflow;

import junit.framework.TestCase;

public class testIAQE4Y extends TestCase {

public void testOracleDataFlow() throws IOException {

DataFlowAnalyzer analyzer = analyseSqlFiles(

gspCommon.BASE_SQL_DIR_PRIVATE + "sqlscripts/IAQE4Y/sqlserverderivationsimpleupdate.sql");

// When we generate the data flow

analyzer.generateDataFlow();

dataflow df = analyzer.getDataFlow();

Dataflow dataFlow = DataFlowAnalyzer.getSqlflowJSONModel(EDbVendor.dbvmssql, df, false);

List<Relationship> lineTotalRels = Arrays.stream(dataFlow.getRelationships())

.filter(r -> r.getTarget().getColumn().equalsIgnoreCase("[LineTotal]")).collect(Collectors.toList());

Relationship lineTotalInsertRel = lineTotalRels.stream()

.filter(r -> r.getEffectType().equalsIgnoreCase("insert")).findFirst().get();

Relationship lineTotalUpdateRel = lineTotalRels.stream()

.filter(r -> r.getEffectType().equalsIgnoreCase("update")).findFirst().get();

assertNotNull(lineTotalInsertRel);

assertNotNull(lineTotalUpdateRel);

Transform[] insertTransforms = lineTotalInsertRel.getSources()[0].getTransforms();

assertTrue(insertTransforms.length == 1);

assertTrue(insertTransforms[0].getCode().equals("*"));

Transform[] updateTransforms = Arrays.stream(lineTotalUpdateRel.getSources())

.flatMap(s -> Arrays.stream(s.getTransforms())).toArray(Transform[]::new);

assertTrue(updateTransforms.length == 3);

assertTrue(Arrays.stream(updateTransforms)

.allMatch(t -> t.getCode().equalsIgnoreCase("[LineTotal] + [UnitPrice] + [UnitPriceDiscount]")));

}

private DataFlowAnalyzer analyseSqlFiles(String path) throws IOException {

File sqlFile = new File(path);

EDbVendor vendor = EDbVendor.dbvmssql;

DataFlowAnalyzer dataFlowAnalyzer = new DataFlowAnalyzer(sqlFile, vendor, true);

dataFlowAnalyzer.setTransform(true);

return dataFlowAnalyzer;

}

}