Skip to content

Commit b4d6a87

Browse files
committed
make truncate_ms() a a function in model_fields_ tests
1 parent 19ef748 commit b4d6a87

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

tests/model_fields_/test_embedded_model.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
Data,
1414
Holder,
1515
)
16+
from .utils import truncate_ms
1617

1718

1819
class MethodTests(SimpleTestCase):
@@ -38,10 +39,6 @@ def test_validate(self):
3839

3940

4041
class ModelTests(TestCase):
41-
def truncate_ms(self, value):
42-
"""Truncate microseconds to milliseconds as supported by MongoDB."""
43-
return value.replace(microsecond=(value.microsecond // 1000) * 1000)
44-
4542
def test_save_load(self):
4643
Holder.objects.create(data=Data(integer="5"))
4744
obj = Holder.objects.get()
@@ -64,12 +61,12 @@ def test_save_load_null(self):
6461
def test_pre_save(self):
6562
"""Field.pre_save() is called on embedded model fields."""
6663
obj = Holder.objects.create(data=Data())
67-
auto_now = self.truncate_ms(obj.data.auto_now)
68-
auto_now_add = self.truncate_ms(obj.data.auto_now_add)
64+
auto_now = truncate_ms(obj.data.auto_now)
65+
auto_now_add = truncate_ms(obj.data.auto_now_add)
6966
self.assertEqual(auto_now, auto_now_add)
7067
# save() updates auto_now but not auto_now_add.
7168
obj.save()
72-
self.assertEqual(self.truncate_ms(obj.data.auto_now_add), auto_now_add)
69+
self.assertEqual(truncate_ms(obj.data.auto_now_add), auto_now_add)
7370
auto_now_two = obj.data.auto_now
7471
self.assertGreater(auto_now_two, obj.data.auto_now_add)
7572
# And again, save() updates auto_now but not auto_now_add.

tests/model_fields_/utils.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
def truncate_ms(value):
2+
"""Truncate microseconds to milliseconds as supported by MongoDB."""
3+
return value.replace(microsecond=(value.microsecond // 1000) * 1000)

0 commit comments

Comments
 (0)