def test_exit(self): storage = ContentStorage() storage.close = Mock() storage.__exit__() storage.close.assert_called_once_with()
def test_close(self): storage = ContentStorage() storage.close()