def test_can_add_2_sources(self): a = Controller() s1 = mock_source('Source 1') s2 = mock_source('Source 2') a.add_source(s1) a.add_source(s2) assert a.get_source_names() == ('Source 1', 'Source 2')
def test_can_add_source(self): a = Controller() s = mock_source('Mock Source') a.add_source(s) assert a.get_source_names() == ('Mock Source',)