def test_club_name_overflow(session):
too_long_name = "blahblah" * 8
too_long_club = mc.Clubs(name=too_long_name,
country=mco.Countries(
name=u"foo",
confederation=enums.ConfederationType.fifa))
with pytest.raises(DataError):
session.add(too_long_club)
session.commit()
def club_data():
england = mco.Countries(name=u"England",
confederation=enums.ConfederationType.europe)
france = mco.Countries(name=u"France",
confederation=enums.ConfederationType.europe)
tz_london = mco.Timezones(name=u"Europe/London",
offset=0.0,
confederation=enums.ConfederationType.europe)
return {
'date':
date(2015, 1, 1),
'competition':
mco.DomesticCompetitions(name=u'Test Competition',
level=1,
country=england),
'season':
mco.Seasons(start_year=mco.Years(yr=2014),
end_year=mco.Years(yr=2015)),
'venue':
mco.Venues(name=u"Emirates Stadium",
city=u"London",
country=england,
timezone=tz_london),
'home_team':
mc.Clubs(name=u"Arsenal FC", country=england),
'away_team':
mc.Clubs(name=u"Lincoln City FC", country=england),
'home_manager':
mcp.Managers(first_name=u"Arsène",
last_name=u"Wenger",
birth_date=date(1949, 10, 22),
country=france),
'away_manager':
mcp.Managers(first_name=u"Gary",
last_name=u"Simpson",
birth_date=date(1961, 4, 11),
country=england),
'referee':
mcp.Referees(first_name=u"Mark",
last_name=u"Clattenburg",
birth_date=date(1975, 3, 13),
country=england)
}
def test_club_insert(session):
club = mc.Clubs(name=u"Arsenal",
country=mco.Countries(
name=u"England",
confederation=enums.ConfederationType.europe))
session.add(club)
result = session.query(mc.Clubs).one()
assert result.name == u"Arsenal"
assert repr(result) == "<Club(name=Arsenal, country=England)>"
def test_club_unicode_insert(session):
club = mc.Clubs(name=u"Фк Спартак Москва",
country=mco.Countries(
name=u"Russia",
confederation=enums.ConfederationType.europe))
session.add(club)
result = session.query(mc.Clubs).join(
mco.Countries).filter(mco.Countries.name == u"Russia").one()
assert result.name == u"Фк Спартак Москва"
assert unicode(result) == u"<Club(name=Фк Спартак Москва, country=Russia)>"
def clubs(self, data_frame):
remote_ids = []
local_ids = []
club_records = []
fields = ['short_name', 'name', 'country_id']
for idx, row in data_frame.iterrows():
club_dict = {field: row[field] for field in fields if row[field]}
if not self.record_exists(mc.Clubs, **club_dict):
club_dict.update(id=uuid.uuid4())
club_records.append(mc.Clubs(**club_dict))
remote_ids.append(row['remote_id'])
local_ids.append(club_dict['id'])
self.session.bulk_save_objects(club_records)
map_records = [mc.ClubMap(id=local_id, remote_id=remote_id, supplier_id=self.supplier_id)
for remote_id, local_id in zip(remote_ids, local_ids) if remote_id]
self.session.bulk_save_objects(map_records)
self.session.commit()