def test_update_reference(self): initialGame = Mock(spec=Game, name="initialGame") nextGame = Mock(spec=Game, name="nextGame") initialGame.transition = Mock(return_value=nextGame) ref = GameReference(initialGame) ref.transition() initialGame.transition.assert_any_call() self.assertIs(ref.game, nextGame)