mirror of
https://github.com/Instagram/LibCST.git
synced 2025-12-23 10:35:53 +00:00
1190dc8a2f
This was a pain, because apparently we can't refer to "static_analysis.libcst" as a module before its finished executing, due to https://bugs.python.org/issue25294. To combat this, I had to change to direct importing inside parser, which was a bit of a pain due to the number of nodes used. But, it works.
38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
# Copyright (c) Facebook, Inc. and its affiliates.
|
|
#
|
|
# This source code is licensed under the MIT license found in the
|
|
# LICENSE file in the root directory of this source tree.
|
|
|
|
# pyre-strict
|
|
from typing import cast
|
|
from unittest.mock import Mock
|
|
|
|
import libcst as cst
|
|
from libcst import BatchableCSTVisitor, parse_module, visit_batched
|
|
from libcst.testing.utils import UnitTest
|
|
|
|
|
|
class BatchedVisitorTest(UnitTest):
|
|
def test(self) -> None:
|
|
mock = Mock()
|
|
|
|
class ABatchable(BatchableCSTVisitor):
|
|
def visit_Pass(self, node: cst.Pass) -> None:
|
|
mock.visited_a()
|
|
object.__setattr__(node, "a_attr", True)
|
|
|
|
class BBatchable(BatchableCSTVisitor):
|
|
def visit_Pass(self, node: cst.Pass) -> None:
|
|
mock.visited_b()
|
|
object.__setattr__(node, "b_attr", 1)
|
|
|
|
module = visit_batched(parse_module("pass"), [ABatchable(), BBatchable()])
|
|
pass_ = cast(cst.SimpleStatementLine, module.body[0]).body[0]
|
|
|
|
# Check properties were set
|
|
self.assertEqual(object.__getattribute__(pass_, "a_attr"), True)
|
|
self.assertEqual(object.__getattribute__(pass_, "b_attr"), 1)
|
|
|
|
# Check that each visitor was only called once
|
|
mock.visited_a.assert_called_once()
|
|
mock.visited_b.assert_called_once()
|