def test_add_money_and_save(self):
money = Money("10.45")
expected = Money("10.76")
model = ModelWithVanillaMoneyField(money=money)
model.money = money + 3 % money
model.save()
self.assertEquals(expected, model.money)
def test_find_models_related_to_money_models(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 test_exact_match(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 test_currency_search(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 test_range_search(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 test_saving(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)