File diff 48f1825d7e4e → ea0e3d3a73b7
tests/date_hooks.py
Show inline comments
 
new file 100644
 
import datetime
 

	
 
class DateHookTestBase:
 
    DATE_FMT_ISO = '%Y-%m-%d'
 
    DATE_FMT_LEDGER = '%d/%m/%Y'
 

	
 
    def _format_config_date(self, section, key, value, date_fmt):
 
        if value is not None:
 
            section[key] = value.strftime(date_fmt)
 

	
 
    def new_config(
 
            self,
 
            fmt=DATE_FMT_ISO,
 
            default_date=datetime.date(2017, 3, 1),
 
            start_date=datetime.date(2017, 2, 1),
 
            end_date=datetime.date(2017, 4, 1)
 
    ):
 
        section = {'date format': fmt.replace('%', '%%')}
 
        self._format_config_date(section, 'default date', default_date, fmt)
 
        self._format_config_date(section, 'import start date', start_date, fmt)
 
        self._format_config_date(section, 'import end date', end_date, fmt)
 
        return {'Dates': section}