|
require 'test_helper'
|
|
require 'tftp/tftp_plugin'
|
|
require "tftp/server"
|
|
|
|
class TftpTest < Test::Unit::TestCase
|
|
def setup
|
|
@tftp = Proxy::TFTP::Server.new
|
|
Proxy::TFTP::Plugin.load_test_settings(:tftproot => "/some/root")
|
|
end
|
|
|
|
def test_should_have_a_logger
|
|
assert_respond_to @tftp, :logger
|
|
end
|
|
|
|
def test_path_to_tftp_directory_without_tftproot_setting
|
|
Proxy::TFTP::Plugin.load_test_settings({})
|
|
assert_equal "/var/lib/tftpboot", @tftp.send(:path)
|
|
end
|
|
|
|
def test_path_to_tftp_directory_with_tftproot_setting
|
|
assert_equal "/some/root", @tftp.send(:path)
|
|
end
|
|
|
|
def test_path_to_tftp_directory_with_relative_tftproot_setting
|
|
Proxy::TFTP::Plugin.load_test_settings(:tftproot => "./some/root")
|
|
assert_equal Pathname.new(__FILE__).join("..", "..", "..", "modules","tftp","some","root").to_s, @tftp.send(:path)
|
|
end
|
|
|
|
def test_paths_inside_tftp_directory_dont_raise_errors
|
|
::Proxy::HttpDownload.any_instance.stubs(:start).returns(true)
|
|
FileUtils.stubs(:mkdir_p).returns(true)
|
|
|
|
assert Proxy::TFTP.send(:fetch_boot_file,'/some/root/boot/file','http://localhost/file')
|
|
end
|
|
|
|
def test_paths_outside_tftp_directory_raise_errors
|
|
::Proxy::HttpDownload.any_instance.stubs(:start).returns(true)
|
|
FileUtils.stubs(:mkdir_p).returns(true)
|
|
|
|
assert_raises RuntimeError do
|
|
Proxy::TFTP.send(:fetch_boot_file,'/other/root/boot/file','http://localhost/file')
|
|
end
|
|
end
|
|
|
|
def test_boot_filename_has_no_dash_when_prefix_ends_with_slash
|
|
assert_equal "a/b/c/somefile", Proxy::TFTP.boot_filename('a/b/c/', '/d/somefile')
|
|
end
|
|
|
|
def test_boot_filename_uses_dash_when_prefix_does_not_end_with_slash
|
|
assert_equal "a/b/c-somefile", Proxy::TFTP.boot_filename('a/b/c', '/d/somefile')
|
|
end
|
|
end
|
