150 lines
6.1 KiB
Python
150 lines
6.1 KiB
Python
# tests/test_input_parser.py
|
|
import os
|
|
from agent.input_parser import InputParser
|
|
|
|
FIXTURE_DIR = os.path.join(os.path.dirname(__file__), 'test_data')
|
|
|
|
|
|
def test_parse_design_basic_info():
|
|
parser = InputParser(
|
|
design_md_path=os.path.join(FIXTURE_DIR, '詳細設計書_ZAN04MAT.md'),
|
|
source_cbl_path='dummy.cbl',
|
|
file_db_md_path='dummy.md',
|
|
cpy_dir='dummy_cpy',
|
|
db_md_path='dummy_db.md'
|
|
)
|
|
parser._design_text = parser._read_file(parser.design_md_path)
|
|
|
|
from agent.models import ProgramMeta
|
|
meta = ProgramMeta(program_id='', program_name='', system_name='',
|
|
pgm_type='', pgm_pattern='',
|
|
summary_lines=[], prerequisites=[],
|
|
files=[], keys=[], modules=[],
|
|
process_detail='', output_records='',
|
|
input_type='file', copy_fields={}, db_tables={})
|
|
|
|
parser._parse_basic_info(meta)
|
|
assert meta.program_id == 'ZAN04MAT'
|
|
assert meta.pgm_pattern == 'マッチング(1:1)'
|
|
assert meta.pgm_type == 'メイン'
|
|
assert '残業統計管理システム' in meta.system_name
|
|
|
|
|
|
def test_parse_use_files():
|
|
parser = InputParser(
|
|
design_md_path=os.path.join(FIXTURE_DIR, '詳細設計書_ZAN04MAT.md'),
|
|
source_cbl_path='dummy.cbl', file_db_md_path='dummy.md',
|
|
cpy_dir='dummy_cpy', db_md_path='dummy_db.md'
|
|
)
|
|
parser._design_text = parser._read_file(parser.design_md_path)
|
|
|
|
from agent.models import ProgramMeta
|
|
meta = ProgramMeta(program_id='', program_name='', system_name='',
|
|
pgm_type='', pgm_pattern='',
|
|
summary_lines=[], prerequisites=[],
|
|
files=[], keys=[], modules=[],
|
|
process_detail='', output_records='',
|
|
input_type='file', copy_fields={}, db_tables={})
|
|
|
|
parser._parse_use_files(meta)
|
|
assert len(meta.files) == 5
|
|
r01 = [f for f in meta.files if f.identifier == 'R01'][0]
|
|
assert r01.file_db_name == 'OVT-SORTED'
|
|
assert r01.copy_group == 'ZAN01REC'
|
|
assert r01.medium == 'PS'
|
|
assert r01.io == 'I'
|
|
|
|
|
|
def test_determine_input_type_file():
|
|
parser = InputParser(
|
|
design_md_path=os.path.join(FIXTURE_DIR, '詳細設計書_ZAN04MAT.md'),
|
|
source_cbl_path='dummy.cbl', file_db_md_path='dummy.md',
|
|
cpy_dir='dummy_cpy', db_md_path='dummy_db.md'
|
|
)
|
|
|
|
from agent.models import ProgramMeta, FileInfo
|
|
meta = ProgramMeta(program_id='ZAN04MAT', program_name='', system_name='',
|
|
pgm_type='', pgm_pattern='',
|
|
summary_lines=[], prerequisites=[],
|
|
files=[], keys=[], modules=[],
|
|
process_detail='', output_records='',
|
|
input_type='file', copy_fields={}, db_tables={})
|
|
meta.files = [
|
|
FileInfo(no=1, file_db_name='F1', identifier='R01', dd_name='DD1',
|
|
io='I', copy_group='C1', format='FB', record_len=80,
|
|
medium='PS', remarks=''),
|
|
FileInfo(no=2, file_db_name='F2', identifier='W01', dd_name='DD2',
|
|
io='O', copy_group='C2', format='FB', record_len=80,
|
|
medium='PS', remarks=''),
|
|
]
|
|
parser._determine_input_type(meta)
|
|
assert meta.input_type == 'file'
|
|
|
|
|
|
def test_determine_input_type_mixed():
|
|
parser = InputParser(
|
|
design_md_path=os.path.join(FIXTURE_DIR, '詳細設計書_ZAN04MAT.md'),
|
|
source_cbl_path='dummy.cbl', file_db_md_path='dummy.md',
|
|
cpy_dir='dummy_cpy', db_md_path='dummy_db.md'
|
|
)
|
|
|
|
from agent.models import ProgramMeta, FileInfo
|
|
meta = ProgramMeta(program_id='TEST', program_name='', system_name='',
|
|
pgm_type='', pgm_pattern='',
|
|
summary_lines=[], prerequisites=[],
|
|
files=[], keys=[], modules=[],
|
|
process_detail='', output_records='',
|
|
input_type='file', copy_fields={}, db_tables={})
|
|
meta.files = [
|
|
FileInfo(no=1, file_db_name='F1', identifier='R01', dd_name='DD1',
|
|
io='I', copy_group='C1', format='FB', record_len=80,
|
|
medium='PS', remarks=''),
|
|
FileInfo(no=2, file_db_name='DB1', identifier='DB', dd_name='',
|
|
io='I', copy_group='', format='', record_len=0,
|
|
medium='DB', remarks=''),
|
|
]
|
|
parser._determine_input_type(meta)
|
|
assert meta.input_type == 'mixed'
|
|
|
|
|
|
def test_parse_process_detail():
|
|
parser = InputParser(
|
|
design_md_path=os.path.join(FIXTURE_DIR, '詳細設計書_ZAN04MAT.md'),
|
|
source_cbl_path='dummy.cbl', file_db_md_path='dummy.md',
|
|
cpy_dir='dummy_cpy', db_md_path='dummy_db.md'
|
|
)
|
|
parser._design_text = parser._read_file(parser.design_md_path)
|
|
|
|
from agent.models import ProgramMeta
|
|
meta = ProgramMeta(program_id='', program_name='', system_name='',
|
|
pgm_type='', pgm_pattern='',
|
|
summary_lines=[], prerequisites=[],
|
|
files=[], keys=[], modules=[],
|
|
process_detail='', output_records='',
|
|
input_type='file', copy_fields={}, db_tables={})
|
|
|
|
parser._parse_process_detail(meta)
|
|
assert '1000ITTSOR' in meta.process_detail
|
|
assert 'マッチの場合' in meta.process_detail
|
|
|
|
|
|
def test_parse_output_records():
|
|
parser = InputParser(
|
|
design_md_path=os.path.join(FIXTURE_DIR, '詳細設計書_ZAN04MAT.md'),
|
|
source_cbl_path='dummy.cbl', file_db_md_path='dummy.md',
|
|
cpy_dir='dummy_cpy', db_md_path='dummy_db.md'
|
|
)
|
|
parser._design_text = parser._read_file(parser.design_md_path)
|
|
|
|
from agent.models import ProgramMeta
|
|
meta = ProgramMeta(program_id='', program_name='', system_name='',
|
|
pgm_type='', pgm_pattern='',
|
|
summary_lines=[], prerequisites=[],
|
|
files=[], keys=[], modules=[],
|
|
process_detail='', output_records='',
|
|
input_type='file', copy_fields={}, db_tables={})
|
|
|
|
parser._parse_output_records(meta)
|
|
assert 'OVT-MATCHED' in meta.output_records
|
|
assert 'OVT-DBCLEAN' in meta.output_records
|