def test_visit_should_call_drop(self):
stmt = MagicMock(spec=DropTableStatement)
convertor = StatementToPlanConvertor()
mock = MagicMock()
convertor.visit_drop = mock
convertor.visit(stmt)
mock.assert_called_once()
mock.assert_called_with(stmt)
def test_visit_should_call_create_udf(self):
stmt = MagicMock(spec=CreateUDFStatement)
convertor = StatementToPlanConvertor()
mock = MagicMock()
convertor.visit_create_udf = mock
convertor.visit(stmt)
mock.assert_called_once()
mock.assert_called_with(stmt)