bluechips/tests/model/test_transfer.py

   1 from unittest import TestCase
   2 from bluechips import model
   3
   4 class TestTransfer(TestCase):
   5     def setUp(self):
   6         self.u1 = model.User('chaz', u'Charles Root', False)
   7         self.u2 = model.User('boo', u'Boo Ghost', True)
   8         self.t = model.Transfer(self.u1, self.u2, 1234)
   9
  10     def test_constructor(self):
  11         assert self.t.debtor == self.u1
  12         assert self.t.creditor == self.u2
  13         assert self.t.amount == 1234
  14
  15     def test_repr(self):
  16         assert (repr(self.t) ==
  17                 '<Transfer: from Charles Root to Boo Ghost for 1234>')
  18
  19     def test_involves(self):
  20         other_u = model.User('jim', u'Jimbo James', True)
  21         assert self.t.involves(other_u) == False
  22         assert self.t.involves(self.u1)
  23         assert self.t.involves(self.u2)