mirror of
https://github.com/thegeeklab/ansible-later.git
synced 2024-07-03 15:40:59 +02:00
112 lines
3.3 KiB
Python
112 lines
3.3 KiB
Python
from unittest import TestCase
|
|
|
|
import pytest
|
|
from django.contrib.auth.models import User
|
|
from testfixtures import OutputCapture, Replacer
|
|
from testfixtures.compat import PY3
|
|
from .models import SampleModel
|
|
from testfixtures.tests.test_django.manage import main
|
|
|
|
from ..test_compare import CompareHelper
|
|
from ... import compare
|
|
from ...django import compare as django_compare
|
|
|
|
|
|
class CompareTests(CompareHelper, TestCase):
|
|
|
|
def test_simple_same(self):
|
|
django_compare(SampleModel(id=1), SampleModel(id=1))
|
|
|
|
def test_simple_diff(self):
|
|
if PY3:
|
|
expected = "'id': 1 != 2"
|
|
else:
|
|
expected = "u'id': 1 != 2"
|
|
|
|
self.check_raises(
|
|
SampleModel(id=1), SampleModel(id=2),
|
|
compare=django_compare,
|
|
message=(
|
|
'SampleModel not as expected:\n'
|
|
'\n'
|
|
'same:\n'
|
|
"['value']\n"
|
|
'\n'
|
|
'values differ:\n'+
|
|
expected
|
|
)
|
|
)
|
|
|
|
def test_simple_ignore_fields(self):
|
|
django_compare(SampleModel(id=1), SampleModel(id=1),
|
|
ignore_fields=['id'])
|
|
|
|
def test_ignored_because_speshul(self):
|
|
django_compare(SampleModel(not_editable=1), SampleModel(not_editable=2))
|
|
|
|
def test_ignored_because_no_longer_speshul(self):
|
|
if PY3:
|
|
same = "['created', 'id', 'value']\n"
|
|
else:
|
|
same = "['created', u'id', 'value']\n"
|
|
self.check_raises(
|
|
SampleModel(not_editable=1), SampleModel(not_editable=2),
|
|
compare=django_compare,
|
|
message=(
|
|
'SampleModel not as expected:\n'
|
|
'\n'
|
|
'same:\n'+
|
|
same+
|
|
'\n'
|
|
'values differ:\n'
|
|
"'not_editable': 1 != 2"
|
|
),
|
|
non_editable_fields=True
|
|
)
|
|
|
|
def test_normal_compare_id_same(self):
|
|
# other diffs ignored
|
|
compare(SampleModel(id=1, value=1), SampleModel(id=1, value=2))
|
|
|
|
def test_normal_compare_id_diff(self):
|
|
if PY3:
|
|
expected = (
|
|
"'id': 3 != 4\n"
|
|
"'value': 1 != 2"
|
|
)
|
|
|
|
else:
|
|
expected = (
|
|
"'value': 1 != 2\n"
|
|
"u'id': 3 != 4"
|
|
)
|
|
|
|
self.check_raises(
|
|
SampleModel(id=3, value=1), SampleModel(id=4, value=2),
|
|
compare=django_compare,
|
|
message=(
|
|
'SampleModel not as expected:\n'
|
|
'\n'
|
|
'values differ:\n'+
|
|
expected
|
|
)
|
|
)
|
|
|
|
def test_manage(self):
|
|
with OutputCapture() as output:
|
|
with Replacer() as r:
|
|
r.replace('os.environ.DJANGO_SETTINGS_MODULE', '', strict=False)
|
|
r.replace('sys.argv', ['x', 'check'])
|
|
main()
|
|
output.compare('System check identified no issues (0 silenced).')
|
|
|
|
@pytest.mark.django_db
|
|
def test_many_to_many_same(self):
|
|
user = User.objects.create(username='foo')
|
|
django_compare(user,
|
|
expected=User(
|
|
username='foo', first_name='', last_name='',
|
|
is_superuser=False
|
|
),
|
|
ignore_fields=['id', 'date_joined'])
|