1 

2import os 

3import sys 

4import pytest 

5sys.path.append(os.path.dirname(os.path.abspath(__file__)) + "/../../../../") 

6from sdc_etl_libs.sdc_data_schema.schema_toolbox import SchemaToolbox 

7 

8 

9def test_determine_dataframe_from_schema_happy_path(): 

10 """ 

11 Ensure get_field_names_for_file returns all field names 

12 """ 

13 

14 data_schema = { 

15 "fields": [ 

16 {"name": "COL1", "type": {"type": "string"}}, 

17 {"name": "COL2", "type": {"type": "string"}}, 

18 {"name": "COL3", "type": {"type": "string"}}, 

19 ] 

20 } 

21 

22 result = SchemaToolbox.get_field_names_for_file(data_schema) 

23 assert result == ["COL1", "COL2", "COL3"] 

24 

25 

26def test_determine_dataframe_from_schema_added_field_removed(): 

27 """ 

28 Ensure a field with "add_column" == True is removed from the list. 

29 """ 

30 

31 data_schema = { 

32 "fields": [ 

33 {"name": "COL1", "type": {"type": "string"}}, 

34 {"name": "COL2", "type": {"type": "string"}, "add_column": True}, 

35 {"name": "COL3", "type": {"type": "string"}}, 

36 ] 

37 } 

38 

39 result = SchemaToolbox.get_field_names_for_file(data_schema) 

40 assert result == ["COL1", "COL3"]