def test_main(self): t = timedelta(days = 5) d = datetime.today() - t dp = datetime.strftime(d,'%Y-%m-%d') arg = ['dummy',1,'epg',dp,'test'] param = params.setstatus_params(arg) actual = setstatus.main(param)
def test_setstatus_params_noexit(self): argv = [0,1,'epg','2015-08-15',4] actual = params.setstatus_params(argv) assert actual.data_type == 'epg'
def test_setstatus_params_exit3(self): argv = [0,1,'epg','20150815',4] params.setstatus_params(argv)
def test_setstatus_params_exit2(self): argv = [0,1,'epga',3,4] params.setstatus_params(argv)
def test_setstatus_params_exit1(self): argv = [0,1,2,3] params.setstatus_params(argv)
import os import sys from tools import aws from tools import log from tools import util from tools import params from tools import ctrl ERROR_STATUS = "{0}_failed" def main(param): try: c = ctrl.data_ctrl_mst(param.maker_id, param.data_type, param.data_date.replace("-", ""), param.status) except Exception as e: log.except_log(logger, e) c.status = ERROR_STATUS.format(c.status) finally: # log to RDS with aws.rdsconn(config) as rdsconn: aws.commit_ctrl_update(rdsconn, c) if __name__ == "__main__": logger = log.init_logger(False, os.path.basename(__file__)) confpath = os.path.join(os.path.dirname(os.path.realpath(__file__)), "conf", "app.config") config = util.get_config(confpath) param = params.setstatus_params(sys.argv) main(param)