Fixed #13700 -- Converted m2m_through_regress tests to unittests, removing...

from datetime import datetime

from django.contrib.auth.models import User

from django.core import management

from django.db import models

# Forward declared intermediate model

class Membership(models.Model):

    person = models.ForeignKey('Person')

class B(models.Model):

    b_text = models.CharField(max_length=20)

    a_list = models.ManyToManyField(A, through=Through)

__test__ = {'API_TESTS':"""

# Create some dummy data

>>> bob = Person.objects.create(name='Bob')

>>> jim = Person.objects.create(name='Jim')

>>> rock = Group.objects.create(name='Rock')

>>> roll = Group.objects.create(name='Roll')

>>> frank = User.objects.create_user('frank','frank@example.com','password')

>>> jane = User.objects.create_user('jane','jane@example.com','password')

# Now test that the forward declared Membership works

>>> Membership.objects.create(person=bob, group=rock)

<Membership: Bob is a member of Rock>

>>> Membership.objects.create(person=bob, group=roll)

<Membership: Bob is a member of Roll>

>>> Membership.objects.create(person=jim, group=rock)

<Membership: Jim is a member of Rock>

>>> bob.group_set.all()

[<Group: Rock>, <Group: Roll>]

>>> roll.members.all()

[<Person: Bob>]

# Error messages use the model name, not repr of the class name

>>> bob.group_set = []

Traceback (most recent call last):

...

AttributeError: Cannot set values on a ManyToManyField which specifies an intermediary model.  Use m2m_through_regress.Membership's Manager instead.

>>> roll.members = []

Traceback (most recent call last):

...

AttributeError: Cannot set values on a ManyToManyField which specifies an intermediary model.  Use m2m_through_regress.Membership's Manager instead.

>>> rock.members.create(name='Anne')

Traceback (most recent call last):

...

AttributeError: Cannot use create() on a ManyToManyField which specifies an intermediary model.  Use m2m_through_regress.Membership's Manager instead.

>>> bob.group_set.create(name='Funk')

Traceback (most recent call last):

...

AttributeError: Cannot use create() on a ManyToManyField which specifies an intermediary model.  Use m2m_through_regress.Membership's Manager instead.

# Now test that the intermediate with a relationship outside

# the current app (i.e., UserMembership) workds

>>> UserMembership.objects.create(user=frank, group=rock)

<UserMembership: frank is a user and member of Rock>

>>> UserMembership.objects.create(user=frank, group=roll)

<UserMembership: frank is a user and member of Roll>

>>> UserMembership.objects.create(user=jane, group=rock)

<UserMembership: jane is a user and member of Rock>

>>> frank.group_set.all()

[<Group: Rock>, <Group: Roll>]

>>> roll.user_members.all()

[<User: frank>]

# Regression test for #8134 --

# m2m-through models shouldn't be serialized as m2m fields on the model.

# First, clean up a lot of objects we don't need.

# The serialization test only requires three objects to work -

# one for each end of the m2m, plus the through model.

>>> User.objects.all().delete()

>>> UserMembership.objects.all().delete()

>>> frank.delete()

>>> rock.delete()

>>> jim.delete()

# Dump the current contents of the database as a JSON fixture

>>> management.call_command('dumpdata', 'm2m_through_regress', format='json', indent=2)

[

  {

    "pk": 2,

    "model": "m2m_through_regress.membership",

    "fields": {

      "person": 1,

      "price": 100,

      "group": 2

    }

  },

  {

    "pk": 1,

    "model": "m2m_through_regress.person",

    "fields": {

      "name": "Bob"

    }

  },

  {

    "pk": 2,

    "model": "m2m_through_regress.group",

    "fields": {

      "name": "Roll"

    }

  }

]

# Check the XML serializer too, since it doesn't use the common implementation

>>> management.call_command('dumpdata', 'm2m_through_regress', format='xml', indent=2)

<?xml version="1.0" encoding="utf-8"?>

<django-objects version="1.0">

  <object pk="2" model="m2m_through_regress.membership">

    <field to="m2m_through_regress.person" name="person" rel="ManyToOneRel">1</field>

    <field to="m2m_through_regress.group" name="group" rel="ManyToOneRel">2</field>

    <field type="IntegerField" name="price">100</field>

  </object>

  <object pk="1" model="m2m_through_regress.person">

    <field type="CharField" name="name">Bob</field>

  </object>

  <object pk="2" model="m2m_through_regress.group">

    <field type="CharField" name="name">Roll</field>

  </object>

</django-objects>

## Regression test for #8046:

Check that we don't involve too many copies of the intermediate table when

doing a join.

>>> bob = Person.objects.create(name='Bob')

>>> jim = Person.objects.create(name='Jim')

>>> rock = Group.objects.create(name='Rock')

>>> roll = Group.objects.create(name='Roll')

>>> _ = Membership.objects.create(person=bob, group=rock)

>>> _ = Membership.objects.create(person=jim, group=rock, price=50)

>>> _ = Membership.objects.create(person=bob, group=roll, price=50)

>>> rock.members.filter(membership__price=50)

[<Person: Jim>]

## Regression test for #8254

>>> bob.group_set.filter(membership__price=50)

[<Group: Roll>]

## Regression test for #9804

# Flush the database, just to make sure we can.

>>> management.call_command('flush', verbosity=0, interactive=False)

## Regression test for #11107

Ensure that sequences on m2m_through tables are being created for the through

model, not for a phantom auto-generated m2m table.

>>> management.call_command('loaddata', 'm2m_through', verbosity=0)

>>> management.call_command('dumpdata', 'm2m_through_regress', format='json')

[{"pk": 1, "model": "m2m_through_regress.usermembership", "fields": {"price": 100, "group": 1, "user": 1}}, {"pk": 1, "model": "m2m_through_regress.person", "fields": {"name": "Guido"}}, {"pk": 1, "model": "m2m_through_regress.group", "fields": {"name": "Python Core Group"}}]

"""}

try:

    from cStringIO import StringIO

except ImportError:

    from StringIO import StringIO

from django.core import management

from django.contrib.auth.models import User

from django.test import TestCase

from models import Person, Group, Membership, UserMembership

class M2MThroughTestCase(TestCase):

    def test_everything(self):

        bob = Person.objects.create(name="Bob")

        jim = Person.objects.create(name="Jim")

        rock = Group.objects.create(name="Rock")

        roll = Group.objects.create(name="Roll")

        frank = User.objects.create_user("frank", "frank@example.com", "password")

        jane = User.objects.create_user("jane", "jane@example.com", "password")

        Membership.objects.create(person=bob, group=rock)

        Membership.objects.create(person=bob, group=roll)

        Membership.objects.create(person=jim, group=rock)

        self.assertQuerysetEqual(

            bob.group_set.all(), [

                "<Group: Rock>",

                "<Group: Roll>",

            ]

        )

        self.assertQuerysetEqual(

            roll.members.all(), [

                "<Person: Bob>",

            ]

        )

        self.assertRaises(AttributeError, setattr, bob, "group_set", [])

        self.assertRaises(AttributeError, setattr, roll, "members", [])

        self.assertRaises(AttributeError, rock.members.create, name="Anne")

        self.assertRaises(AttributeError, bob.group_set.create, name="Funk")

        UserMembership.objects.create(user=frank, group=rock)

        UserMembership.objects.create(user=frank, group=roll)

        UserMembership.objects.create(user=jane, group=rock)

        self.assertQuerysetEqual(

            frank.group_set.all(), [

                "<Group: Rock>",

                "<Group: Roll>",

            ]

        )

        self.assertQuerysetEqual(

            roll.user_members.all(), [

                "<User: frank>",

            ]

        )

    def test_serialization(self):

        "m2m-through models aren't serialized as m2m fields. Refs #8134"

        p = Person.objects.create(name="Bob")

        g = Group.objects.create(name="Roll")

        Membership.objects.create(person=p, group=g)

        out = StringIO()

        management.call_command("dumpdata", "m2m_through_regress", format="json", stdout=out)

        self.assertEqual(out.getvalue().strip(), """[{"pk": 1, "model": "m2m_through_regress.membership", "fields": {"person": 1, "price": 100, "group": 1}}, {"pk": 1, "model": "m2m_through_regress.person", "fields": {"name": "Bob"}}, {"pk": 1, "model": "m2m_through_regress.group", "fields": {"name": "Roll"}}]""")

        out = StringIO()

        management.call_command("dumpdata", "m2m_through_regress", format="xml",

            indent=2, stdout=out)

        self.assertEqual(out.getvalue().strip(), """

<?xml version="1.0" encoding="utf-8"?>

<django-objects version="1.0">

  <object pk="1" model="m2m_through_regress.membership">

    <field to="m2m_through_regress.person" name="person" rel="ManyToOneRel">1</field>

    <field to="m2m_through_regress.group" name="group" rel="ManyToOneRel">1</field>

    <field type="IntegerField" name="price">100</field>

  </object>

  <object pk="1" model="m2m_through_regress.person">

    <field type="CharField" name="name">Bob</field>

  </object>

  <object pk="1" model="m2m_through_regress.group">

    <field type="CharField" name="name">Roll</field>

  </object>

</django-objects>

        """.strip())

    def test_join_trimming(self):

        "Check that we don't involve too many copies of the intermediate table when doing a join. Refs #8046, #8254"

        bob  = Person.objects.create(name="Bob")

        jim = Person.objects.create(name="Jim")

        rock = Group.objects.create(name="Rock")

        roll = Group.objects.create(name="Roll")

        Membership.objects.create(person=bob, group=rock)

        Membership.objects.create(person=jim, group=rock, price=50)

        Membership.objects.create(person=bob, group=roll, price=50)

        self.assertQuerysetEqual(

            rock.members.filter(membership__price=50), [

                "<Person: Jim>",

            ]

        )

        self.assertQuerysetEqual(

            bob.group_set.filter(membership__price=50), [

                "<Group: Roll>",

            ]

        )

class ThroughLoadDataTestCase(TestCase):

    fixtures = ["m2m_through"]

    def test_sequence_creation(self):

        "Check that sequences on an m2m_through are created for the through model, not a phantom auto-generated m2m table. Refs #11107"

        out = StringIO()

        management.call_command("dumpdata", "m2m_through_regress", format="json", stdout=out)

        self.assertEqual(out.getvalue().strip(), """[{"pk": 1, "model": "m2m_through_regress.usermembership", "fields": {"price": 100, "group": 1, "user": 1}}, {"pk": 1, "model": "m2m_through_regress.person", "fields": {"name": "Guido"}}, {"pk": 1, "model": "m2m_through_regress.group", "fields": {"name": "Python Core Group"}}]""")