add tests and fix linter

aryanmehrotra · aryanmehrotra · commit 088e19aae0ba · 2025-02-24T16:56:45.000+05:30
diff --git a/main.go b/main.go
@@ -36,12 +36,10 @@ func main() {
 			if r.URL.Path == rootPath {
 				filePath += indexHTML
 			} else if !ok {
-				if stat, err := os.Stat(filePath); err == nil && stat.IsDir() {
-					if _, err = os.Stat(filePath + ".html"); os.IsNotExist(err) {
-						filePath += indexHTML
-					} else {
-						filePath += htmlExtension
-					}
+				if _, err := os.Stat(filePath + ".html"); err == nil {
+					filePath += htmlExtension
+				} else if stat, err := os.Stat(filePath); err == nil && stat.IsDir() {
+					filePath += indexHTML
 				}
 			}
 
diff --git a/main_test.go b/main_test.go
@@ -76,3 +76,62 @@ func TestServer(t *testing.T) {
 	//nolint:staticcheck // Ignore as we are testing the server
 	os.RemoveAll(tempDir)
 }
+
+func TestServer_FileAndDirectorySameName(t *testing.T) {
+	// Create a temporary directory
+	tempDir := t.TempDir()
+
+	// Create an "index.html" file
+	indexFilePath := filepath.Join(tempDir, "index.html")
+	if err := os.WriteFile(indexFilePath, []byte("<html><body>Index File</body></html>"), 0600); err != nil {
+		t.Fatalf("Failed to write file %s: %v", indexFilePath, err)
+	}
+
+	// Create an "index" directory
+	indexDirPath := filepath.Join(tempDir, "index")
+	if err := os.MkdirAll(indexDirPath, 0600); err != nil {
+		t.Fatalf("Failed to create directory %s: %v", indexDirPath, err)
+	}
+
+	// Create an "index/index.html" file inside the directory
+	indexDirFilePath := filepath.Join(indexDirPath, "index.html")
+	if err := os.WriteFile(indexDirFilePath, []byte("<html><body>Index Directory</body></html>"), 0600); err != nil {
+		t.Fatalf("Failed to write file %s: %v", indexDirFilePath, err)
+	}
+
+	// Set the environment variable for the static file path
+	t.Setenv("STATIC_DIR_PATH", tempDir)
+
+	go main()
+
+	time.Sleep(3 * time.Second) // Allow time for the server to start
+
+	tests := []struct {
+		path       string
+		statusCode int
+	}{
+		{"/index", http.StatusOK},  // Should serve "index.html" file
+		{"/index/", http.StatusOK}, // Should serve "index/index.html"
+		{"/index.html", http.StatusOK},
+	}
+
+	for _, test := range tests {
+		req, err := http.NewRequestWithContext(context.Background(), http.MethodGet, "http://localhost:8000"+test.path, http.NoBody)
+		if err != nil {
+			t.Fatalf("Failed to create request: %v", err)
+		}
+
+		client := &http.Client{}
+
+		resp, err := client.Do(req)
+		if err != nil {
+			t.Fatalf("Failed to perform request: %v", err)
+		}
+
+		if resp.StatusCode != test.statusCode {
+			t.Errorf("Expected status code %v, got %v for path %v", test.statusCode, resp.StatusCode, test.path)
+		}
+
+		resp.Body.Close()
+	}
+}
