fix test case

This commit is contained in:
Jeremy Long
2017-08-31 05:48:34 -04:00
parent 74a2326e0e
commit ff7ebf405c

View File

@@ -117,11 +117,11 @@ public class RubyBundleAuditAnalyzerTest extends BaseDBTestCase {
*/ */
@Test @Test
public void testAnalysis() throws AnalysisException, DatabaseException { public void testAnalysis() throws AnalysisException, DatabaseException {
try { try (Engine engine = new Engine(getSettings())){
analyzer.initialize(null); engine.openDatabase();
analyzer.initialize(engine);
final String resource = "ruby/vulnerable/gems/rails-4.1.15/Gemfile.lock"; final String resource = "ruby/vulnerable/gems/rails-4.1.15/Gemfile.lock";
final Dependency result = new Dependency(BaseTest.getResourceAsFile(this, resource)); final Dependency result = new Dependency(BaseTest.getResourceAsFile(this, resource));
final Engine engine = new Engine(getSettings());
analyzer.analyze(result, engine); analyzer.analyze(result, engine);
int size = engine.getDependencies().size(); int size = engine.getDependencies().size();
assertTrue(size >= 1); assertTrue(size >= 1);
@@ -142,12 +142,12 @@ public class RubyBundleAuditAnalyzerTest extends BaseDBTestCase {
*/ */
@Test @Test
public void testAddCriticalityToVulnerability() throws AnalysisException, DatabaseException { public void testAddCriticalityToVulnerability() throws AnalysisException, DatabaseException {
try { try (Engine engine = new Engine(getSettings())) {
analyzer.initialize(null); engine.openDatabase();
analyzer.initialize(engine);
final Dependency result = new Dependency(BaseTest.getResourceAsFile(this, final Dependency result = new Dependency(BaseTest.getResourceAsFile(this,
"ruby/vulnerable/gems/sinatra/Gemfile.lock")); "ruby/vulnerable/gems/sinatra/Gemfile.lock"));
final Engine engine = new Engine(getSettings());
analyzer.analyze(result, engine); analyzer.analyze(result, engine);
Dependency dependency = engine.getDependencies().get(0); Dependency dependency = engine.getDependencies().get(0);