@@ -191,8 +191,7 @@ def setUp(self):
191
191
class TestSmallFlatDocCreation (SmallFlatDocTest , TestCase ):
192
192
def do_task (self ):
193
193
for doc in self .documents :
194
- model = SmallFlatModel (** doc )
195
- model .save ()
194
+ SmallFlatModel .objects .create (** doc )
196
195
197
196
def after (self ):
198
197
SmallFlatModel .objects .all ().delete ()
@@ -202,8 +201,7 @@ class TestSmallFlatDocUpdate(SmallFlatDocTest, TestCase):
202
201
def setUp (self ):
203
202
super ().setUp ()
204
203
for doc in self .documents :
205
- model = SmallFlatModel (** doc )
206
- model .save ()
204
+ SmallFlatModel .objects .create (** doc )
207
205
self .models = list (SmallFlatModel .objects .all ())
208
206
self .data_size = len (encode ({"field1" : "updated_value0" })) * NUM_DOCS
209
207
self .iteration = 0
@@ -223,8 +221,7 @@ def setUp(self):
223
221
super ().setUp ()
224
222
self .ids = []
225
223
for doc in self .documents :
226
- model = SmallFlatModel (** doc )
227
- model .save ()
224
+ model = SmallFlatModel .objects .create (** doc )
228
225
self .ids .append (model .id )
229
226
230
227
def do_task (self ):
@@ -273,8 +270,7 @@ def setUp(self):
273
270
class TestLargeFlatDocCreation (LargeFlatDocTest , TestCase ):
274
271
def do_task (self ):
275
272
for doc in self .documents :
276
- model = LargeFlatModel (** doc )
277
- model .save ()
273
+ LargeFlatModel .objects .create (** doc )
278
274
279
275
def after (self ):
280
276
LargeFlatModel .objects .all ().delete ()
@@ -284,8 +280,7 @@ class TestLargeFlatDocUpdate(LargeFlatDocTest, TestCase):
284
280
def setUp (self ):
285
281
super ().setUp ()
286
282
for doc in self .documents :
287
- model = LargeFlatModel (** doc )
288
- model .save ()
283
+ LargeFlatModel .objects .create (** doc )
289
284
self .models = list (LargeFlatModel .objects .all ())
290
285
self .data_size = len (encode ({"field1" : "updated_value0" })) * NUM_DOCS
291
286
self .iteration = 0
0 commit comments