From aee7aed755af07a4e96be1b09ccae53acaf32ec3 Mon Sep 17 00:00:00 2001
From: Brendan Coles <bcoles@gmail.com>
Date: Mon, 11 May 2015 21:52:13 +0000
Subject: [PATCH] Add Network extension test case stub

---
 test/unit/extensions/tc_network.rb | 46 ++++++++++++++++++++++++++++++
 test/unit/ts_unit.rb               |  2 ++
 2 files changed, 48 insertions(+)
 create mode 100644 test/unit/extensions/tc_network.rb

diff --git a/test/unit/extensions/tc_network.rb b/test/unit/extensions/tc_network.rb
new file mode 100644
index 000000000..0416bb518
--- /dev/null
+++ b/test/unit/extensions/tc_network.rb
@@ -0,0 +1,46 @@
+#
+# Copyright (c) 2006-2015 Wade Alcorn - wade@bindshell.net
+# Browser Exploitation Framework (BeEF) - http://beefproject.com
+# See the file 'doc/COPYING' for copying permission
+#
+require 'test/unit'
+
+class TC_Network < Test::Unit::TestCase
+
+  def setup
+    $:.unshift(File.join(File.expand_path(File.dirname(__FILE__)), '.'))
+    $root_dir = File.expand_path('../../../../', __FILE__)
+
+    require 'extensions/network/models/network_service'
+    require 'extensions/network/models/network_host'
+  end
+
+  def shutdown
+    $root_dir = nil
+  end
+
+  # Connects to in-memory database (does not test anything)
+  def test_01_database
+    DataMapper.setup(:default, 'sqlite3::memory:')
+    DataMapper.auto_migrate!
+  end
+
+  # Tests procedure for properly adding new host
+  def test_02_add_host_good
+    assert_nothing_raised do
+      r = BeEF::Core::Models::NetworkHost.new(:hooked_browser_id => '1234', :ip => '127.0.0.1')
+      r.save
+      raise "Adding network host failed" if BeEF::Core::Models::NetworkHost.all(:hooked_browser_id => '1234', :ip => '127.0.0.1').empty?
+    end
+  end
+
+  # Tests procedure for properly adding new service
+  def test_03_add_service_good
+    assert_nothing_raised do
+      r = BeEF::Core::Models::NetworkService.new(:hooked_browser_id => '1234', :proto => 'http', :ip => '127.0.0.1', :port => 80, :type => 'Apache', :cid => 1)
+      r.save
+      raise "Adding network service failed" if BeEF::Core::Models::NetworkService.all(:hooked_browser_id => '1234', :ip => '127.0.0.1').empty?
+    end
+  end
+
+end
diff --git a/test/unit/ts_unit.rb b/test/unit/ts_unit.rb
index a3c2aed72..d108bd572 100644
--- a/test/unit/ts_unit.rb
+++ b/test/unit/ts_unit.rb
@@ -27,6 +27,7 @@ require './extensions/tc_hooks'
 require './extensions/tc_proxy'
 require './extensions/tc_requester'
 require './extensions/tc_event_logger'
+require './extensions/tc_network'
 #require './extensions/tc_dns'
 require './tc_grep'
 require './tc_filesystem'
@@ -54,6 +55,7 @@ class TS_BeefTests
     suite << TC_Proxy.suite
     suite << TC_Hackverter.suite
     suite << TC_EventLogger.suite
+    suite << TC_Network.suite
     suite << TC_Hooks.suite
     suite << TC_Redirector.suite
     #suite << TC_Dns.suite