JezK
Edit File: hosts_file_parser.rb
require File.expand_path(File.dirname(__FILE__) + '/../spec_helper') require 'phusion_passenger/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_true end it "returns true if the host name resolves to ::1" do @parser.resolves_to_localhost?("ageha").should be_true end it "returns true if the host name resolves to 0.0.0.0" do @parser.resolves_to_localhost?("mizusawa").should be_true end it "returns false if the host name resolves to something else" do @parser.resolves_to_localhost?("sawako").should be_false @parser.resolves_to_localhost?("kanako").should be_false end it "returns false if the host name does not resolve" do @parser.resolves_to_localhost?("foo.com").should be_false 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