enova · kav121988 · Dec 3, 2018 · Dec 3, 2018
diff --git a/exercise-003-web/.idea/encodings.xml b/exercise-003-web/.idea/encodings.xml
diff --git a/exercise-003-web/.idea/exercise-003-web.iml b/exercise-003-web/.idea/exercise-003-web.iml
diff --git a/exercise-003-web/.idea/misc.xml b/exercise-003-web/.idea/misc.xml
diff --git a/exercise-003-web/.idea/modules.xml b/exercise-003-web/.idea/modules.xml
diff --git a/exercise-003-web/.idea/vcs.xml b/exercise-003-web/.idea/vcs.xml
diff --git a/exercise-003-web/.idea/workspace.xml b/exercise-003-web/.idea/workspace.xml
diff --git a/exercise-003-web/build.sh b/exercise-003-web/build.sh
@@ -0,0 +1,3 @@
+gofmt -w .
+golint ./...
+go run exhibit-f/server.go
diff --git a/exercise-003-web/exhibit-c/server.go b/exercise-003-web/exhibit-c/server.go
@@ -5,6 +5,7 @@ import (
 	"net/http"
 )
 
+// View structure
 type View struct {
 	Name string
 	Age  int

diff --git a/exercise-003-web/exhibit-d/home.html b/exercise-003-web/exhibit-d/home.html
@@ -1,11 +1,11 @@
 <!DOCTYPE html>
 <html>
-  <title>Learning Go</title>
+<title>Learning Go</title>
 
-  <body>
-    <form action="/signup" method="POST">
-      <input type="text" name="username" placeholder="Enter Name">
-      <input type="submit" value="Sign-Up"/>
-    </form>
-  </body>
-</html>
+<body>
+<form action="/signup" method="POST">
+    <input type="text" name="username" placeholder="Enter Name">
+    <input type="submit" value="Sign-Up"/>
+</form>
+</body>
+</html>
diff --git a/exercise-003-web/exhibit-e/server_test.go b/exercise-003-web/exhibit-e/server_test.go
@@ -1,10 +1,10 @@
 package main
 
 import (
+	"github.com/stretchr/testify/assert"
 	"net/http"
 	"net/http/httptest"
 	"testing"
-	"github.com/stretchr/testify/assert"
 )
 
 func init() {

diff --git a/exercise-003-web/exhibit-f/home.html b/exercise-003-web/exhibit-f/home.html
@@ -0,0 +1,23 @@
+<!DOCTYPE html>
+<html>
+  <title>Learning Go</title>
+
+  <body>
+    <form action="/home" method="POST">
+      <input type="text" name="username" placeholder="Enter Name">
+      <input type="submit" value="Sign-Up"/>
+    </form>
+    <table>
+        <tr>
+            <th>Username</th>
+            <th>Attempts</th>
+        </tr>
+        {{range $key, $value := .}}
+            <tr>
+                <td><strong>{{$key}}</strong></td>
+                <td><strong>{{$value}}</strong></td>
+            </tr>
+        {{end}}
+    </table>
+  </body>
+</html>
diff --git a/exercise-003-web/exhibit-f/server.go b/exercise-003-web/exhibit-f/server.go
@@ -0,0 +1,47 @@
+package main
+
+import (
+	"fmt"
+	"html/template"
+	"log"
+	"net/http"
+)
+
+var loginInfo = make(map[string]int)
+
+var homeT *template.Template
+
+func setup(dir string) {
+	homeT = template.Must(template.ParseFiles(dir + "/exhibit-f/home.html"))
+}
+
+func home(w http.ResponseWriter, r *http.Request) {
+	switch r.Method {
+	case "POST":
+		r.ParseForm()
+		username := r.Form.Get("username")
+		if username != "" {
+			loginInfo[username]++
+		}
+		err := homeT.Execute(w, loginInfo)
+		if err != nil {
+			log.Println(err)
+		}
+	case "GET":
+		err := homeT.Execute(w, loginInfo)
+		if err != nil {
+			log.Println(err)
+		}
+	default:
+		fmt.Fprint(w, "Invalid HTTP method")
+	}
+}
+
+func main() {
+	setup(".")
+	http.HandleFunc("/home", home)
+	err := http.ListenAndServe(":8080", nil)
+	if err != nil {
+		panic(err)
+	}
+}
diff --git a/exercise-003-web/exhibit-f/server_test.go b/exercise-003-web/exhibit-f/server_test.go
@@ -0,0 +1,65 @@
+package main
+
+import (
+	"github.com/stretchr/testify/assert"
+	"net/http"
+	"net/http/httptest"
+	"net/url"
+	"testing"
+)
+
+func init() {
+	setup("../")
+}
+
+func TestServer(t *testing.T) {
+	assert := assert.New(t)
+
+	req, err := http.NewRequest("GET", "http://localhost:8080/home", nil)
+	assert.Nil(err)
+
+	w := httptest.NewRecorder()
+	home(w, req)
+
+	if status := w.Code; status != http.StatusOK {
+		t.Errorf("Status code differs. Expected %d .\n Got %d instead", http.StatusOK, status)
+	}
+
+	assert.Contains(w.Body.String(), "Enter Name")
+	assert.Contains(w.Body.String(), "Username")
+	assert.Contains(w.Body.String(), "Attempts")
+}
+
+func TestPost(t *testing.T) {
+	assert := assert.New(t)
+
+	v := url.Values{}
+
+	req, err := http.NewRequest("POST", "http://localhost:8080/home", nil)
+	v.Add("username", "Kavya")
+	req.PostForm = v
+	req.Header.Add("Content-Type", "application/x-www-form-urlencoded")
+	assert.Nil(err)
+
+	w := httptest.NewRecorder()
+	home(w, req)
+	assert.Contains(w.Body.String(), "Kavya")
+	assert.Contains(w.Body.String(), "1")
+}
+
+func TestConnect(t *testing.T) {
+	assert := assert.New(t)
+
+	req, err := http.NewRequest("PUT", "http://localhost:8080/home", nil)
+	assert.Nil(err)
+
+	w := httptest.NewRecorder()
+	home(w, req)
+
+	if status := w.Code; status != http.StatusOK {
+		t.Errorf("Status code differs. Expected %d .\n Got %d instead", http.StatusOK, status)
+	}
+
+	assert.Contains(w.Body.String(), "Invalid HTTP method")
+
+}