def test_pop_state_doesnt_fail_if_none_added(self):
c = Context(base_state={
'caller': 'stu',
'signer': 'stu',
'this': 'contract',
'owner': None
})
c._pop_state()
self.assertEqual(c._get_state(), c._base_state)
def test_pop_state_removes_last_state(self):
c = Context(base_state={
'caller': 'stu',
'signer': 'stu',
'this': 'contract',
'owner': None
})
new_state = {
'caller': 'stuart',
'signer': 'stuart',
'this': 'contracts',
'owner': 123
}
c._add_state(new_state)
self.assertEqual(c._get_state(), new_state)
c._pop_state()
self.assertEqual(c._get_state(), c._base_state)