def test_shouldGenerateDefaultValue_whenSchedulePeriodIsInvalid(self):
valid_values = [
member
for member in DltaskScheduler.SchedulePeriods.__members__.values()
]
invalid_values = [-1, 4, -9999, 9999, 'some_string']
default_value = DltaskScheduler.SchedulePeriods.HOURLY
for valid_value in valid_values:
self.create_test_settings_file({
constants.SCHEDULE_SECTION: {
constants.SCHEDULE_PERIOD: valid_value.value
}
})
settings_read = Settings(TEST_SETTINGS_FILE).get()
assert settings_read[constants.SCHEDULE_SECTION][
constants.SCHEDULE_PERIOD] == valid_value
for invalid_value in invalid_values:
self.create_test_settings_file({
constants.SCHEDULE_SECTION: {
constants.SCHEDULE_PERIOD: invalid_value
}
})
settings_read = Settings(TEST_SETTINGS_FILE).get()
assert settings_read[constants.SCHEDULE_SECTION][
constants.SCHEDULE_PERIOD] == default_value
def test_shouldGenerateDefaultValue_whenStartMinuteIsInvalid(self):
valid_values = [0, 12, 35, 59]
invalid_values = [-1, 60, -9999, 9999, 'some_string']
default_value = 0
for valid_value in valid_values:
self.create_test_settings_file({
constants.SCHEDULE_SECTION: {
constants.START_MINUTE: valid_value
}
})
settings_read = Settings(TEST_SETTINGS_FILE).get()
assert settings_read[constants.SCHEDULE_SECTION][
constants.START_MINUTE] == valid_value
for invalid_value in invalid_values:
self.create_test_settings_file({
constants.SCHEDULE_SECTION: {
constants.START_MINUTE: invalid_value
}
})
settings_read = Settings(TEST_SETTINGS_FILE).get()
assert settings_read[constants.SCHEDULE_SECTION][
constants.START_MINUTE] == default_value
def test_shouldGenerateDefaultValue_whenIsScheduleIsInvalid(self):
valid_values = [0, 1]
invalid_values = [-1, 2, -9999, 9999, 'some_string']
default_value = False
for valid_value in valid_values:
self.create_test_settings_file({
constants.SCHEDULE_SECTION: {
constants.IS_SCHEDULED: valid_value
}
})
settings_read = Settings(TEST_SETTINGS_FILE).get()
assert settings_read[constants.SCHEDULE_SECTION][
constants.IS_SCHEDULED] == valid_value
for invalid_value in invalid_values:
self.create_test_settings_file({
constants.SCHEDULE_SECTION: {
constants.IS_SCHEDULED: invalid_value
}
})
settings_read = Settings(TEST_SETTINGS_FILE).get()
assert settings_read[constants.SCHEDULE_SECTION][
constants.IS_SCHEDULED] == default_value
def test_setShouldReplaceSettingState(self):
new_settings = {
constants.GENERAL_SECTION: {
constants.SAVE_LOCATION: Path('new'),
},
constants.API_SECTION: {
constants.ACCESS_TOKEN: 'new',
constants.ACCESS_SECRET: 'new',
constants.CONSUMER_KEY: 'new',
constants.CONSUMER_SECRET: 'new',
},
constants.SCHEDULE_SECTION: {
constants.IS_SCHEDULED: True,
constants.SCHEDULE_PERIOD:
DltaskScheduler.SchedulePeriods.HOURLY,
constants.START_HOUR: 1,
constants.START_MINUTE: 1,
},
}
app_settings = Settings(SAMPLE_SETTINGS_FILE)
app_settings.set(new_settings)
settings = app_settings.get()
for section_name, section in new_settings.items():
for option_name, option in section.items():
assert settings[section_name][option_name] == option
def __init__(self, app_path):
logging.info('Initializing objects')
self._settings = Settings(app_path / constants.FILENAME_SETTINGS)
self._history = DownloadHistory()
self._retriever = TweetsRetriever_TwitterAPI(self._history,
self._settings)
self._scheduler = DltaskScheduler(app_path)
self._abort = AbortFlag()
logging.info('Finished initializing objects')
def test_writeShouldWriteValueToNonExistantSettingsFile(self):
assert TEST_SETTINGS_FILE.exists() == False
general_settings = Settings(TEST_SETTINGS_FILE)
sample_settings = Settings(SAMPLE_SETTINGS_FILE).get()
general_settings.set(sample_settings)
general_settings.write()
assert TEST_SETTINGS_FILE.exists() == True
settings = Settings(TEST_SETTINGS_FILE).get()
for section_name, section in sample_settings.items():
for option_name, option in section.items():
assert settings[section_name][option_name] == option
def test_writeShouldWriteValueToSettingsFile(self):
shutil.copyfile(SAMPLE_SETTINGS_FILE, TEST_SETTINGS_FILE)
general_settings = Settings(TEST_SETTINGS_FILE)
new_settings = {
constants.GENERAL_SECTION: {
constants.SAVE_LOCATION: Path('new'),
},
constants.API_SECTION: {
constants.ACCESS_TOKEN: 'new',
constants.ACCESS_SECRET: 'new',
constants.CONSUMER_KEY: 'new',
constants.CONSUMER_SECRET: 'new',
},
constants.SCHEDULE_SECTION: {
constants.IS_SCHEDULED: True,
constants.SCHEDULE_PERIOD:
DltaskScheduler.SchedulePeriods.HOURLY,
constants.START_HOUR: 1,
constants.START_MINUTE: 1,
},
}
general_settings.set(new_settings)
general_settings.write()
settings = Settings(TEST_SETTINGS_FILE).get()
for section_name, section in new_settings.items():
for option_name, option in section.items():
assert settings[section_name][option_name] == option
def test_shouldGenerateFilePathDirAsDefaultValue_ForNonExistingGeneralSection(
self):
no_general_section = TEST_DATA_DIR / 'no_general_settings.txt'
file_location = no_general_section.parents[0]
settings = Settings(no_general_section).get()
assert settings[constants.GENERAL_SECTION][
constants.SAVE_LOCATION] == file_location
def create_test_settings_file(self, new_options):
shutil.copyfile(SAMPLE_SETTINGS_FILE, TEST_SETTINGS_FILE)
settings = Settings(TEST_SETTINGS_FILE).get()
for section, options in new_options.items():
settings[section].update(options)
parser = ConfigParser()
parser.read_dict(settings)
with TEST_SETTINGS_FILE.open('w', encoding='utf-8') as f:
parser.write(f, settings)
def test_shouldInsertEmptyStringeAsDefaultValue_ForNonExistingSection(
self):
no_api_section = TEST_DATA_DIR / 'no_twitter_api.txt'
settings = Settings(no_api_section).get()
assert settings[constants.API_SECTION][constants.ACCESS_TOKEN] == ''
assert settings[constants.API_SECTION][constants.ACCESS_SECRET] == ''
assert settings[constants.API_SECTION][constants.CONSUMER_KEY] == ''
assert settings[constants.API_SECTION][constants.CONSUMER_SECRET] == ''
def test_shouldGenerateDefaultValue_ForNonExistingScheduleSection(self):
no_schedule_section = TEST_DATA_DIR / 'no_schedule_settings.txt'
file_location = no_schedule_section.parents[0]
settings = Settings(no_schedule_section).get()
assert settings[constants.SCHEDULE_SECTION][
constants.IS_SCHEDULED] == False
assert settings[constants.SCHEDULE_SECTION][
constants.
SCHEDULE_PERIOD] == DltaskScheduler.SchedulePeriods.HOURLY
assert settings[constants.SCHEDULE_SECTION][constants.START_HOUR] == 0
assert settings[constants.SCHEDULE_SECTION][
constants.START_MINUTE] == 0
def test_shouldGenerateDefaultValues_whenFileNonExistant(self):
nonexistant_file = TEST_DATA_DIR / '123120381028309128309123.txt'
file_location = nonexistant_file.parents[0]
settings = Settings(nonexistant_file).get()
assert settings[constants.GENERAL_SECTION][
constants.SAVE_LOCATION] == file_location
assert settings[constants.API_SECTION][constants.ACCESS_TOKEN] == ''
assert settings[constants.API_SECTION][constants.ACCESS_SECRET] == ''
assert settings[constants.API_SECTION][constants.CONSUMER_KEY] == ''
assert settings[constants.API_SECTION][constants.CONSUMER_SECRET] == ''
assert settings[constants.SCHEDULE_SECTION][
constants.IS_SCHEDULED] == False
assert settings[constants.SCHEDULE_SECTION][
constants.
SCHEDULE_PERIOD] == DltaskScheduler.SchedulePeriods.HOURLY
assert settings[constants.SCHEDULE_SECTION][constants.START_HOUR] == 0
assert settings[constants.SCHEDULE_SECTION][
constants.START_MINUTE] == 0
def test_shouldGenerateEmptyString_whenOptionsAreEmpty(self):
empty_file = TEST_DATA_DIR / 'empty.txt'
file_location = empty_file.parents[0]
settings = Settings(empty_file).get()
assert settings[constants.GENERAL_SECTION][
constants.SAVE_LOCATION] == file_location
assert settings[constants.API_SECTION][constants.ACCESS_TOKEN] == ''
assert settings[constants.API_SECTION][constants.ACCESS_SECRET] == ''
assert settings[constants.API_SECTION][constants.CONSUMER_KEY] == ''
assert settings[constants.API_SECTION][constants.CONSUMER_SECRET] == ''
assert settings[constants.SCHEDULE_SECTION][
constants.IS_SCHEDULED] == False
assert settings[constants.SCHEDULE_SECTION][
constants.
SCHEDULE_PERIOD] == DltaskScheduler.SchedulePeriods.HOURLY
assert settings[constants.SCHEDULE_SECTION][constants.START_HOUR] == 0
assert settings[constants.SCHEDULE_SECTION][
constants.START_MINUTE] == 0
def test_shouldReadValueFromGoodSettingFile(self):
settings = Settings(SAMPLE_SETTINGS_FILE).get()
assert settings[constants.GENERAL_SECTION]['save_location'] == Path(
r'C:\Users\user1\dir1\dir\dir3')
assert settings[constants.API_SECTION][
constants.ACCESS_TOKEN] == 'test_access_token'
assert settings[constants.API_SECTION][
constants.ACCESS_SECRET] == 'test_access_secret'
assert settings[constants.API_SECTION][
constants.CONSUMER_KEY] == 'test_consumer_key'
assert settings[constants.API_SECTION][
constants.CONSUMER_SECRET] == 'test_consumer_secret'
assert settings[constants.SCHEDULE_SECTION][
constants.IS_SCHEDULED] == True
assert settings[constants.SCHEDULE_SECTION][
constants.SCHEDULE_PERIOD] == DltaskScheduler.SchedulePeriods.DAILY
assert settings[constants.SCHEDULE_SECTION][constants.START_HOUR] == 1
assert settings[constants.SCHEDULE_SECTION][
constants.START_MINUTE] == 1
class RuntimeBindings:
def __init__(self, app_path):
logging.info('Initializing objects')
self._settings = Settings(app_path / constants.FILENAME_SETTINGS)
self._history = DownloadHistory()
self._retriever = TweetsRetriever_TwitterAPI(self._history,
self._settings)
self._scheduler = DltaskScheduler(app_path)
self._abort = AbortFlag()
logging.info('Finished initializing objects')
def get_settings(self):
return self._settings
def get_save_location(self):
return Path(self._settings.get()[constants.GENERAL_SECTION][
constants.SAVE_LOCATION])
def get_users(self):
return readUserList(self.get_save_location() /
constants.FILENAME_USERS)
def get_history(self):
return self._history
def get_tweet_retriever(self):
return self._retriever
def get_scheduler(self):
return self._scheduler
def get_abort(self):
return self._abort
def download_media(self, url, savepath):
downloadMedia(url, savepath)
def test_shouldStripWhitespace_AfterParsing(self):
space_settings_file = TEST_DATA_DIR / 'settings_spaces.txt'
settings = Settings(space_settings_file).get()
assert settings[constants.GENERAL_SECTION]['save_location'] == Path(
r'C:\Users\user1\dir1\dir\dir3')
assert settings[constants.API_SECTION][
constants.ACCESS_TOKEN] == 'test_access_token'
assert settings[constants.API_SECTION][
constants.ACCESS_SECRET] == 'test_access_secret'
assert settings[constants.API_SECTION][
constants.CONSUMER_KEY] == 'test_consumer_key'
assert settings[constants.API_SECTION][
constants.CONSUMER_SECRET] == 'test_consumer_secret'
assert settings[constants.SCHEDULE_SECTION][
constants.IS_SCHEDULED] == True
assert settings[constants.SCHEDULE_SECTION][
constants.SCHEDULE_PERIOD] == DltaskScheduler.SchedulePeriods.DAILY
assert settings[constants.SCHEDULE_SECTION][constants.START_HOUR] == 1
assert settings[constants.SCHEDULE_SECTION][
constants.START_MINUTE] == 1
def test_raiseErrorWhenSaveLocationDoesNotExist(self):
general_settings = Settings(SAMPLE_SETTINGS_FILE)
with pytest.raises(exceptions.SaveLocationNotExist) as e:
general_settings.validate_settings()
def test_raiseErrorWhenNoAPIOptionsAreFoundInSettings(self):
empty_file = TEST_DATA_DIR / 'empty.txt'
general_settings = Settings(empty_file)
with pytest.raises(exceptions.APINotFound) as e:
general_settings.validate_settings()
