def test_layer_mute():
map_view = Mock()
map_view.image_sources = [sentinel.source]
source_limits = Mock(spec=["add_source", "remove_source"])
layer = layers.Layer(map_view, sentinel.visible, source_limits)
layer.mute()
source_limits.add_source.assert_called_once_with(sentinel.source)
source_limits.remove_source.assert_called_once_with(sentinel.source)
def test_layer_unmute():
map_view = Mock()
map_view.image_sources = [sentinel.source]
source_limits = Mock(spec=["add_source", "remove_source"])
layer = layers.Layer(map_view, sentinel.visible, source_limits)
layer.unmute()
# One call during __init__ and one during unmute()
calls = [call(sentinel.source), call(sentinel.source)]
source_limits.add_source.assert_has_calls(calls)