From 6af023dafd52327507e9a191c5c9e64919ab58a8 Mon Sep 17 00:00:00 2001 From: Diederik van der Boor Date: Mon, 15 Jul 2019 17:56:53 +0200 Subject: [PATCH] Added unit test for refresh_from_db() --- polymorphic/tests/test_orm.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/polymorphic/tests/test_orm.py b/polymorphic/tests/test_orm.py index af73b95..cf9180e 100644 --- a/polymorphic/tests/test_orm.py +++ b/polymorphic/tests/test_orm.py @@ -1234,3 +1234,14 @@ class PolymorphicTests(TransactionTestCase): self.assertEqual( len(objects[1].many2many.non_polymorphic()), 1 ) # base object does exist + + def test_refresh_from_db_fields(self): + """Test whether refresh_from_db(fields=..) works as it performs .only() queries""" + obj = Model2B.objects.create(field1="aa", field2="bb") + Model2B.objects.filter(pk=obj.pk).update(field1="aa1", field2="bb2") + obj.refresh_from_db(fields=["field2"]) + assert obj.field1 == "aa" + assert obj.field2 == "bb2" + + obj.refresh_from_db(fields=["field1"]) + assert obj.field1 == "aa1"