13
13
Data ,
14
14
Holder ,
15
15
)
16
+ from .utils import truncate_ms
16
17
17
18
18
19
class MethodTests (SimpleTestCase ):
@@ -38,10 +39,6 @@ def test_validate(self):
38
39
39
40
40
41
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
-
45
42
def test_save_load (self ):
46
43
Holder .objects .create (data = Data (integer = "5" ))
47
44
obj = Holder .objects .get ()
@@ -64,12 +61,12 @@ def test_save_load_null(self):
64
61
def test_pre_save (self ):
65
62
"""Field.pre_save() is called on embedded model fields."""
66
63
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 )
69
66
self .assertEqual (auto_now , auto_now_add )
70
67
# save() updates auto_now but not auto_now_add.
71
68
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 )
73
70
auto_now_two = obj .data .auto_now
74
71
self .assertGreater (auto_now_two , obj .data .auto_now_add )
75
72
# And again, save() updates auto_now but not auto_now_add.
0 commit comments