Maintain brand reference on series

We need to set the parentRef on series when writing a
series.

Author: Tom McAdam

src/main/java/org/atlasapi/remotesite/five/FiveBrandProcessor.java

@@ -71,7 +71,7 @@ public void processShow(Element element) {
 
         String id = childValue(element, "id");
         EpisodeProcessingNodeFactory nodeFactory 
-            = new EpisodeProcessingNodeFactory(episodeProcessor, brand.getSpecialization());
+            = new EpisodeProcessingNodeFactory(brand, episodeProcessor);
 
         try {
             String responseBody = httpClient.get(getShowUri(id) + WATCHABLES_URL_SUFFIX).body();
@@ -219,18 +219,18 @@ private class EpisodeProcessingNodeFactory extends NodeFactory {
 
         private final FiveEpisodeProcessor episodeProcessor;
         private final List<Item> items = Lists.newArrayList();
-        private final Specialization specialization;
+        private final Brand brand;
 
-        public EpisodeProcessingNodeFactory(FiveEpisodeProcessor episodeProcessor, Specialization specialization) {
-            this.episodeProcessor = episodeProcessor;
-            this.specialization = specialization;
+        public EpisodeProcessingNodeFactory(Brand brand, FiveEpisodeProcessor episodeProcessor) {
+            this.brand = checkNotNull(brand);
+            this.episodeProcessor = checkNotNull(episodeProcessor);
         }
 
         @Override
         public Nodes finishMakingElement(Element element) {
             if (element.getLocalName().equalsIgnoreCase("watchable")) {
                 try {
-                    items.add(episodeProcessor.processEpisode(element, specialization));
+                    items.add(episodeProcessor.processEpisode(element, brand));
                 }
                 catch (Exception e) {
                     log.error("Exception when processing episode", e);

src/main/java/org/atlasapi/remotesite/five/FiveEpisodeProcessor.java

@@ -11,6 +11,7 @@
 import org.atlasapi.media.TransportSubType;
 import org.atlasapi.media.TransportType;
 import org.atlasapi.media.channel.Channel;
+import org.atlasapi.media.entity.Brand;
 import org.atlasapi.media.entity.Broadcast;
 import org.atlasapi.media.entity.Encoding;
 import org.atlasapi.media.entity.Episode;
@@ -58,12 +59,12 @@ public FiveEpisodeProcessor(String baseApiUrl, RemoteSiteClient<HttpResponse> ht
         this.channelMap = channelMap;
     }
 
-    public Item processEpisode(Element element, Specialization specialization) throws Exception {
+    public Item processEpisode(Element element, Brand brand) throws Exception {
 
         String id = childValue(element, "id");
 
         Item item;
-        if(specialization == Specialization.FILM) {
+        if (brand.getSpecialization() == Specialization.FILM) {
             item = new Film(getEpisodeUri(id), getEpisodeCurie(id), Publisher.FIVE);
             item.setMediaType(MediaType.VIDEO);
             item.setSpecialization(Specialization.FILM);
@@ -76,7 +77,7 @@ public Item processEpisode(Element element, Specialization specialization) throw
             if (!Strings.isNullOrEmpty(episodeNumber)) {
                 episode.setEpisodeNumber(Integer.valueOf(episodeNumber));
             }
-            processSeries(episode, element);
+            processSeries(episode, element, brand);
             item = episode;
         }
 
@@ -169,14 +170,15 @@ private Location getLocation(Element element) throws Exception {
     }
 
 
-    private void processSeries(Episode episode, Element element) {
+    private void processSeries(Episode episode, Element element, Brand brand) {
         Element seasonLinkElement = element.getFirstChildElement("season_link");
         if (seasonLinkElement != null) {
             Element seasonElement = seasonLinkElement.getFirstChildElement("season");
             String id = childValue(seasonElement, "id");
             Series series = getSeriesMap().get(id);
             if (series == null){ 
                 series = new Series(seasonLinkElement.getAttributeValue("href"), getSeriesCurie(id), Publisher.FIVE);
+                series.setParent(brand);
                 series.setGenres(genreMap.mapRecognised(ImmutableSet.of(childValue(seasonElement, "genre"))));
 
                 Maybe<String> image = getImage(seasonElement);