def test_dont_get_primary_key(self):
meta = Person.objects.first()._meta
update_fields = None
exclude_fields = None
fields = helper.get_fields(update_fields, exclude_fields, meta)
self._assertIn(['id'], meta.get_fields()) # sanity check
self._assertNotIn(['id'], fields) # actual test
meta = PersonUUID.objects.create(age=3)._meta
update_fields = None
exclude_fields = None
fields = helper.get_fields(update_fields, exclude_fields, meta)
self._assertIn(['uuid'], meta.get_fields()) # sanity check
self._assertNotIn(['uuid'], fields) # actual test
def test_dont_get_primary_key(self):
meta = Person.objects.first()._meta
update_fields = None
exclude_fields = None
fields = helper.get_fields(update_fields, exclude_fields, meta)
self._assertIn(['id'], meta.get_fields()) # sanity check
self._assertNotIn(['id'], fields) # actual test
meta = PersonUUID.objects.create(age=3)._meta
update_fields = None
exclude_fields = None
fields = helper.get_fields(update_fields, exclude_fields, meta)
self._assertIn(['uuid'], meta.get_fields()) # sanity check
self._assertNotIn(['uuid'], fields) # actual test
def test_exclude_foreignkey_with_id_suffix(self):
meta = Person.objects.first()._meta
update_fields = None
exclude_fields = ['email', 'role_id']
fields = helper.get_fields(update_fields, exclude_fields, meta)
self.assertEquals(len(fields), self.total_fields - 2)
self._assertNotIn(['email', 'role'], fields)
def test_dont_get_many_to_many_relations(self):
meta = Person.objects.first()._meta
update_fields = None
exclude_fields = None
fields = helper.get_fields(update_fields, exclude_fields, meta)
self._assertIn(['jobs'], meta.get_fields()) # sanity check
self._assertNotIn(['jobs'], fields) # actual test
def test_update_fields_over_not_deferred_field_02(self):
obj = Person.objects.only('name', 'age', 'height').first()
meta = obj._meta
update_fields = []
exclude_fields = None
fields = helper.get_fields(update_fields, exclude_fields, meta, obj)
self._assertEquals(fields, [])
def test_obj_argument(self):
obj = Person.objects.first()
meta = obj._meta
update_fields = None
exclude_fields = None
fields = helper.get_fields(update_fields, exclude_fields, meta, obj)
self.assertEquals(len(fields), self.total_fields)
def test_dont_get_many_to_many_relations(self):
meta = Person.objects.first()._meta
update_fields = None
exclude_fields = None
fields = helper.get_fields(update_fields, exclude_fields, meta)
self._assertIn(['jobs'], meta.get_fields()) # sanity check
self._assertNotIn(['jobs'], fields) # actual test
def test_exclude_foreignkey_with_id_suffix(self):
meta = Person.objects.first()._meta
update_fields = None
exclude_fields = ['email', 'role_id']
fields = helper.get_fields(update_fields, exclude_fields, meta)
self.assertEquals(len(fields), self.total_fields - 2)
self._assertNotIn(['email', 'role'], fields)
def test_update_fields_over_not_deferred_field_02(self):
obj = Person.objects.only('name', 'age', 'height').first()
meta = obj._meta
update_fields = []
exclude_fields = None
fields = helper.get_fields(update_fields, exclude_fields, meta, obj)
self._assertEquals(fields, [])
def test_only_and_exclude_fields(self):
obj = Person.objects.only('name', 'age', 'height').first()
meta = obj._meta
update_fields = None
exclude_fields = ['age', 'date']
fields = helper.get_fields(update_fields, exclude_fields, meta, obj)
self._assertEquals(fields, ['name', 'height'])
def test_only_and_exclude_fields(self):
obj = Person.objects.only('name', 'age', 'height').first()
meta = obj._meta
update_fields = None
exclude_fields = ['age', 'date']
fields = helper.get_fields(update_fields, exclude_fields, meta, obj)
self._assertEquals(fields, ['name', 'height'])
def test_obj_argument(self):
obj = Person.objects.first()
meta = obj._meta
update_fields = None
exclude_fields = None
fields = helper.get_fields(update_fields, exclude_fields, meta, obj)
self.assertEquals(len(fields), self.total_fields)
def test_only_and_exclude_fields_02(self):
obj = Person.objects.defer('age', 'height').first()
meta = obj._meta
update_fields = None
exclude_fields = ['image', 'data']
fields = helper.get_fields(update_fields, exclude_fields, meta, obj)
self.assertEquals(len(fields), self.total_fields - 4)
self._assertNotIn(['age', 'height', 'image', 'data'], fields)
def test_only_and_exclude_fields_02(self):
obj = Person.objects.defer('age', 'height').first()
meta = obj._meta
update_fields = None
exclude_fields = ['image', 'data']
fields = helper.get_fields(update_fields, exclude_fields, meta, obj)
self.assertEquals(len(fields), self.total_fields - 4)
self._assertNotIn(['age', 'height', 'image', 'data'], fields)
def test_get_foreignkey_with_id_suffix(self):
meta = Person.objects.first()._meta
update_fields = ['role_id', 'my_file']
exclude_fields = None
fields = helper.get_fields(update_fields, exclude_fields, meta)
self._assertEquals(fields, ['role', 'my_file'])
def test_empty_update_fields(self):
meta = Person.objects.first()._meta
update_fields = []
exclude_fields = ['email', 'height']
fields = helper.get_fields(update_fields, exclude_fields, meta)
self._assertEquals(fields, [])
def test_update_fields_and_exclude_fields(self):
meta = Person.objects.first()._meta
update_fields = ['age', 'email', 'text']
exclude_fields = ['email', 'height']
fields = helper.get_fields(update_fields, exclude_fields, meta)
self._assertEquals(fields, ['age', 'text'])
def test_update_fields_and_exclude_fields(self):
meta = Person.objects.first()._meta
update_fields = ['age', 'email', 'text']
exclude_fields = ['email', 'height']
fields = helper.get_fields(update_fields, exclude_fields, meta)
self._assertEquals(fields, ['age', 'text'])
def test_get_all_fields(self):
meta = Person.objects.first()._meta
update_fields = None
exclude_fields = None
fields = helper.get_fields(update_fields, exclude_fields, meta)
self.assertEquals(len(fields), self.total_fields)
def test_empty_update_fields(self):
meta = Person.objects.first()._meta
update_fields = []
exclude_fields = ['email', 'height']
fields = helper.get_fields(update_fields, exclude_fields, meta)
self._assertEquals(fields, [])
def test_get_foreignkey_with_id_suffix(self):
meta = Person.objects.first()._meta
update_fields = ['role_id', 'my_file']
exclude_fields = None
fields = helper.get_fields(update_fields, exclude_fields, meta)
self._assertEquals(fields, ['role', 'my_file'])
def test_arguments_as_tuples(self):
meta = Person.objects.first()._meta
update_fields = ('age', 'email', 'text')
exclude_fields = ('email', 'height')
fields = helper.get_fields(update_fields, exclude_fields, meta)
self._assertEquals(fields, ['age', 'text'])
def test_get_a_foreignkey(self):
meta = Person.objects.first()._meta
update_fields = ['role', 'my_file']
exclude_fields = None
fields = helper.get_fields(update_fields, exclude_fields, meta)
self._assertEquals(fields, ['role', 'my_file', 'auto_now'])
def test_arguments_as_tuples(self):
meta = Person.objects.first()._meta
update_fields = ('age', 'email', 'text')
exclude_fields = ('email', 'height')
fields = helper.get_fields(update_fields, exclude_fields, meta)
self._assertEquals(fields, ['age', 'text'])
def test_get_all_fields(self):
meta = Person.objects.first()._meta
update_fields = None
exclude_fields = None
fields = helper.get_fields(update_fields, exclude_fields, meta)
self.assertEquals(len(fields), self.total_fields)
