36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
|
"""
|
||
|
|
Test custom Django management commands.
|
||
|
|
"""
|
||
|
|
from unittest.mock import patch
|
||
|
|
|
||
|
|
from psycopg2 import OperationalError as Psycopg2OpError
|
||
|
|
|
||
|
|
from django.core.management import call_command
|
||
|
|
from django.db.utils import OperationalError
|
||
|
|
from django.test import SimpleTestCase
|
||
|
|
|
||
|
|
|
||
|
|
@patch('core.management.commands.wait_for_db.Command.check')
|
||
|
|
class CommandTests(SimpleTestCase):
|
||
|
|
"""Test Comands."""
|
||
|
|
|
||
|
|
def test_wait_for_db_ready(self, patched_check):
|
||
|
|
"""Test waiting for database if database ready."""
|
||
|
|
patched_check.return_value = True
|
||
|
|
|
||
|
|
call_command('wait_for_db')
|
||
|
|
|
||
|
|
patched_check.assert_called_once_with(databases=['default'])
|
||
|
|
|
||
|
|
# ojo con el orden de los parametros "desde dentro hacia fuera"
|
||
|
|
@patch('time.sleep')
|
||
|
|
def test_wait_for_db_delay(self, patched_sleep, patched_check):
|
||
|
|
"""Test waiting for database when getting OperationalError."""
|
||
|
|
patched_check.side_effect = [Psycopg2OpError] * 2 + \
|
||
|
|
[OperationalError] * 3 + [True]
|
||
|
|
|
||
|
|
call_command('wait_for_db')
|
||
|
|
|
||
|
|
self.assertEqual(patched_check.call_count, 6)
|
||
|
|
patched_check.assert_called_with(databases=['default'])
|