Current File : //opt/cpanel/ea-ruby27/src/passenger-release-6.0.23/test/ruby/utils/hosts_file_parser.rb
require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
PhusionPassenger.require_passenger_lib 'utils/hosts_file_parser'
require 'stringio'
module PhusionPassenger
describe Utils::HostsFileParser do
before :each do
@io = StringIO.new
end
def create
@io.rewind
@parser = Utils::HostsFileParser.new(@io)
end
describe "parsing" do
it "ignores comments" do
@io.puts("# 127.0.0.1 foo.com")
create
@parser.ip_count.should == 0
@parser.host_count.should == 0
end
it "ignores comments that come after leading spaces" do
@io.puts(" # 127.0.0.1 foo.com")
create
@parser.ip_count.should == 0
@parser.host_count.should == 0
end
it "ignores comments that come after leading tabs" do
@io.puts("\t# 127.0.0.1 foo.com")
create
@parser.ip_count.should == 0
@parser.host_count.should == 0
end
it "ignores empty lines" do
@io.puts("127.0.0.1 foo.com")
@io.puts
@io.puts("127.0.0.1 bar.com")
@io.puts("127.0.0.2 baz.com")
create
@parser.ip_count.should == 2
@parser.host_count.should == 3
end
it "ignores leading and trailing spaces" do
@io.puts(" 127.0.0.1 foo.com")
@io.puts
@io.puts(" 127.0.0.1 bar.com ")
@io.puts("127.0.0.2 baz.com ")
create
@parser.ip_count.should == 2
@parser.host_count.should == 3
@parser.resolve("foo.com").should == "127.0.0.1"
@parser.resolve("bar.com").should == "127.0.0.1"
@parser.resolve("baz.com").should == "127.0.0.2"
end
it "ignores leading and trailing tabs" do
@io.puts("\t\t127.0.0.1 foo.com")
@io.puts
@io.puts("\t127.0.0.1 bar.com\t")
@io.puts("127.0.0.2 baz.com\t\t")
create
@parser.ip_count.should == 2
@parser.host_count.should == 3
@parser.resolve("foo.com").should == "127.0.0.1"
@parser.resolve("bar.com").should == "127.0.0.1"
@parser.resolve("baz.com").should == "127.0.0.2"
end
it "correctly handles spaces as seperators" do
@io.puts("127.0.0.1 foo.com bar.com baz.com")
create
@parser.ip_count.should == 1
@parser.host_count.should == 3
@parser.resolve("foo.com").should == "127.0.0.1"
@parser.resolve("bar.com").should == "127.0.0.1"
@parser.resolve("baz.com").should == "127.0.0.1"
end
it "correctly handles tabs as seperators" do
@io.puts("127.0.0.1\tfoo.com\t\tbar.com baz.com")
create
@parser.ip_count.should == 1
@parser.host_count.should == 3
@parser.resolve("foo.com").should == "127.0.0.1"
@parser.resolve("bar.com").should == "127.0.0.1"
@parser.resolve("baz.com").should == "127.0.0.1"
end
end
describe "#resolve" do
it "returns nil if the host name is not in the file" do
@io.puts("127.0.0.1 foo.com")
create
@parser.resolve("bar.com").should be_nil
end
it "returns the IP address associated with the host name if it exists" do
@io.puts("255.255.255.255 foo.com")
create
@parser.resolve("foo.com").should == "255.255.255.255"
end
it "is case-insensitive" do
@io.puts("255.255.255.255 fOO.com")
create
@parser.resolve("foo.COM").should == "255.255.255.255"
end
it "correctly handles lines that contain multiple host names" do
@io.puts("255.255.255.255 foo.com bar.com")
create
@parser.resolve("foo.com").should == "255.255.255.255"
@parser.resolve("bar.com").should == "255.255.255.255"
@parser.resolve("baz.com").should be_nil
end
it "always returns 127.0.0.1 for localhost" do
create
@parser.resolve("localhost").should == "127.0.0.1"
@parser.resolve("localHOST").should == "127.0.0.1"
end
end
describe "#resolves_to_localhost?" do
before :each do
@io.puts "127.0.0.1 kotori"
@io.puts "192.168.0.1 kanako"
@io.puts "::1 ageha"
@io.puts "::2 sawako"
@io.puts "0.0.0.0 mizusawa"
create
end
it "returns true if the host name resolves to 127.0.0.1" do
@parser.resolves_to_localhost?("kotori").should be_truthy
end
it "returns true if the host name resolves to ::1" do
@parser.resolves_to_localhost?("ageha").should be_truthy
end
it "returns true if the host name resolves to 0.0.0.0" do
@parser.resolves_to_localhost?("mizusawa").should be_truthy
end
it "returns false if the host name resolves to something else" do
@parser.resolves_to_localhost?("sawako").should be_falsey
@parser.resolves_to_localhost?("kanako").should be_falsey
end
it "returns false if the host name does not resolve" do
@parser.resolves_to_localhost?("foo.com").should be_falsey
end
end
describe "#add_group_data" do
before :each do
@standard_entries =
"127.0.0.1 kotori hazuki\n" +
" 192.168.0.1 kanako\n\n" +
"\t::1 ageha sawako\n" +
"0.0.0.0 mizusawa naru\n"
@target = StringIO.new
end
it "adds the group data if it doesn't exist" do
@io.puts @standard_entries
create
@parser.add_group_data("some marker",
"# a comment\n" +
"127.0.0.1 foo\n")
@parser.write(@target)
@target.string.should ==
@standard_entries +
"###### BEGIN some marker ######\n" +
"# a comment\n" +
"127.0.0.1 foo\n" +
"###### END some marker ######\n"
end
it "replaces the existing group data if it does exist" do
@io.puts "###### BEGIN some marker ######\n" +
"# another comment\n" +
"127.0.0.1 bar\n" +
"###### END some marker ######\n" +
"\n" +
@standard_entries +
"###### BEGIN some other marker ######\n" +
"# another comment\n" +
"127.0.0.1 bar\n" +
"###### END some other marker ######\n"
create
@parser.add_group_data("some marker", "127.0.0.1 foo\n")
@parser.write(@target)
@target.string.should ==
"###### BEGIN some marker ######\n" +
"127.0.0.1 foo\n" +
"###### END some marker ######\n" +
"\n" +
@standard_entries +
"###### BEGIN some other marker ######\n" +
"# another comment\n" +
"127.0.0.1 bar\n" +
"###### END some other marker ######\n"
end
it "correctly handles the lack of a terminating newline in the group data" do
@io.puts @standard_entries
create
@parser.add_group_data("some marker",
"# a comment\n" +
"127.0.0.1 foo")
@parser.write(@target)
@target.string.should ==
@standard_entries +
"###### BEGIN some marker ######\n" +
"# a comment\n" +
"127.0.0.1 foo\n" +
"###### END some marker ######\n"
end
it "ensures that the group data starts at a new line" do
@io.write "127.0.0.1 foo.com"
create
@parser.add_group_data("some marker", "127.0.0.1 bar.com")
@parser.write(@target)
@target.string.should ==
"127.0.0.1 foo.com\n" +
"###### BEGIN some marker ######\n" +
"127.0.0.1 bar.com\n" +
"###### END some marker ######\n"
end
end
describe "#write" do
it "preserves all comments and leading and trailing whitespaces" do
@io.puts "127.0.0.1 foo.com "
@io.puts "# a comment"
@io.puts "\t127.0.0.1 bar.com\t"
create
original_data = @io.string
target = StringIO.new
@parser.write(target)
target.string.should == original_data
end
end
end
end # module PhusionPassenger
Mini Shell