def remove(cls, alias):
"""
Remove source by alias.
Required arguments:
alias -- alias of source to remove
"""
try:
source = cls._sources[alias]
except KeyError:
raise UnknownSourceError(alias)
else:
EosSourceManager.remove(alias)
source.edb.close()
del cls._sources[alias]
def test_removing_known_source(mock_data_handler, mock_cache_handler):
SourceManager.add('test', mock_data_handler, mock_cache_handler)
SourceManager.remove('test')
assert 'test' not in SourceManager._sources
def test_removing_unknown_source(mock_data_handler, mock_cache_handler):
with pytest.raises(UnknownSourceError):
SourceManager.remove('test')
