Skip to content

Commit 1a9f5c5

Browse files
committed
refactor tests
1 parent 09c1d69 commit 1a9f5c5

File tree

2 files changed

+88
-27
lines changed

2 files changed

+88
-27
lines changed

tests/schema_/models.py

Lines changed: 0 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -15,33 +15,19 @@ class Address(models.Model):
1515
state = models.CharField(max_length=2)
1616
zip_code = models.IntegerField(db_index=True)
1717
uid = models.IntegerField(unique=True)
18-
unique_together_one = models.CharField(max_length=10)
19-
unique_together_two = models.CharField(max_length=10)
20-
indexed_by_index_one = models.CharField(max_length=10)
21-
unique_constraint_one = models.CharField(max_length=10)
2218

2319
class Meta:
2420
apps = new_apps
25-
constraints = [models.UniqueConstraint(fields=["unique_constraint_one"], name="unique_one")]
26-
indexes = [models.Index(fields=["indexed_by_index_one"])]
27-
unique_together = [("unique_together_one", "unique_together_two")]
2821

2922

3023
class Author(models.Model):
3124
name = models.CharField(max_length=10)
3225
age = models.IntegerField(db_index=True)
3326
address = EmbeddedModelField(Address)
3427
employee_id = models.IntegerField(unique=True)
35-
unique_together_three = models.CharField(max_length=10)
36-
unique_together_four = models.CharField(max_length=10)
37-
indexed_by_index_two = models.CharField(max_length=10)
38-
unique_constraint_two = models.CharField(max_length=10)
3928

4029
class Meta:
4130
apps = new_apps
42-
constraints = [models.UniqueConstraint(fields=["unique_constraint_two"], name="unique_two")]
43-
indexes = [models.Index(fields=["indexed_by_index_two"])]
44-
unique_together = [("unique_together_three", "unique_together_four")]
4531

4632

4733
class Book(models.Model):

tests/schema_/test_embedded_model.py

Lines changed: 88 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -213,7 +213,7 @@ class Address(models.Model):
213213
index_together_two = models.CharField(max_length=10)
214214

215215
class Meta:
216-
app_label = "schema"
216+
app_label = "schema_"
217217
index_together = [("index_together_one", "index_together_two")]
218218

219219
class Author(models.Model):
@@ -222,14 +222,14 @@ class Author(models.Model):
222222
index_together_four = models.CharField(max_length=10)
223223

224224
class Meta:
225-
app_label = "schema"
225+
app_label = "schema_"
226226
index_together = [("index_together_three", "index_together_four")]
227227

228228
class Book(models.Model):
229229
author = EmbeddedModelField(Author)
230230

231231
class Meta:
232-
app_label = "schema"
232+
app_label = "schema_"
233233

234234
with connection.schema_editor() as editor:
235235
editor.create_model(Book)
@@ -239,20 +239,45 @@ class Meta:
239239
self.get_constraints_for_columns(
240240
Book, ["author.address.index_together_one", "author.address.index_together_two"]
241241
),
242-
["schema_addr_index_t_a0305a_idx"],
242+
["schema__add_index_t_efa93e_idx"],
243243
)
244244
self.assertEqual(
245245
self.get_constraints_for_columns(
246246
Book,
247247
["author.index_together_three", "author.index_together_four"],
248248
),
249-
["schema_auth_index_t_65817b_idx"],
249+
["schema__aut_index_t_df32aa_idx"],
250250
)
251-
editor.delete_model(Author)
252-
self.assertTableNotExists(Author)
251+
editor.delete_model(Book)
252+
self.assertTableNotExists(Book)
253253

254+
@isolate_apps("schema_")
254255
def test_unique_together(self):
255256
"""Meta.unique_together on an embedded model."""
257+
258+
class Address(models.Model):
259+
unique_together_one = models.CharField(max_length=10)
260+
unique_together_two = models.CharField(max_length=10)
261+
262+
class Meta:
263+
app_label = "schema_"
264+
unique_together = [("unique_together_one", "unique_together_two")]
265+
266+
class Author(models.Model):
267+
address = EmbeddedModelField(Address)
268+
unique_together_three = models.CharField(max_length=10)
269+
unique_together_four = models.CharField(max_length=10)
270+
271+
class Meta:
272+
app_label = "schema_"
273+
unique_together = [("unique_together_three", "unique_together_four")]
274+
275+
class Book(models.Model):
276+
author = EmbeddedModelField(Author)
277+
278+
class Meta:
279+
app_label = "schema_"
280+
256281
with connection.schema_editor() as editor:
257282
editor.create_model(Book)
258283
self.assertTableExists(Book)
@@ -274,31 +299,81 @@ def test_unique_together(self):
274299
"schema__address_author.address.unique_together_one_author.address.unique_together_two_de682e30_uniq"
275300
],
276301
)
277-
editor.delete_model(Author)
278-
self.assertTableNotExists(Author)
302+
editor.delete_model(Book)
303+
self.assertTableNotExists(Book)
279304

305+
@isolate_apps("schema_")
280306
def test_indexes(self):
281307
"""Meta.indexes on an embedded model."""
308+
309+
class Address(models.Model):
310+
indexed_one = models.CharField(max_length=10)
311+
312+
class Meta:
313+
app_label = "schema_"
314+
indexes = [models.Index(fields=["indexed_one"])]
315+
316+
class Author(models.Model):
317+
address = EmbeddedModelField(Address)
318+
indexed_two = models.CharField(max_length=10)
319+
320+
class Meta:
321+
app_label = "schema_"
322+
indexes = [models.Index(fields=["indexed_two"])]
323+
324+
class Book(models.Model):
325+
author = EmbeddedModelField(Author)
326+
327+
class Meta:
328+
app_label = "schema_"
329+
282330
with connection.schema_editor() as editor:
283331
editor.create_model(Book)
284332
self.assertTableExists(Book)
285333
# Embedded uniques are created.
286334
self.assertEqual(
287-
self.get_constraints_for_columns(Book, ["author.indexed_by_index_two"]),
288-
["schema__aut_indexed_7e3a5c_idx"],
335+
self.get_constraints_for_columns(Book, ["author.indexed_two"]),
336+
["schema__aut_indexed_b19137_idx"],
289337
)
290338
self.assertEqual(
291339
self.get_constraints_for_columns(
292340
Book,
293-
["author.address.indexed_by_index_one"],
341+
["author.address.indexed_one"],
294342
),
295-
["schema__add_indexed_ef5dd6_idx"],
343+
["schema__add_indexed_b64972_idx"],
296344
)
297345
editor.delete_model(Author)
298346
self.assertTableNotExists(Author)
299347

348+
@isolate_apps("schema_")
300349
def test_constraints(self):
301350
"""Meta.constraints on an embedded model."""
351+
352+
class Address(models.Model):
353+
unique_constraint_one = models.CharField(max_length=10)
354+
355+
class Meta:
356+
app_label = "schema_"
357+
constraints = [
358+
models.UniqueConstraint(fields=["unique_constraint_one"], name="unique_one")
359+
]
360+
361+
class Author(models.Model):
362+
address = EmbeddedModelField(Address)
363+
unique_constraint_two = models.CharField(max_length=10)
364+
365+
class Meta:
366+
app_label = "schema_"
367+
constraints = [
368+
models.UniqueConstraint(fields=["unique_constraint_two"], name="unique_two")
369+
]
370+
371+
class Book(models.Model):
372+
author = EmbeddedModelField(Author)
373+
374+
class Meta:
375+
app_label = "schema_"
376+
302377
with connection.schema_editor() as editor:
303378
editor.create_model(Book)
304379
self.assertTableExists(Book)

0 commit comments

Comments
 (0)