Test for Pruner

lib/api/app.rb

@@ -6,27 +6,26 @@ require_relative '../exceptions'
 
 module Pruning
   module API
+    # Handles general concers regarding API calls
     class App < Sinatra::Base
       before { content_type :json }
       after  { serialise_response }
       set :show_exceptions, true
-      
 
-      error Pruning::Exceptions::UnexpectedError do 
+      error Pruning::Exceptions::UnexpectedError do
         status 500
       end
 
       error Pruning::Exceptions::OriginCannotFindTheResource do
-        status 404     
+        status 404
       end
 
-
       get '/' do
-        "Try /tree/:name"
+        'Try /tree/:name?indicator_ids[]=32&indicator_ids[]=31'
       end
 
-
       private
+
       def serialise_response
         return unless content_type == 'application/json'
         response.body = [JSON(response.body)]

lib/api/tree.rb

@@ -7,17 +7,15 @@ require_relative '../exceptions'
 
 module Pruning
   module API
+    # Handles all tree related API requests
     class Tree < App
       get '/tree/:name' do
-          tree_repo = Pruning::Repos::Tree.new(RestClient, ENV['TREE_SOURCE_API_HOSTNAME'])
-          complete_tree = tree_repo.get(query.name)
-          pruner = Pruning::Processing::Pruner.new(complete_tree)
-          pruner.prune_tree(query.indicator_ids) 
+        tree_repo = Pruning::Repos::Tree.new(RestClient,
+                                             ENV['TREE_SOURCE_API_HOSTNAME'])
+        complete_tree = tree_repo.get(query.name)
+        pruner = Pruning::Processing::Pruner.new(complete_tree)
+        pruner.prune_tree(query.indicator_ids.to_a)
       end
     end
-  end 
+  end
 end
-
-
-
-