Improve test coverage.

model_clone/tests/test_clone_mixin.py

@@ -8,7 +8,7 @@
 from django.utils.text import slugify
 from django.db.utils import IntegrityError
 from django.utils.timezone import make_naive
-from mock import patch, PropertyMock
+from unittest.mock import patch, PropertyMock
 
 from sample.models import (
     Edition,
@@ -22,7 +22,8 @@
     Cover,
     BackCover,
     BookTag,
-    Tag,
+    BookSaleTag,
+    Tag, SaleTag,
 )
 
 User = get_user_model()
@@ -234,6 +235,24 @@ def test_cloning_with_explicit_clone_m2m_fields(
             list(book_clone_2.tags.values_list("name")),
         )
 
+        sale_tag_1 = SaleTag.objects.create(name="test-tag-3")
+        sale_tag_2 = SaleTag.objects.create(name="test-tag-4")
+
+        _clone_m2m_fields_mock.return_value = ["tags"]
+
+        book_3 = Book.objects.create(
+            name="New Book 3", created_by=self.user1, slug=slugify("New Book 3")
+        )
+        BookSaleTag.objects.create(book=book_3, sale_tag=sale_tag_1)
+        BookSaleTag.objects.create(book=book_3, sale_tag=sale_tag_2)
+
+        book_clone_3 = book_3.make_clone()
+
+        self.assertEqual(
+            list(book_3.tags.values_list("name")),
+            list(book_clone_3.tags.values_list("name")),
+        )
+
     @patch("sample.models.Author._clone_excluded_fields", new_callable=PropertyMock)
     def test_cloning_with_clone_excluded_fields(
         self,

sample/migrations/0015_auto_20210423_0935.py

@@ -0,0 +1,43 @@
+# Generated by Django 3.2 on 2021-04-23 09:35
+
+from django.db import migrations, models
+import django.db.models.deletion
+import model_clone.mixins.clone
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('sample', '0014_auto_20210422_1449'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='SaleTag',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('name', models.CharField(max_length=255)),
+            ],
+            options={
+                'abstract': False,
+            },
+            bases=(model_clone.mixins.clone.CloneMixin, models.Model),
+        ),
+        migrations.CreateModel(
+            name='BookSaleTag',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('book', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='sample.book')),
+                ('sale_tag', models.ForeignKey(on_delete=django.db.models.deletion.PROTECT, to='sample.saletag')),
+            ],
+            options={
+                'unique_together': {('book', 'sale_tag')},
+            },
+            bases=(model_clone.mixins.clone.CloneMixin, models.Model),
+        ),
+        migrations.AddField(
+            model_name='book',
+            name='sale_tags',
+            field=models.ManyToManyField(through='sample.BookSaleTag', to='sample.SaleTag'),
+        ),
+    ]

sample/models.py

@@ -42,6 +42,12 @@ def __str__(self):
         return _(self.name)
 
 
+class SaleTag(CloneModel):
+    name = models.CharField(max_length=255)
+
+    def __str__(self):
+        return _(self.name)
+
 class Book(CloneModel):
     name = models.CharField(max_length=2000)
     slug = models.SlugField(unique=True)
@@ -52,6 +58,7 @@ class Book(CloneModel):
     )
     created_at = models.DateTimeField(auto_now_add=True)
     tags = models.ManyToManyField(Tag, through="BookTag")
+    sale_tags = models.ManyToManyField(SaleTag, through="BookSaleTag")
 
     def __str__(self):
         return _(self.name)
@@ -67,6 +74,16 @@ class Meta:
         ]
 
 
+class BookSaleTag(CloneModel):
+    book = models.ForeignKey(Book, on_delete=models.CASCADE)
+    sale_tag = models.ForeignKey(SaleTag, on_delete=models.PROTECT)
+
+    class Meta:
+        unique_together = [
+            ("book", "sale_tag"),
+        ]
+
+
 class Page(CloneModel):
     content = models.CharField(max_length=20000)
     book = models.ForeignKey(Book, on_delete=models.CASCADE)