def test_builder_gets_correct_broker_fee_from_list():
  fee = 'True'

  builder = ListingBuilder(broker_fee=[fee])

  builder._build_broker_fee()

  broker_fee_attr = builder.listing_attrs_output[listing_builder.BROKER_FEE]

  expected_fee = True

  assert broker_fee_attr == expected_fee
def test_builder_gets_correct_broker_fee_from_url_if_not_in_list():
  home_parser_mock = MagicMock(spec=home_parser)

  expected_fee = True
  home_parser_mock.get_broker_fee_from_url = MagicMock(return_value=expected_fee)

  builder = ListingBuilder(home_parser=home_parser_mock)

  builder.listing_attrs_output = MagicMock()
  builder.listing_attrs_output.get.return_value = True

  builder._build_broker_fee()

  assert builder.listing_attrs_output.__setitem__.call_args_list[0] == call(listing_builder.BROKER_FEE,
                                                                            expected_fee)