def testFindModelsRelatedToMoneyModels(self):
moneyModel = ModelWithVanillaMoneyField(money=Money("100.0", moneyed.ZWN))
moneyModel.save()
relatedModel = ModelRelatedToModelWithMoney(moneyModel=moneyModel)
relatedModel.save()
ModelRelatedToModelWithMoney.objects.get(moneyModel__money=Money("100.0", moneyed.ZWN))
ModelRelatedToModelWithMoney.objects.get(moneyModel__money__lt=Money("1000.0", moneyed.ZWN))
def testExactMatch(self):
somemoney = Money("100.0")
model = ModelWithVanillaMoneyField()
model.money = somemoney
model.save()
retrieved = ModelWithVanillaMoneyField.objects.get(money=somemoney)
self.assertEquals(model.pk, retrieved.pk)
def testCurrencySearch(self):
otherMoney = Money("1000", moneyed.USD)
correctMoney = Money("1000", moneyed.ZWN)
model = ModelWithVanillaMoneyField(money=Money("100.0", moneyed.ZWN))
model.save()
shouldBeEmpty = ModelWithVanillaMoneyField.objects.filter(money__lt=otherMoney)
self.assertEquals(shouldBeEmpty.count(), 0)
shouldBeOne = ModelWithVanillaMoneyField.objects.filter(money__lt=correctMoney)
self.assertEquals(shouldBeOne.count(), 1)
def testRangeSearch(self):
minMoney = Money("3")
model = ModelWithVanillaMoneyField(money=Money("100.0"))
model.save()
retrieved = ModelWithVanillaMoneyField.objects.get(money__gt=minMoney)
self.assertEquals(model.pk, retrieved.pk)
shouldBeEmpty = ModelWithVanillaMoneyField.objects.filter(money__lt=minMoney)
self.assertEquals(shouldBeEmpty.count(), 0)
def testSaving(self):
somemoney = Money("100.0")
model = ModelWithVanillaMoneyField(money=somemoney)
model.save()
retrieved = ModelWithVanillaMoneyField.objects.get(pk=model.pk)
self.assertEquals(somemoney.currency, retrieved.money.currency)
self.assertEquals(somemoney, retrieved.money)
# Try setting the value directly
retrieved.money = Money(1, moneyed.DKK)
retrieved.save()
retrieved = ModelWithVanillaMoneyField.objects.get(pk=model.pk)
self.assertEquals(Money(1, moneyed.DKK), retrieved.money)
