diff --git a/scripts/commands/update-guides.js b/scripts/commands/update-guides.js index b66431e3..b2d50819 100644 --- a/scripts/commands/update-guides.js +++ b/scripts/commands/update-guides.js @@ -36,15 +36,16 @@ async function generateMainXML() { async function generateCountries() { logger.info(`Generating countries/...`) - const countryCodes = Object.keys(programs).map(id => channels[id].country) + const filteredChannels = Object.keys(programs).map(id => channels[id]) - for (let code of countryCodes) { + for (let channel of filteredChannels) { + const code = channel.country const output = { channels: [], programs: [] } - output.channels = _.flatten(Object.values(channels)) + output.channels = filteredChannels .filter(c => c.country === code) .map(c => { c.site = sources[c.id] diff --git a/tests/commands/update-guides.test.js b/tests/commands/update-guides.test.js index 31965ba3..91163c74 100644 --- a/tests/commands/update-guides.test.js +++ b/tests/commands/update-guides.test.js @@ -32,7 +32,7 @@ it('can generate epg.xml', () => { expect(output).toBe(expected) }) -fit('can generate /countries guides', () => { +it('can generate /countries guides', () => { const output1 = content('tests/__data__/output/guides/countries/us.epg.xml') const expected1 = content('tests/__data__/expected/guides/countries/us.epg.xml')