def test_getPosters_When_struck_out_Then_skip_delegate_and_return_nextProviders_result(self): # Setup provider = OneStrikeAndYoureOutFanartProvider(self.platform, self.delegate, self.nextProvider) key = provider.createKey('getPosters', self.program) provider.strikeOut(key, self.program) when(self.nextProvider).getPosters(any()).thenReturn(['blah.png']) # Test posters = provider.getPosters(self.program) # Verify self.assertEqual('blah.png', posters[0]) self.assertIn(self.program.title(), provider.struckOut[key].values()) verifyZeroInteractions(self.delegate)
def test_getSeasonAndEpisode_When_struck_out_Then_skip_delegate_and_return_nextProviders_result(self): # Setup provider = OneStrikeAndYoureOutFanartProvider(self.platform, self.delegate, self.nextProvider) key = provider.createKey('getSeasonAndEpisode', self.program) provider.strikeOut(key, self.program) when(self.nextProvider).getSeasonAndEpisode(any()).thenReturn(('1','2')) # Test season, episode = provider.getSeasonAndEpisode(self.program) # Verify self.assertEqual('1', season) self.assertEqual('2', episode) self.assertIn(self.program.title(), provider.struckOut[key].values()) verifyZeroInteractions(self.delegate)