def test_update(self):
user = db.User.query().get()
role = db.Role(parent=user.key)
role.type_index = RoleType.COMMITTEE_ADMIN
role.committee = 'EDU'
role.put()
supplier = db.Supplier(id=6)
supplier.name = 'S1'
supplier.put()
self.assertEquals(db.create_key('Supplier-6'), supplier.key)
fund = db.Fund(id=11)
fund.committee = 'EDU'
fund.name = 'myFund'
fund.put()
purchase = db.Purchase(id=12, parent=fund.key)
purchase.description = 'a'
purchase.put()
with app.test_request_context('/p',
method='POST',
data={
'_action': 'update',
'supplier': 'Supplier-6',
'quote_amount-value': '11',
'description': 'b'
}):
purchases.view_purchase('Purchase-12')
self.assertEqual('b', purchase.description)
def test_purchase_list(self):
fund = db.Fund(id=11)
fund.committee = 'EDU'
fund.name = 'myFund'
fund.put()
with app.test_request_context('/', method='GET'):
purchases.view_purchase_list('Fund-11')
def view_fund_list(db_id):
committee = data_models.lookup_committee(db_id)
new_fund = db.Fund(committee=db_id)
new_fund.key = ndb.Key('Fund', None)
model = data_models.Model(new_fund, db_id, db.Fund)
form = FundForm(request.form, new_fund)
model.add_form(ACTION_CREATE.name, form)
property_list = (name_field, code_field)
fund_query = db.Fund.query(db.Fund.committee == db_id).order(
-db.Fund.state_index, db.Fund.name)
return views.view_std_entity_list(model, 'Fund List', ACTION_CREATE,
property_list, fund_query, committee)
def test_view_purchase(self):
fund = db.Fund(id=11)
fund.committee = 'EDU'
fund.name = 'myFund'
fund.put()
purchase = db.Purchase(id=12, parent=fund.key)
purchase.po_number = 'MB006'
purchase.state_index = 1
purchase.quote_amount = db.Money(value=1234)
purchase.description = "desc"
purchase.put()
with app.test_request_context('/', method='GET'):
purchases.view_purchase('Purchase-12')
def test_checked(self):
user = db.User.query().get()
role = db.Role(parent=user.key)
role.type_index = RoleType.FUND_ADMIN
role.put()
fund = db.Fund(id=11)
fund.put()
purchase = db.Purchase(id=12, parent=fund.key)
purchase.put()
with app.test_request_context('/p',
method='POST',
data={'_action': 'checked'}):
purchases.view_purchase('Purchase-12')
self.assertEqual("MB0001", purchase.po_number)
def test_create_purchase(self):
user = db.User.query().get()
role = db.Role(parent=user.key)
role.type_index = RoleType.COMMITTEE_ADMIN
role.committee = 'EDU'
role.put()
fund = db.Fund(id=11)
fund.committee = 'EDU'
fund.name = 'myFund'
fund.put()
supplier = db.Supplier(id=6)
supplier.name = 'S1'
supplier.put()
with app.test_request_context('/',
method='POST',
data={
'_action': 'create',
'supplier': 'Supplier-6',
'quote_amount-value': '11',
'description': 'b'
}):
purchases.view_purchase_list('Fund-11')