def test_set_rating_with_invalid_value_fails(self): rbplayer = RBHandler(self.shell) with self.assertRaises(Exception): rbplayer.set_rating(1, 'x')
def test_set_rating_works(self): rbplayer = RBHandler(self.shell) self.db.entry_lookup_by_id.side_effect = lambda x: x rbplayer.set_rating(1, 5) self.db.entry_set.assert_called_with(1, 'rating', 5)