Merge pull request #31 from luthermonson/fix-regression

Author: luthermonson

hosts.go

@@ -72,6 +72,13 @@ func (h *Hosts) Load() error {
 	}
 	defer file.Close()
 
+	// if you're reloading from disk confirm you refresh the hash maps and lines
+	if len(h.Lines) != 0 {
+		h.ips = lookup{l: make(map[string][]int)}
+		h.hosts = lookup{l: make(map[string][]int)}
+		h.Lines = []HostsLine{}
+	}
+
 	scanner := bufio.NewScanner(utfbom.SkipOnly(file))
 	for scanner.Scan() {
 		hl := NewHostsLine(scanner.Text())

hosts_test.go

@@ -263,3 +263,21 @@ func benchmarkHosts_Flush(c int, b *testing.B) {
 	assert.Nil(b, hosts.Flush())
 	assert.Nil(b, os.Remove("hostsfile"))
 }
+
+func TestHosts_Flush(t *testing.T) {
+	f, err := os.Create("hostsfile")
+	defer func() {
+		assert.Nil(t, f.Close())
+		assert.Nil(t, os.Remove("hostsfile"))
+	}()
+
+	assert.Nil(t, err)
+	hosts, err := NewCustomHosts("./hostsfile")
+	assert.Nil(t, err)
+	assert.Nil(t, hosts.Add("127.0.0.2", "host1"))
+	assert.Equal(t, 1, len(hosts.Lines))
+	assert.Equal(t, "127.0.0.2 host1", hosts.Lines[0].Raw)
+	assert.Nil(t, hosts.Flush())
+	assert.Equal(t, 1, len(hosts.Lines))
+	assert.Equal(t, "127.0.0.2 host1", hosts.Lines[0].Raw)
+}