Updated interface unit tests with new method names.

2014-04-23 21:50:23 -04:00
parent 3ed4098c2f
commit 820ba3a2e7
1 changed files with 2 additions and 3 deletions
--- a/test/unit/extensions/tc_dns.rb
+++ b/test/unit/extensions/tc_dns.rb
@@ -49,12 +49,11 @@ class TC_Dns < Test::Unit::TestCase
  def test_03_interface
    @@dns = BeEF::Extension::Dns::Server.instance

-    assert_respond_to(@@dns, :run_server)
    assert_respond_to(@@dns, :add_rule)
-    assert_respond_to(@@dns, :remove_rule)
    assert_respond_to(@@dns, :get_rule)
+    assert_respond_to(@@dns, :remove_rule!)
    assert_respond_to(@@dns, :get_ruleset)
-    assert_respond_to(@@dns, :remove_ruleset)
+    assert_respond_to(@@dns, :remove_ruleset!)
  end

  # Tests that DNS server runs correctly on desired address and port