def test_get_base_event_is_none_when_base_event_missing():
# A financing statement without a base event is technically in valid data, but it should not cause an error to occur
base_reg_num = '123456A'
model = FinancingStatement(
registration_number=base_reg_num,
events=[
FinancingStatementEvent(registration_number='123457B',
base_registration_number=base_reg_num),
FinancingStatementEvent(registration_number='123459C',
base_registration_number=base_reg_num),
FinancingStatementEvent(registration_number='123458D',
base_registration_number=base_reg_num)
])
event = model.get_base_event()
assert event is None
def test_get_base_event_should_get_record_with_same_registration_number():
base_reg_num = '123456A'
model = FinancingStatement(
registration_number=base_reg_num,
events=[
FinancingStatementEvent(registration_number='123457B',
base_registration_number=base_reg_num),
FinancingStatementEvent(registration_number='123459C',
base_registration_number=base_reg_num),
FinancingStatementEvent(registration_number=base_reg_num,
base_registration_number=base_reg_num),
FinancingStatementEvent(registration_number='123458D',
base_registration_number=base_reg_num)
])
event = model.get_base_event()
assert event.registration_number == base_reg_num