created tests

cmd/main.go

@@ -31,22 +31,22 @@ func main() {
 		if q.Query == "" {
 			break
 		}
-		fmt.Printf("Time: %s\nUser: %s\nHost: %s\nID: %d\nSchema: %s\nLast_errno: %d\nKilled: %d\nQuery_time: %s\nLock_time: %s\nRows_sent: %d\nRows_examined: %d\nRows_affected: %d\nBytes_sent: %d\nQuery: %s\n",
-			q.Time,
-			q.User,
-			q.Host,
-			q.ID,
-			q.Schema,
-			q.LastErrNo,
-			q.Killed,
-			q.QueryTime,
-			q.LockTime,
-			q.RowsSent,
-			q.RowsExamined,
-			q.RowsAffected,
-			q.BytesSent,
-			q.Query,
-		)
+		// fmt.Printf("Time: %s\nUser: %s\nHost: %s\nID: %d\nSchema: %s\nLast_errno: %d\nKilled: %d\nQuery_time: %s\nLock_time: %s\nRows_sent: %d\nRows_examined: %d\nRows_affected: %d\nBytes_sent: %d\nQuery: %s\n",
+		// 	q.Time,
+		// 	q.User,
+		// 	q.Host,
+		// 	q.ID,
+		// 	q.Schema,
+		// 	q.LastErrNo,
+		// 	q.Killed,
+		// 	q.QueryTime,
+		// 	q.LockTime,
+		// 	q.RowsSent,
+		// 	q.RowsExamined,
+		// 	q.RowsAffected,
+		// 	q.BytesSent,
+		// 	q.Query,
+		// )
 		count++
 	}
 

query.go

@@ -29,10 +29,13 @@ type Query struct {
 }
 
 // Fingerprint returns Query.query's MD5 fingerprint.
-func (q Query) Fingerprint() string {
+func (q Query) Fingerprint() (string, error) {
 	h := md5.New()
-	io.WriteString(h, q.Query)
-	return fmt.Sprintf("%x", h.Sum(nil))
+	_, err := io.WriteString(h, q.Query)
+	if err != nil {
+		return "", err
+	}
+	return fmt.Sprintf("%x", h.Sum(nil)), nil
 }
 
 // parseHeader parses everything that begin with #

query_test.go

@@ -0,0 +1,29 @@
+package slowql
+
+import "testing"
+
+func TestQuery_Fingerprint(t *testing.T) {
+	tests := []struct {
+		name    string
+		query   string
+		want    string
+		wantErr bool
+	}{
+		{name: "pass", query: "foobar", want: "3858f62230ac3c915f300c664312c63f", wantErr: false},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			q := Query{
+				Query: tt.query,
+			}
+			got, err := q.Fingerprint()
+			if (err != nil) != tt.wantErr {
+				t.Errorf("Query.Fingerprint() error = %v, wantErr %v", err, tt.wantErr)
+				return
+			}
+			if got != tt.want {
+				t.Errorf("Query.Fingerprint() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}