import mock
import sys

class MockConfigMgr(object):
  def __setattr__(self, k, v):
    return None

mocked_config = mock.MagicMock(return_value=False)
mocked_config.config = ['config/mbj_79800.json']
mocked_config.analysisName_base = ''
mocked_config.luminosity = 0
mocked_config.regionOutput = ''
mocked_config.noSyst = True
mocked_config.doFlatUnc30 = False
mocked_config.doFlatUnc50 = False
mocked_config.doAllSyst = False
mocked_config.doTheoSyst = False
mocked_config.cutsJson = 'metadata/cuts_mbj.json'
mocked_byteify = mock.MagicMock(side_effect=lambda x: x)

mocked_cfgMgr = mock.MagicMock()
mocked_cfgMgr.ConfigManager = MockConfigMgr()
@mock.patch.dict('sys.modules',
    ROOT=mock.MagicMock(),
    configManager=mocked_cfgMgr,
    configWriter=mock.MagicMock(),
    systematic=mock.MagicMock(),)
@mock.patch('argparse.ArgumentParser.add_argument')
@mock.patch('argparse.ArgumentParser.parse_args')
@mock.patch('argparse.ArgumentParser.parse_known_args', return_value=[mocked_config,None])
@mock.patch('ROOT.TColor.GetColor')
def test(*args):
  filename = 'python/3b.py'
  vars_global = globals()
  #vars_local = locals()
  vars_global['byteify'] = mocked_byteify
  vars_global['configMgr'] = mock.MagicMock()
  vars_global['ROOT'] = mock.MagicMock()
  exec(compile(open(filename, "rb").read(), filename, 'exec'), vars_global, {})
  return mocked_cfgMgr.configMgr.cutsDict

cuts = test()
print cuts.keys()