• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• django-sqlserver-master
  • acceptance_test.py
  • sqlserver
    • __init__.py
    • base.py
  • LICENSE
  • appveyor.yml
  • README.rst
  • setup.py
  • test-results
    • results-django1.8-mssql-096bd99-3.txt
    • results-django1.7-mssql-1.6.2.txt
    • results-django1.8-mssql-1.7.txt
    • results-django1.7-mssql-1.7.txt
    • results-django1.8-mssql-096bd99-2.txt
    • results-django1.8-mssql-096bd99.txt
  • setup.cfg
  • tests
    • m2m_and_m2o
      • models.py
      • tests.py
      • __init__.py
    • ordering
      • models.py
      • tests.py
      • __init__.py
    • select_related
      • models.py
      • tests.py
      • __init__.py
    • force_insert_update
      • models.py
      • tests.py
      • __init__.py
    • update_only_fields
      • models.py
      • tests.py
      • __init__.py
    • pagination
      • models.py
      • tests.py
      • __init__.py
      • custom.py
    • raw_query
      • models.py
      • tests.py
      • __init__.py
    • multiple_database
      • routers.py
      • fixtures
        • pets.json
        • multidb.default.json
        • multidb.other.json
        • multidb-common.json
      • models.py
      • tests.py
      • __init__.py
    • transactions
      • models.py
      • tests.py
      • __init__.py
    • nested_foreign_keys
      • models.py
      • tests.py
      • __init__.py
    • select_related_regress
      • models.py
      • tests.py
      • __init__.py
    • m2m_through
      • models.py
      • tests.py
      • __init__.py
    • defer_regress
      • models.py
      • tests.py
      • __init__.py
    • m2m_recursive
      • models.py
      • tests.py
      • __init__.py
    • delete_regress
      • models.py
      • tests.py
      • __init__.py
    • custom_columns
      • models.py
      • tests.py
      • __init__.py
    • m2m_multiple
      • models.py
      • tests.py
      • __init__.py
    • unmanaged_models
      • models.py
      • tests.py
      • __init__.py
    • bulk_create
      • models.py
      • tests.py
      • __init__.py
    • null_fk_ordering
      • models.py
      • tests.py
      • __init__.py
    • field_subclassing
      • fields.py
      • tests.py
      • __init__.py
    • expressions
      • models.py
      • test_queryset_values.py
      • tests.py
      • __init__.py
    • null_fk
      • models.py
      • tests.py
      • __init__.py
    • transaction_hooks
      • models.py
      • tests.py
      • __init__.py
    • aggregation_regress
      • models.py
      • tests.py
      • __init__.py
    • urls.py
    • update
      • models.py
      • tests.py
      • __init__.py
    • aggregation
      • models.py
      • tests.py
      • __init__.py
    • reverse_lookup
      • models.py
      • tests.py
      • __init__.py
    • distinct_on_fields
      • models.py
      • tests.py
      • __init__.py
    • datatypes
      • models.py
      • tests.py
      • __init__.py
    • runtests.py
    • __init__.py
    • db_typecasts
      • tests.py
      • __init__.py
    • m2m_intermediary
      • models.py
      • tests.py
      • __init__.py
    • field_defaults
      • models.py
      • tests.py
      • __init__.py
    • dates
      • models.py
      • tests.py
      • __init__.py
    • select_for_update
      • models.py
      • tests.py
      • __init__.py
    • select_related_onetoone
      • models.py
      • tests.py
      • __init__.py
    • or_lookups
      • models.py
      • tests.py
      • __init__.py
    • queries
      • models.py
      • tests.py
      • __init__.py
      • test_qs_combinators.py
    • m2m_signals
      • models.py
      • tests.py
      • __init__.py
    • migrations2
      • test_migrations_2_no_deps
        • 0001_initial.py
        • __init__.py
      • models.py
      • __init__.py
      • test_migrations_2_first
        • 0002_second.py
        • 0001_initial.py
        • __init__.py
      • test_migrations_2
        • 0001_initial.py
        • __init__.py
    • many_to_one_null
      • models.py
      • tests.py
      • __init__.py
    • indexes
      • models.py
      • tests.py
      • __init__.py
    • test_mssql.py
    • field_deconstruction
      • tests.py
      • __init__.py
    • generic_relations_regress
      • models.py
      • tests.py
      • __init__.py
    • migration_test_data_persistence
      • models.py
      • tests.py
      • __init__.py
      • migrations
        • 0001_initial.py
        • __init__.py
        • 0002_add_book.py
    • custom_pk
      • fields.py
      • models.py
      • tests.py
      • __init__.py
    • m2m_regress
      • models.py
      • tests.py
      • __init__.py
    • datetimes
      • models.py
      • tests.py
      • __init__.py
    • defer
      • models.py
      • tests.py
      • __init__.py
    • m2o_recursive
      • models.py
      • tests.py
      • __init__.py
  • broken-tests
    • backends
      • test_utils.py
      • test_postgresql.py
      • models.py
      • test_creation.py
      • test_features.py
      • tests.py
      • __init__.py
      • test_mysql.py
    • null_queries
      • models.py
      • tests.py
      • __init__.py
    • inspectdb
      • models.py
      • tests.py
      • __init__.py
    • many_to_many
      • models.py
      • tests.py
      • __init__.py
    • generic_relations
      • models.py
      • tests.py
      • __init__.py
    • m2m_through_regress
      • fixtures
        • m2m_through.json
      • test_multitable.py
      • models.py
      • tests.py
      • __init__.py
    • get_or_create
      • models.py
      • tests.py
      • __init__.py
    • many_to_one
      • models.py
      • tests.py
      • __init__.py
    • expressions_case
      • models.py
      • tests.py
      • __init__.py
    • introspection
      • models.py
      • tests.py
      • __init__.py
    • db_functions
      • test_cast.py
      • models.py
      • tests.py
      • __init__.py
      • test_datetime.py
    • migrate_signals
      • custom_migrations
        • 0001_initial.py
        • __init__.py
      • models.py
      • tests.py
      • __init__.py
    • one_to_one
      • models.py
      • tests.py
      • __init__.py
    • migrations
      • test_migrations_squashed_complex_multi_apps
        • app1
          • 4_auto.py
          • 2_squashed_3.py
          • 3_auto.py
          • 2_auto.py
          • 1_auto.py
          • __init__.py
        • __init__.py
        • app2
          • 2_auto.py
          • 1_auto.py
          • 1_squashed_2.py
          • __init__.py
      • deprecated_field_migrations
        • 0002_remove_ipaddressfield_ip.py
        • 0001_initial.py
        • __init__.py
      • test_optimizer.py
      • test_migrations_squashed_extra
        • 0002_second.py
        • 0001_initial.py
        • __init__.py
        • 0003_third.py
        • 0001_squashed_0002.py
      • test_base.py
      • test_state.py
      • migrations_test_apps
        • migrated_app
          • models.py
          • __init__.py
          • migrations
            • 0001_initial.py
            • __init__.py
        • conflicting_app_with_dependencies
          • __init__.py
          • migrations
            • 0002_conflicting_second.py
            • 0002_second.py
            • 0001_initial.py
            • __init__.py
        • mutate_state_a
          • __init__.py
          • migrations
            • 0001_initial.py
            • __init__.py
        • without_init_file
          • __init__.py
          • migrations
            • .keep
        • alter_fk
          • book_app
            • __init__.py
            • migrations
              • 0001_initial.py
              • __init__.py
          • __init__.py
          • author_app
            • __init__.py
            • migrations
              • 0001_initial.py
              • 0002_alter_id.py
              • __init__.py
        • lookuperror_a
          • models.py
          • __init__.py
          • migrations
            • 0003_a3.py
            • 0002_a2.py
            • 0001_initial.py
            • __init__.py
            • 0004_a4.py
        • normal
          • __init__.py
        • mutate_state_b
          • __init__.py
          • migrations
            • 0002_add_field.py
            • 0001_initial.py
            • __init__.py
        • with_package_model
          • models
            • __init__.py
          • __init__.py
        • unmigrated_app_syncdb
          • models.py
          • __init__.py
        • migrated_unapplied_app
          • models.py
          • __init__.py
          • migrations
            • 0001_initial.py
            • __init__.py
        • lookuperror_b
          • models.py
          • __init__.py
          • migrations
            • 0002_b2.py
            • 0003_b3.py
            • 0001_initial.py
            • __init__.py
        • __init__.py
        • unmigrated_app
          • models.py
          • __init__.py
        • unspecified_app_with_conflict
          • models.py
          • __init__.py
          • migrations
            • 0002_conflicting_second.py
            • 0002_second.py
            • 0001_initial.py
            • __init__.py
        • lookuperror_c
          • models.py
          • __init__.py
          • migrations
            • 0002_c2.py
            • 0001_initial.py
            • __init__.py
            • 0003_c3.py
      • test_migrations_squashed_ref_squashed
        • app1
          • 4_auto.py
          • 2_squashed_3.py
          • 3_auto.py
          • 2_auto.py
          • 1_auto.py
          • __init__.py
        • __init__.py
        • app2
          • 2_auto.py
          • 1_auto.py
          • 1_squashed_2.py
          • __init__.py
      • test_migrations_squash_noop
        • 0002_second.py
        • 0001_initial.py
        • __init__.py
      • routers.py
      • test_exceptions.py
      • test_migrations_backwards_deps_1
        • 0002_second.py
        • 0001_initial.py
      • test_questioner.py
      • test_operations.py
      • test_migrations_squashed
        • 0002_second.py
        • 0001_initial.py
        • __init__.py
        • 0001_squashed_0002.py
      • test_migrations_run_before
        • 0002_second.py
        • 0001_initial.py
        • __init__.py
        • 0003_third.py
      • test_migrations_no_default
        • 0001_initial.py
        • __init__.py
      • test_commands.py
      • test_loader.py
      • test_migrations_initial_false
        • 0001_not_initial.py
        • __init__.py
      • test_migrations_custom_user
        • 0001_initial.py
        • __init__.py
      • test_graph.py
      • test_migrations_fake_split_initial
        • 0002_second.py
        • 0001_initial.py
        • __init__.py
      • faulty_migrations
        • namespace
          • foo
            • __init__.py
        • file.py
        • __init__.py
      • related_models_app
        • __init__.py
      • test_deprecated_fields.py
      • models.py
      • test_migrations_squashed_complex
        • 7_auto.py
        • 4_auto.py
        • 3_auto.py
        • 5_auto.py
        • 6_auto.py
        • 3_squashed_5.py
        • 2_auto.py
        • 1_auto.py
        • __init__.py
      • test_migrations_first
        • thefirst.py
        • second.py
        • __init__.py
      • test_migrations_non_atomic
        • 0001_initial.py
        • __init__.py
      • test_executor.py
      • test_migrations_no_ancestor
        • 0002_conflicting_second.py
        • 0002_second.py
        • 0001_initial.py
        • __init__.py
      • __init__.py
      • test_auto_now_add
        • 0001_initial.py
        • __init__.py
      • test_add_many_to_many_field_initial
        • 0001_initial.py
        • 0002_initial.py
        • __init__.py
      • test_migrations_conflict
        • 0002_conflicting_second.py
        • 0002_second.py
        • 0001_initial.py
        • __init__.py
      • test_migrations_unmigdep
        • 0001_initial.py
        • __init__.py
      • test_migrations_no_changes
        • 0002_second.py
        • 0001_initial.py
        • __init__.py
        • 0003_third.py
      • test_migrations_empty
        • __init__.py
      • test_migrations_order
        • 0001.py
        • __init__.py
      • test_migrations_atomic_operation
        • 0001_initial.py
        • __init__.py
      • test_writer.py
      • test_migrations_squashed_erroneous
        • 7_auto.py
        • 6_auto.py
        • 3_squashed_5.py
        • 2_auto.py
        • 1_auto.py
        • __init__.py
      • test_autodetector.py
      • test_multidb.py
      • test_migrations
        • 0002_second.py
        • 0001_initial.py
        • __init__.py
    • timezones
      • models.py
      • urls.py
      • forms.py
      • tests.py
      • __init__.py
      • admin.py
    • schema
      • fields.py
      • models.py
      • tests.py
      • __init__.py
    • order_with_respect_to
      • base_tests.py
      • models.py
      • tests.py
      • __init__.py
  • changelog.rst
  • .gitignore
  • MANIFEST.in

from __future__ import unicode_literals

from django.db.models.signals import post_save, pre_save
from django.test import TestCase

from .models import Account, Employee, Person, Profile, ProxyEmployee


class UpdateOnlyFieldsTests(TestCase):
    def test_update_fields_basic(self):
        s = Person.objects.create(name='Sara', gender='F')
        self.assertEqual(s.gender, 'F')

        s.gender = 'M'
        s.name = 'Ian'
        s.save(update_fields=['name'])

        s = Person.objects.get(pk=s.pk)
        self.assertEqual(s.gender, 'F')
        self.assertEqual(s.name, 'Ian')

    def test_update_fields_deferred(self):
        s = Person.objects.create(name='Sara', gender='F', pid=22)
        self.assertEqual(s.gender, 'F')

        s1 = Person.objects.defer("gender", "pid").get(pk=s.pk)
        s1.name = "Emily"
        s1.gender = "M"

        with self.assertNumQueries(1):
            s1.save()

        s2 = Person.objects.get(pk=s1.pk)
        self.assertEqual(s2.name, "Emily")
        self.assertEqual(s2.gender, "M")

    def test_update_fields_only_1(self):
        s = Person.objects.create(name='Sara', gender='F')
        self.assertEqual(s.gender, 'F')

        s1 = Person.objects.only('name').get(pk=s.pk)
        s1.name = "Emily"
        s1.gender = "M"

        with self.assertNumQueries(1):
            s1.save()

        s2 = Person.objects.get(pk=s1.pk)
        self.assertEqual(s2.name, "Emily")
        self.assertEqual(s2.gender, "M")

    def test_update_fields_only_2(self):
        s = Person.objects.create(name='Sara', gender='F', pid=22)
        self.assertEqual(s.gender, 'F')

        s1 = Person.objects.only('name').get(pk=s.pk)
        s1.name = "Emily"
        s1.gender = "M"

        with self.assertNumQueries(2):
            s1.save(update_fields=['pid'])

        s2 = Person.objects.get(pk=s1.pk)
        self.assertEqual(s2.name, "Sara")
        self.assertEqual(s2.gender, "F")

    def test_update_fields_only_repeated(self):
        s = Person.objects.create(name='Sara', gender='F')
        self.assertEqual(s.gender, 'F')

        s1 = Person.objects.only('name').get(pk=s.pk)
        s1.gender = 'M'
        with self.assertNumQueries(1):
            s1.save()
        # save() should not fetch deferred fields
        s1 = Person.objects.only('name').get(pk=s.pk)
        with self.assertNumQueries(1):
            s1.save()

    def test_update_fields_inheritance_defer(self):
        profile_boss = Profile.objects.create(name='Boss', salary=3000)
        e1 = Employee.objects.create(name='Sara', gender='F', employee_num=1, profile=profile_boss)
        e1 = Employee.objects.only('name').get(pk=e1.pk)
        e1.name = 'Linda'
        with self.assertNumQueries(1):
            e1.save()
        self.assertEqual(Employee.objects.get(pk=e1.pk).name, 'Linda')

    def test_update_fields_fk_defer(self):
        profile_boss = Profile.objects.create(name='Boss', salary=3000)
        profile_receptionist = Profile.objects.create(name='Receptionist', salary=1000)
        e1 = Employee.objects.create(name='Sara', gender='F', employee_num=1, profile=profile_boss)
        e1 = Employee.objects.only('profile').get(pk=e1.pk)
        e1.profile = profile_receptionist
        with self.assertNumQueries(1):
            e1.save()
        self.assertEqual(Employee.objects.get(pk=e1.pk).profile, profile_receptionist)
        e1.profile_id = profile_boss.pk
        with self.assertNumQueries(1):
            e1.save()
        self.assertEqual(Employee.objects.get(pk=e1.pk).profile, profile_boss)

    def test_select_related_only_interaction(self):
        profile_boss = Profile.objects.create(name='Boss', salary=3000)
        e1 = Employee.objects.create(name='Sara', gender='F', employee_num=1, profile=profile_boss)
        e1 = Employee.objects.only('profile__salary').select_related('profile').get(pk=e1.pk)
        profile_boss.name = 'Clerk'
        profile_boss.salary = 1000
        profile_boss.save()
        # The loaded salary of 3000 gets saved, the name of 'Clerk' isn't
        # overwritten.
        with self.assertNumQueries(1):
            e1.profile.save()
        reloaded_profile = Profile.objects.get(pk=profile_boss.pk)
        self.assertEqual(reloaded_profile.name, profile_boss.name)
        self.assertEqual(reloaded_profile.salary, 3000)

    def test_update_fields_m2m(self):
        profile_boss = Profile.objects.create(name='Boss', salary=3000)
        e1 = Employee.objects.create(name='Sara', gender='F', employee_num=1, profile=profile_boss)
        a1 = Account.objects.create(num=1)
        a2 = Account.objects.create(num=2)
        e1.accounts.set([a1, a2])

        with self.assertRaises(ValueError):
            e1.save(update_fields=['accounts'])

    def test_update_fields_inheritance(self):
        profile_boss = Profile.objects.create(name='Boss', salary=3000)
        profile_receptionist = Profile.objects.create(name='Receptionist', salary=1000)
        e1 = Employee.objects.create(name='Sara', gender='F', employee_num=1, profile=profile_boss)

        e1.name = 'Ian'
        e1.gender = 'M'
        e1.save(update_fields=['name'])

        e2 = Employee.objects.get(pk=e1.pk)
        self.assertEqual(e2.name, 'Ian')
        self.assertEqual(e2.gender, 'F')
        self.assertEqual(e2.profile, profile_boss)

        e2.profile = profile_receptionist
        e2.name = 'Sara'
        e2.save(update_fields=['profile'])

        e3 = Employee.objects.get(pk=e1.pk)
        self.assertEqual(e3.name, 'Ian')
        self.assertEqual(e3.profile, profile_receptionist)

        with self.assertNumQueries(1):
            e3.profile = profile_boss
            e3.save(update_fields=['profile_id'])

        e4 = Employee.objects.get(pk=e3.pk)
        self.assertEqual(e4.profile, profile_boss)
        self.assertEqual(e4.profile_id, profile_boss.pk)

    def test_update_fields_inheritance_with_proxy_model(self):
        profile_boss = Profile.objects.create(name='Boss', salary=3000)
        profile_receptionist = Profile.objects.create(name='Receptionist', salary=1000)
        e1 = ProxyEmployee.objects.create(name='Sara', gender='F', employee_num=1, profile=profile_boss)

        e1.name = 'Ian'
        e1.gender = 'M'
        e1.save(update_fields=['name'])

        e2 = ProxyEmployee.objects.get(pk=e1.pk)
        self.assertEqual(e2.name, 'Ian')
        self.assertEqual(e2.gender, 'F')
        self.assertEqual(e2.profile, profile_boss)

        e2.profile = profile_receptionist
        e2.name = 'Sara'
        e2.save(update_fields=['profile'])

        e3 = ProxyEmployee.objects.get(pk=e1.pk)
        self.assertEqual(e3.name, 'Ian')
        self.assertEqual(e3.profile, profile_receptionist)

    def test_update_fields_signals(self):
        p = Person.objects.create(name='Sara', gender='F')
        pre_save_data = []

        def pre_save_receiver(**kwargs):
            pre_save_data.append(kwargs['update_fields'])
        pre_save.connect(pre_save_receiver)
        post_save_data = []

        def post_save_receiver(**kwargs):
            post_save_data.append(kwargs['update_fields'])
        post_save.connect(post_save_receiver)
        p.save(update_fields=['name'])
        self.assertEqual(len(pre_save_data), 1)
        self.assertEqual(len(pre_save_data[0]), 1)
        self.assertIn('name', pre_save_data[0])
        self.assertEqual(len(post_save_data), 1)
        self.assertEqual(len(post_save_data[0]), 1)
        self.assertIn('name', post_save_data[0])

        pre_save.disconnect(pre_save_receiver)
        post_save.disconnect(post_save_receiver)

    def test_update_fields_incorrect_params(self):
        s = Person.objects.create(name='Sara', gender='F')

        with self.assertRaises(ValueError):
            s.save(update_fields=['first_name'])

        with self.assertRaises(ValueError):
            s.save(update_fields="name")

    def test_empty_update_fields(self):
        s = Person.objects.create(name='Sara', gender='F')
        pre_save_data = []

        def pre_save_receiver(**kwargs):
            pre_save_data.append(kwargs['update_fields'])
        pre_save.connect(pre_save_receiver)
        post_save_data = []

        def post_save_receiver(**kwargs):
            post_save_data.append(kwargs['update_fields'])
        post_save.connect(post_save_receiver)
        # Save is skipped.
        with self.assertNumQueries(0):
            s.save(update_fields=[])
        # Signals were skipped, too...
        self.assertEqual(len(pre_save_data), 0)
        self.assertEqual(len(post_save_data), 0)

        pre_save.disconnect(pre_save_receiver)
        post_save.disconnect(post_save_receiver)

    def test_num_queries_inheritance(self):
        s = Employee.objects.create(name='Sara', gender='F')
        s.employee_num = 1
        s.name = 'Emily'
        with self.assertNumQueries(1):
            s.save(update_fields=['employee_num'])
        s = Employee.objects.get(pk=s.pk)
        self.assertEqual(s.employee_num, 1)
        self.assertEqual(s.name, 'Sara')
        s.employee_num = 2
        s.name = 'Emily'
        with self.assertNumQueries(1):
            s.save(update_fields=['name'])
        s = Employee.objects.get(pk=s.pk)
        self.assertEqual(s.name, 'Emily')
        self.assertEqual(s.employee_num, 1)
        # A little sanity check that we actually did updates...
        self.assertEqual(Employee.objects.count(), 1)
        self.assertEqual(Person.objects.count(), 1)
        with self.assertNumQueries(2):
            s.save(update_fields=['name', 'employee_num'])