test_distro_contract.py

# absolute minimal contract test case

import boa

from eth_account import Account

# static test data of a product build
git_sha="8a645f5782b507202c75ee7fbeaf7bb21d34dd5c2eda4118bb76a31a39226e30"
primary_sha="562a8a6891053541a9cb4d6b252e0cdde4b9da98fc99e87aaeac3ecccb05b91755a2ffcdfd3cdd3e3a55fc5deb96157be4565f4e1c4eb177f7b08075a15e2b70"
product_name="example-1"
build_kind=1

# our roles
foundation_owner = Account.create('KEYSMASH FJAFJKLDSKF7JKFDJ 1530')
attestator = Account.create('KEYSMASH AJFFJKLDSKF7JKFDJ 1531')
security_team = Account.create('KEYSMASH AAAFJKLDSKF7JKFDJ 1532')
product_creator = Account.create('KEYSMASH BBBFJKLDSKF7JKFDJ 1533')
random_guy = Account.create('KEYSMASH FFFFJKLDSKF7JKFDJ 1534')

# deploy our contract
boa.env.eoa = foundation_owner.address
contract = boa.load('ape/contracts/distro.vy', product_creator.address, attestator.address, security_team.address)

#
# Register a product build
#
boa.env.eoa = product_creator.address
print("Setting balance...")
boa.env.set_balance(boa.env.eoa, 1000 * 10**18)
product_id = contract.add_product(product_name, git_sha)
print(f"Created Product with ID {product_id}")
contract.add_product_build(git_sha, build_kind, primary_sha)
print(f"Created Product Build")

#
# The verification tool would do the following
#
boa.env.eoa = random_guy.address
print("Setting balance...")
product_build = contract.get_product_build(primary_sha)
if product_build[0] != 1 or product_build[1] != 1:
    print("Product Build not found!")
    exit(1)
print("Found Product Build")
product = contract.get_product(product_build[0])
if product[0] != product_name or product[1] != git_sha:
    print("Product Build not found!")
    exit(1)
print("Found Product")
verification = contract.current_product_build(product_name, build_kind)
if verification != primary_sha:
    print("Is not current verification!")
    exit(1)
print("Product is verified to be current")

# security team sets the warn flag
boa.env.eoa = security_team.address
contract.set_critical(product_build[0], True)

# Attestator approves
boa.env.eoa = attestator.address
product_build = contract.get_product_build(primary_sha)
if product_build[2] != 1:
    print("Attestation is not outstanding")
    exit(1)
contract.reject_attestation(primary_sha)
product_build = contract.get_product_build(primary_sha)
if product_build[2] != 4:
    print("Attestation is not rejected")
    exit(1)
contract.approve_attestation(primary_sha)
product_build = contract.get_product_build(primary_sha)
if product_build[2] != 2:
    print("Attestation is not approved")
    exit(1)

# test address changes, take away permissions for everyone
boa.env.eoa = foundation_owner.address
contract.set_product_creator(boa.env.eoa)
contract.set_official_validator(boa.env.eoa)
contract.set_security_team(boa.env.eoa)

### FIXME: add validations that functions are not working when not permitted

print("  SUCCESS :)  ")