ephraimbuddy commented on a change in pull request #19931:
URL: https://github.com/apache/airflow/pull/19931#discussion_r760395447
##########
File path: tests/models/test_pool.py
##########
@@ -15,27 +15,51 @@
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
+import random
+import string
+
+import pytest
from airflow import settings
+from airflow.exceptions import AirflowBadRequest, PoolNotFound
from airflow.models.pool import Pool
from airflow.models.taskinstance import TaskInstance as TI
from airflow.operators.dummy import DummyOperator
from airflow.utils import timezone
+from airflow.utils.session import create_session
from airflow.utils.state import State
from tests.test_utils.db import clear_db_dags, clear_db_pools, clear_db_runs,
set_default_pool_slots
DEFAULT_DATE = timezone.datetime(2016, 1, 1)
class TestPool:
+
+ USER_POOL_COUNT = 2
+ TOTAL_POOL_COUNT = USER_POOL_COUNT + 1 # including default_pool
+
@staticmethod
def clean_db():
clear_db_dags()
clear_db_runs()
clear_db_pools()
- def setup_method(self):
+ def setup_method(self, session):
Review comment:
```suggestion
def setup_method(self):
```
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
