Added unit tests for Dns::Server public interface.

2013-07-15 02:12:31 -04:00
parent 3865aab7ee
commit 8270abd2d5
1 changed files with 11 additions and 0 deletions
--- a/test/unit/extensions/tc_dns.rb
+++ b/test/unit/extensions/tc_dns.rb
@@ -42,4 +42,15 @@ class TC_Dns < Test::Unit::TestCase
    assert(@@dns_config.has_key?('port'))
  end

+  # Verifies public interface
+  def test_2_interface
+    @@dns = BeEF::Extension::Dns::Server.instance
+
+    assert(@@dns.respond_to?('run_server'))
+    assert(@@dns.respond_to?('add_rule'))
+    assert(@@dns.respond_to?('remove_rule'))
+    assert(@@dns.respond_to?('get_ruleset'))
+    assert(@@dns.respond_to?('get_rule'))
+  end
+
 end