From ae6552af91cc925fea1275a7606ba9a420150dba Mon Sep 17 00:00:00 2001
From: Craig Johnston <cj@imti.co>
Date: Sun, 23 Jul 2023 00:10:27 -0700
Subject: [PATCH] added simple test

---
 txeh_test.go | 78 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 78 insertions(+)
 create mode 100644 txeh_test.go

diff --git a/txeh_test.go b/txeh_test.go
new file mode 100644
index 0000000..c4bd246
--- /dev/null
+++ b/txeh_test.go
@@ -0,0 +1,78 @@
+package txeh
+
+import (
+	"testing"
+)
+
+func TestNewHostsDefault(t *testing.T) {
+	hosts, err := NewHostsDefault()
+	if err != nil {
+		panic(err)
+	}
+
+	testHostName := "test-new-hosts-default"
+	hosts.AddHost("127.100.100.100", testHostName)
+	foundHostnames := hosts.ListHostsByIP("127.100.100.100")
+	found := false
+	for _, hn := range foundHostnames {
+		if hn == testHostName {
+			found = true
+			break
+		}
+	}
+
+	if !found {
+		t.Fatal("TestNewHostsDefault was unable find the hostname it added")
+	}
+}
+
+func TestNewHosts(t *testing.T) {
+	rawData := ""
+	blankHosts, err := NewHosts(&HostsConfig{
+		RawText: &rawData,
+	})
+	if err != nil {
+		t.Fatalf("TestNewHosts failed on NewHosts: %v", err)
+	}
+
+	testHostName := "test-new-hosts-default"
+	blankHosts.AddHost("127.100.100.100", testHostName)
+	foundHostnames := blankHosts.ListHostsByIP("127.100.100.100")
+	found := false
+	for _, hn := range foundHostnames {
+		if hn == testHostName {
+			found = true
+			break
+		}
+	}
+
+	if !found {
+		t.Fatal("TestNewHosts was unable find the hostname it added")
+	}
+
+	renderedData := blankHosts.RenderHostsFile()
+	fromRenderedHosts, err := NewHosts(&HostsConfig{
+		RawText: &renderedData,
+	})
+
+	for _, k := range []string{"A", "T", "T", "X", "X", "E", "E", "H", "H"} {
+		fromRenderedHosts.AddHost("127.100.100.100", k)
+	}
+
+	fromRenderedHosts.RemoveHosts([]string{"A "})
+
+	err = fromRenderedHosts.Save()
+	if err == nil {
+		t.Fatalf("Should not be able to save when using RawText.")
+	}
+
+	foundHostnames = fromRenderedHosts.ListHostsByIP("127.100.100.100")
+	found = false
+	for _, hn := range foundHostnames {
+		if hn == testHostName {
+			found = true
+			break
+		}
+	}
+
+}