openvas-cli 0.1.1 → 0.2.0

data/spec/openvas-cli/vas_report_spec.rb

@@ -1,75 +1,77 @@
 require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
 require 'nokogiri'
 
-describe "VasReport" do
-  it "should pull all available reports" do
-    repts = VasReport.get_all
+module OpenvasCli
+  describe VasReport do
+    it "should pull all available reports" do
+      repts = VasReport.get_all
+      
+      repts.should_not be nil
+      
+      repts.each { |r| r.should be_valid }
+    end
     
-    repts.should_not be nil
+    it 'should pull a raw XML dump for a single report' do
+      rept = VasReport.get_all[0]
+      
+      xml = rept.to_xml
+      
+      xdoc = Nokogiri::XML(xml) #it should be a valid xml document
+      
+      xdoc.at_xpath("//report/@id").value.should == rept.report_id
+    end
     
-    repts.each { |r| r.should be_valid }
-  end
-  
-  it 'should pull a raw XML dump for a single report' do
-    rept = VasReport.get_all[0]
-    
-    xml = rept.to_xml
-    
-    xdoc = Nokogiri::XML(xml) #it should be a valid xml document
-    
-    xdoc.at_xpath("//report/@id").value.should == rept.report_id
-  end
-  
-  it 'should pull a specific report by id' do
-    all = VasReport.get_all
-    
-    o_rept = all[rand(all.count)]
-    o_rept.should_not be nil
-    n_rept = VasReport.get_all(:report_id => o_rept.report_id)[0]
-    n_rept.should_not be nil
-    
-    o_rept.report_id.should == n_rept.report_id
-  end
-  
-  it 'should report the associated results' do
-    all = VasReport.get_all
+    it 'should pull a specific report by id' do
+      all = VasReport.get_all
+      
+      o_rept = all[rand(all.count)]
+      o_rept.should_not be nil
+      n_rept = VasReport.get_all(:report_id => o_rept.report_id)[0]
+      n_rept.should_not be nil
+      
+      o_rept.report_id.should == n_rept.report_id
+    end
     
-    all.each { |r|
-      r.result_count[:filtered].should == r.results.count
-    }
-  end
-  
-  it 'should filter the results' do
-    all = VasReport.get_all(:filter => [:high])
-    
-    all.each { |report|
-      report.results.each{ |result|
-        result.threat.should =~ /high/i
+    it 'should report the associated results' do
+      all = VasReport.get_all
+      
+      all.each { |r|
+        r.result_count[:filtered].should == r.results.count
       }
-    }
+    end
     
-    all = VasReport.get_all(:filter => [:log])
-    
-    all.each { |report|
-      report.results.each{ |result|
-        result.threat.should =~ /log/i
+    it 'should filter the results' do
+      all = VasReport.get_all(:filter => [:high])
+      
+      all.each { |report|
+        report.results.each{ |result|
+          result.threat.should =~ /high/i
+        }
       }
-    }
-    
-    all = VasReport.get_all(:filter => [:medium, :debug, :low])
-    
-    all.each { |report|
-      report.results.each{ |result|
-        result.threat.should =~ /(medium)|(debug)|(low)/i
+      
+      all = VasReport.get_all(:filter => [:log])
+      
+      all.each { |report|
+        report.results.each{ |result|
+          result.threat.should =~ /log/i
+        }
       }
-    }
-  end
-  
-  it 'should return an empty set when passed a bad report_id' do
-    lambda {
-      r = VasReport.get_all(:report_id => '0000000000')
-      r.should_not be nil
-      r.should be_empty
-    }.should_not raise_error
+      
+      all = VasReport.get_all(:filter => [:medium, :debug, :low])
+      
+      all.each { |report|
+        report.results.each{ |result|
+          result.threat.should =~ /(medium)|(debug)|(low)/i
+        }
+      }
+    end
+    
+    it 'should return an empty set when passed a bad report_id' do
+      lambda {
+        r = VasReport.get_all(:report_id => '0000000000')
+        r.should_not be nil
+        r.should be_empty
+      }.should_not raise_error
+    end
   end
 end

data/spec/openvas-cli/vas_result_spec.rb

@@ -1,109 +1,110 @@
 require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
 
-describe 'VasResult' do
-  it 'should pull all results' do
-    all_results = VasResult.get_all 
-    
-    all_results.should_not be nil
-    
-    all_results.count.should > 0
-    
-    all_results.each { |r| r.should be_valid }
-  end
-  
-  it 'should be able to paginate results', :slow=>true do
-    result_count = VasResult.get_all.count
-    
-    start = 0
-    num = 20
-    
-    while start < result_count do
-      results = VasResult.get_all(:start => start, :count => num)
+module OpenvasCli
+  describe VasResult do
+    it 'should pull all results' do
+      all_results = VasResult.get_all 
       
-      results.count.should <= 20
+      all_results.should_not be nil
       
-      start += results.count
+      all_results.each { |r| r.should be_valid }
     end
-  end
-  
-  it 'should sort by threat by default' do
-    results = VasResult.get_all
     
-    last_val = 9
-    results.each { |r|
-      r.threat_level.should <= last_val
-      last_val = r.threat_level
-    }
-  end
-  
-  it 'should sort the results by ID' do
-    results = VasResult.get_all(:sort_by => :result_id)
+    it 'should be able to paginate results', :slow=>true do
+      result_count = VasResult.get_all.count
+      
+      start = 0
+      num = 20
+      
+      while start < result_count do
+        results = VasResult.get_all(:start => start, :count => num)
+        
+        results.count.should <= 20
+        
+        start += results.count
+      end
+    end
     
-    last_id = ""
-    results.each { |r|
-      r.result_id.should >= last_id
-      last_id = r.result_id
-    }
-  end
-  
-  it 'should sort the results by host' do
-    results = VasResult.get_all(:sort_by => :host)
+    it 'should sort by threat by default' do
+      results = VasResult.get_all
+      
+      last_val = 9
+      results.each { |r|
+        r.threat_level.should <= last_val
+        last_val = r.threat_level
+      }
+    end
     
-    last_host = ""
-    results.each { |r|
-      r.host.should >= last_host
-      last_host = r.host
-    }
-  end
-  
-  it 'should sort the results by rule_id' do
-    results = VasResult.get_all(:sort_by => :rule_id)
+    it 'should sort the results by ID' do
+      results = VasResult.get_all(:sort_by => :result_id)
+      
+      last_id = ""
+      results.each { |r|
+        r.result_id.should >= last_id
+        last_id = r.result_id
+      }
+    end
     
-    last_val = ""
-    results.each { |r|
-      r.rule_id.should >= last_val
-      last_val = r.rule_id
-    }
-  end
-  
-  it 'should sort the results by subnet' do
-    results = VasResult.get_all(:sort_by => :subnet)
+    it 'should sort the results by host' do
+      results = VasResult.get_all(:sort_by => :host)
+      
+      last_host = ""
+      results.each { |r|
+        r.host.should >= last_host
+        last_host = r.host
+      }
+    end
     
-    last_val = ""
-    results.each { |r|
-      r.subnet.should >= last_val
-      last_val = r.subnet
-    }
-  end
-  
-  it 'should sort the results by threat' do
-    results = VasResult.get_all(:sort_by => :threat)
+    it 'should sort the results by rule_id' do
+      results = VasResult.get_all(:sort_by => :rule_id)
+      
+      last_val = ""
+      results.each { |r|
+        r.rule_id.should >= last_val
+        last_val = r.rule_id
+      }
+    end
     
-    last_val = 9
-    results.each { |r|
-      r.threat_level.should <= last_val
-      last_val = r.threat_level
-    }
-  end
-  
-  it 'shoudl pull results for a specified task' do
-    tasks = VasTask.get_all
+    it 'should sort the results by subnet' do
+      results = VasResult.get_all(:sort_by => :subnet)
+      
+      last_val = ""
+      results.each { |r|
+        r.subnet.should >= last_val
+        last_val = r.subnet
+      }
+    end
     
-    tasks.count.should > 0
-    task_id = tasks[0].task_id
-    results = VasResult.get_all(:task_id => task_id)
-    results.each { |r| r.task_id.should == task_id }
+    it 'should sort the results by threat' do
+      results = VasResult.get_all(:sort_by => :threat)
+      
+      last_val = 9
+      results.each { |r|
+        r.threat_level.should <= last_val
+        last_val = r.threat_level
+      }
+    end
     
-    all_tasks = VasResult.get_all
+    it 'shoudl pull results for a specified task' do
+      tasks = VasTask.get_all
+      
+      unless tasks.empty?
+        task_id = tasks[0].task_id
+        results = VasResult.get_all(:task_id => task_id)
+        results.each { |r| r.task_id.should == task_id }
+        
+        all_tasks = VasResult.get_all
+        
+        all_tasks.count.should >= results.count
+      end
+    end
     
-    all_tasks.count.should >= results.count
-  end
-  
-  it 'should filter based off of threat level' do
-    results = VasResult.get_all(:filter=>[:high, :medium])
-    threats = ["High", "Medium"]
-    results.each { |r|
-      threats.should include r.threat
-    }
+    it 'should filter based off of threat level' do
+      results = VasResult.get_all(:filter=>[:high, :medium])
+      threats = ["High", "Medium"]
+      results.each { |r|
+        threats.should include r.threat
+      }
+    end
   end
 end

data/spec/openvas-cli/vas_schedule_spec.rb

@@ -1,61 +1,158 @@
 require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
 
-describe 'VasSchedule' do
-  it 'should pull all available schedules' do
-    schedules = VasSchedule.get_all
-    schedules.should_not be nil
-    schedules.count.should > 0
-    schedules.each{ |s| s.should be_valid }
-  end
-  
-  it 'should pull a schedule by ID' do  
-    all = VasSchedule.get_all
-    all.should_not be nil
-    all.count.should > 0
-    
-    schedule_id = all[rand(all.count)].schedule_id
-    schedules = VasSchedule.get_all(:schedule_id => schedule_id)
-    schedules.count.should == 1
-    schedules[0].schedule_id.should == schedule_id
-  end
-  
-  it 'default sort should be by name' do
-    all = VasSchedule.get_all
-    
-    old_val = ""
-    all.each { |s|
-      s.name.should >= old_val
-      old_val = s.name
-    }
-  end
-  
-  it 'should sort by schedule_id' do
-    all = VasSchedule.get_all(:sort_by=>:schedule_id)
-    
-    old_val = ""
-    all.each { |s|
-      s.schedule_id.should >= old_val
-      old_val = s.schedule_id
-    }
-  end
-  
-  it 'should sort_by first_time' do
-    all = VasSchedule.get_all(:sort_by=>:first_time)
-    
-    old_val = Time.at(0)
-    all.each { |s|
-      s.first_time.should >= old_val
-      old_val = s.first_time
-    }
-  end
-  
-  it 'should sort_by next_time' do
-    all = VasSchedule.get_all(:sort_by=>:next_time)
-    
-    old_val = Time.at(0)
-    all.each { |s|
-      s.next_time.should >= old_val
-      old_val = s.next_time
-    }
+module OpenvasCli
+  describe VasSchedule do
+    def valid_params
+      {
+        :name => "Test_#{Time.now.strftime('%Y-%m-%d_%H:%M:%S')}", 
+        :comment => "Test Schedule [Automatically Generated]", 
+        :first_time => Time.now + 86400,
+        :period => VasPeriod.new(:number => 1, :period => :day)
+      }
+    end
+    
+    def get_t_string(t)
+      t.strftime("%Y-%m-%d_%H:%M")
+    end
+    
+    after(:each) do
+      all = VasSchedule.get_all
+      all.each { |s| s.destroy! if s.name =~ /Test_.*/i }
+    end
+
+    it 'should pull all available schedules' do
+      schedules = VasSchedule.get_all
+      schedules.should_not be nil
+      schedules.each{ |s| s.should be_valid }
+    end
+    
+    it 'should pull a schedule by ID' do  
+      all = VasSchedule.get_all
+      all.should_not be nil
+
+      unless all.empty?    
+        schedule_id = all.choice.id
+        schedules = VasSchedule.get_all(:id => schedule_id)
+        schedules.count.should == 1
+        schedules[0].id.should == schedule_id
+      end
+    end
+    
+    it 'default sort should be by name' do
+      all = VasSchedule.get_all
+      
+      old_val = ""
+      all.each { |s|
+        s.name.should >= old_val
+        old_val = s.name
+      }
+    end
+    
+    it 'should sort by schedule_id' do
+      all = VasSchedule.get_all(:sort_by=>:schedule_id)
+      
+      old_val = ""
+      all.each { |s|
+        s.id.should >= old_val
+        old_val = s.id
+      }
+    end
+    
+    it 'should sort_by first_time' do
+      all = VasSchedule.get_all(:sort_by=>:first_time)
+      
+      old_val = Time.at(0)
+      all.each { |s|
+        s.first_time.should >= old_val
+        old_val = s.first_time
+      }
+    end
+    
+    it 'should sort_by next_time' do
+      all = VasSchedule.get_all(:sort_by=>:next_time)
+      
+      old_val = Time.at(0)
+      all.each { |s|
+        s.next_time.should >= old_val
+        old_val = s.next_time
+      }
+    end
+    
+    it 'should create a new schedule' do
+      s = VasSchedule.new(valid_params)
+      s.should be_valid
+      s.save!
+      
+      s.should_not be_changed
+      s.id.should_not be nil
+      
+      new_s = VasSchedule.get_all(:id => s.id)[0]
+      new_s.should_not be nil
+      new_s.id.should            == s.id
+      new_s.name.should          == s.name
+      new_s.comment.should       == s.comment
+      get_t_string(new_s.first_time).should    == get_t_string(s.first_time)
+      new_s.period.should        == s.period
+    end
+      
+    it 'should modify an existing schedule' do
+      s = VasSchedule.new(valid_params)
+      s.save!
+      
+      old_id = s.id
+      
+      s.comment = "NEW COMMENT"
+      s.save!
+      
+      s.id.should_not == old_id
+      
+      VasSchedule.get_all(:id => old_id).should be_empty
+      
+      new_s = VasSchedule.get_all(:id => s.id)[0]
+      new_s.should_not be nil
+      new_s.id.should            == s.id
+      new_s.name.should          == s.name
+      new_s.comment.should       == s.comment
+      get_t_string(new_s.first_time).should    == get_t_string(s.first_time)
+      new_s.period.should        == s.period
+    end
+    
+    it 'should report changed when name is changed' do
+      s = VasSchedule.new(valid_params)
+      s.should_not be_changed
+      
+      s.name = "NEW NAME"
+      s.should be_changed
+    end
+    
+    it 'should report changed when comment is changed' do
+      s = VasSchedule.new(valid_params)
+      s.should_not be_changed
+      
+      s.comment = "NEW COMMENT"
+      s.should be_changed
+    end
+    
+    it 'should report changed when first_time is changed' do
+      s = VasSchedule.new(valid_params)
+      s.should_not be_changed
+      
+      s.first_time = s.first_time
+      s.should_not be_changed
+      
+      s.first_time = Time.now + 30
+      s.should be_changed
+    end
+    
+    it 'should report changed when period is changed' do
+      s = VasSchedule.new(valid_params)
+      s.should_not be_changed
+      
+      s.period = s.period
+      s.should_not be_changed
+      
+      s.period = s.period + VasPeriod.new(:number => 2, :period => :day)
+      s.should be_changed
+    end
   end
 end