def test_fulfill_contract():
sim = Simulator(dict(), None)
manager = Manager(sim)
producer = Producer("producer_id", manager)
job = Job(est=0, lst=0, id="job_id", load_profile=pd.Series())
consumer = Consumer.start(producers=[producer], job=job, manager=manager)
schedule_object = dict(consumer=consumer, job=job)
producer.schedule.append(schedule_object)
contract = producer.create_contract(job)
consumer.stop()
assert producer.schedule == [schedule_object]
producer.fulfill_contract(contract)
assert producer.schedule == []
def test_create_contract():
sim = Simulator(dict(), None)
manager = Manager(sim)
producer = Producer("producer_id", manager)
job = Job(est=0, lst=0, id="job_id", load_profile=pd.Series())
expected = dict(id='producer_id;job_id;0',
time=0,
time_of_agreement=0,
load_profile=pd.Series(),
job_id='job_id',
producer_id='producer_id')
actual = producer.create_contract(job)
assert expected['id'] == actual['id']
assert expected['time'] == actual['time']
assert expected['time_of_agreement'] == actual['time_of_agreement']
assert expected['job_id'] == actual['job_id']
assert expected['producer_id'] == actual['producer_id']