def test_one2one_simple(self):
SSN.objects.create(person=self.a, number='1')
merge(self.a, self.b)
# move FKs pointing at A to B
#TODO: assert self.a.ssn is None
assert Person.objects.get(pk=self.b.pk).ssn.number == '1'
assert SSN.objects.count() == 1
def test_many2many_same_class(self):
f = Person.objects.create(name='friend')
self.a.friends.add(f)
merge(self.a, self.b)
assert self.a.friends.all().count() == 0
assert self.b.friends.get().name == f.name
assert f.friends.get().name == self.b.name
def test_many2many_simple(self):
self.g.people.add(self.a)
merge(self.a, self.b)
# A's membership in G has been moved to B
assert self.g.people.get().name == self.b.name
assert self.a.groups.all().count() == 0
assert self.b.groups.all().count() == 1
def test_many2many_redundant(self):
self.g.people.add(self.a)
self.g.people.add(self.b)
merge(self.a, self.b)
# A's membership in G is redundant with B's
assert self.g.people.all().count() == 1
assert self.a.groups.all().count() == 0
assert self.b.groups.all().count() == 1
def test_one2one_conflict_delete(self):
SSN.objects.create(person=self.a, number='1')
SSN.objects.create(person=self.b, number='2')
merge(self.a, self.b, DELETE)
assert Person.objects.get(pk=self.b.pk).ssn.number == '1'
assert SSN.objects.count() == 1
def test_one2one_conflict_keep(self):
SSN.objects.create(person=self.a, number='1')
SSN.objects.create(person=self.b, number='2')
merge(self.a, self.b, KEEP)
assert Person.objects.get(pk=self.a.pk).ssn.number == '1'
assert Person.objects.get(pk=self.b.pk).ssn.number == '2'
def test_fk_existing(self):
Number.objects.create(person=self.a, number='555-1112')
Number.objects.create(person=self.b, number='555-1113')
merge(self.a, self.b)
# everything now on B
assert self.a.numbers.count() == 0
assert self.b.numbers.count() == 2
assert Number.objects.count() == 2
def test_fk_simple(self):
Number.objects.create(person=self.a, number='555-1111')
Number.objects.create(person=self.a, number='555-1112')
merge(self.a, self.b)
# move FKs pointing at A to C
assert self.a.numbers.count() == 0
assert self.b.numbers.count() == 2
assert Number.objects.count() == 2
def test_one2one_conflict(self):
SSN.objects.create(person=self.a, number='1')
SSN.objects.create(person=self.b, number='2')
with self.assertRaises(OneToOneConflict):
merge(self.a, self.b)
def test_same_object(self):
with self.assertRaises(ValueError):
# make sure this doesn't rely on is
merge(self.a, Person.objects.get(name=self.a.name))
def test_same_class(self):
with self.assertRaises(ValueError):
merge(self.a, self.g)