From 8270abd2d5cddfeec2ef9588ac9f3190d105c451 Mon Sep 17 00:00:00 2001 From: soh_cah_toa Date: Mon, 15 Jul 2013 02:12:31 -0400 Subject: [PATCH] Added unit tests for Dns::Server public interface. --- test/unit/extensions/tc_dns.rb | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/test/unit/extensions/tc_dns.rb b/test/unit/extensions/tc_dns.rb index 9c1164e95..5c516f99e 100644 --- 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