def test_organize(self, mock_stat): # make sure all out faked files have content mock_stat.return_value = MagicMock(st_size=1) p_d = push.PillarData(False) organized = p_d.organize('policy.cfg') assert len(organized.keys()) == len(nodes) organized = p_d.organize('policy.cfg_commented1') assert len(organized.keys()) == len(nodes) organized = p_d.organize('policy.cfg_commented2') assert len(organized.keys()) == len(nodes) organized = p_d.organize('policy.cfg_ml_commented') assert len(organized.keys()) == len(nodes) organized = p_d.organize('policy.cfg_leading_whitespace') assert len(organized.keys()) == len(nodes) organized = p_d.organize('policy.cfg_trailing_whitespace') assert len(organized.keys()) == len(nodes) organized = p_d.organize('policy.cfg_trailing_and_leading_whitespace') assert len(organized.keys()) == len(nodes) organized = p_d.organize( 'policy.cfg_trailing_and_leading_whitespace_and_leading_comment') assert len(organized.keys()) == 0 organized = p_d.organize( 'policy.cfg_trailing_and_leading_whitespace_and_trailing_comment') assert len(organized.keys()) == len(nodes)
def test_parse(self): p_d = push.PillarData(False) parsed = p_d._parse('{}/*.sls'.format(proposal_dir)) assert len(parsed) == len(nodes) parsed = p_d._parse('{}/mon*.sls'.format(proposal_dir)) assert len(parsed) == len([n for n in nodes if n.startswith('mon')]) parsed = p_d._parse('{}/mon[1,2].sls'.format(proposal_dir)) assert len(parsed) == 2 parsed = p_d._parse('{}/*.sls slice=[2:5]'.format(proposal_dir)) assert len(parsed) == 3 parsed = p_d._parse('{}/*.sls re=.*1\.sls$'.format(proposal_dir)) assert len(parsed) == len([n for n in nodes if '1' in n]) parsed = p_d._parse('{}/*.sls FOO=.*1\.sls$'.format(proposal_dir)) assert len(parsed) == len(nodes)