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')