mirror of
https://github.com/encode/django-rest-framework.git
synced 2024-09-21 11:29:01 +03:00
c674687782
* Identify code that needs to be pulled out of/removed from compat.py * Extract modern code from get_names_and_managers in compat.py and remove compat code * Extract modern code from is_authenticated() in compat.py and remove. * Extract modern code from is_anonymous() in compat.py and remove * Extract modern code from get_related_model() from compat.py and remove * Extract modern code from value_from_object() in compat.py and remove * Update postgres compat JSONField now always available. * Remove DecimalValidator compat * Remove get_remote_field compat * Remove template_render compat Plus isort. * Remove set_many compat * Remove include compat
54 lines
1.8 KiB
Python
54 lines
1.8 KiB
Python
from django.test import TestCase
|
|
|
|
from rest_framework import compat
|
|
|
|
|
|
class CompatTests(TestCase):
|
|
|
|
def setUp(self):
|
|
self.original_django_version = compat.django.VERSION
|
|
self.original_transaction = compat.transaction
|
|
|
|
def tearDown(self):
|
|
compat.django.VERSION = self.original_django_version
|
|
compat.transaction = self.original_transaction
|
|
|
|
def test_total_seconds(self):
|
|
class MockTimedelta(object):
|
|
days = 1
|
|
seconds = 1
|
|
microseconds = 100
|
|
timedelta = MockTimedelta()
|
|
expected = (timedelta.days * 86400.0) + float(timedelta.seconds) + (timedelta.microseconds / 1000000.0)
|
|
assert compat.total_seconds(timedelta) == expected
|
|
|
|
def test_set_rollback_for_transaction_in_managed_mode(self):
|
|
class MockTransaction(object):
|
|
called_rollback = False
|
|
called_leave_transaction_management = False
|
|
|
|
def is_managed(self):
|
|
return True
|
|
|
|
def is_dirty(self):
|
|
return True
|
|
|
|
def rollback(self):
|
|
self.called_rollback = True
|
|
|
|
def leave_transaction_management(self):
|
|
self.called_leave_transaction_management = True
|
|
|
|
dirty_mock_transaction = MockTransaction()
|
|
compat.transaction = dirty_mock_transaction
|
|
compat.set_rollback()
|
|
assert dirty_mock_transaction.called_rollback is True
|
|
assert dirty_mock_transaction.called_leave_transaction_management is True
|
|
|
|
clean_mock_transaction = MockTransaction()
|
|
clean_mock_transaction.is_dirty = lambda: False
|
|
compat.transaction = clean_mock_transaction
|
|
compat.set_rollback()
|
|
assert clean_mock_transaction.called_rollback is False
|
|
assert clean_mock_transaction.called_leave_transaction_management is True
|