Fix convert_pdb (#21)

Author: raphink

lib/puppet/catalog-diff/compilecatalog.rb

@@ -1,4 +1,6 @@
 require 'puppet/network/http_pool'
+require File.expand_path(File.join(File.dirname(__FILE__), 'preprocessor.rb'))
+
 # Puppet::CatalogDiff
 module Puppet::CatalogDiff
   # Puppet::CatalogDiff::CompileCatalog

lib/puppet/catalog-diff/preprocessor.rb

@@ -47,22 +47,22 @@ def convert25(resource, collector)
         collector << resource
       end
     end
-  end
 
-  # Converts PuppetDB catalogs to our intermediate format
-  def convert_pdb(catalog)
-    catalog = catalog[0]
-    # Fix "data" level in PuppetDB catalog
-    catalog['resources'] = catalog['resources']['data']
-    # Fix edges
-    new_edges = []
-    catalog['edges']['data'].each do |edge|
-      new_edges << {
-        'source' => "#{edge['source_type']}[#{edge['source_title']}]",
-        'target' => "#{edge['target_type']}[#{edge['target_title']}]",
-      }
+    # Converts PuppetDB catalogs to our intermediate format
+    def convert_pdb(catalog)
+      catalog = catalog[0]
+      # Fix "data" level in PuppetDB catalog
+      catalog['resources'] = catalog['resources']['data']
+      # Fix edges
+      new_edges = []
+      catalog['edges']['data'].each do |edge|
+        new_edges << {
+          'source' => "#{edge['source_type']}[#{edge['source_title']}]",
+          'target' => "#{edge['target_type']}[#{edge['target_title']}]",
+        }
+      end
+      catalog['edges'] = new_edges
+      catalog
     end
-    catalog['edges'] = new_edges
-    catalog
   end
 end