test(clap counter): test cases for clap counter GET and POST methods

Author: prinzpiuz

go.mod

@@ -8,6 +8,7 @@ require (
 )
 
 require (
+	github.com/DATA-DOG/go-sqlmock v1.5.2 // indirect
 	github.com/didip/tollbooth v4.0.2+incompatible // indirect
 	github.com/didip/tollbooth/v8 v8.0.1 // indirect
 	github.com/go-pkgz/expirable-cache/v3 v3.0.0 // indirect

go.sum

@@ -1,3 +1,5 @@
+github.com/DATA-DOG/go-sqlmock v1.5.2 h1:OcvFkGmslmlZibjAjaHm3L//6LiuBgolP7OputlJIzU=
+github.com/DATA-DOG/go-sqlmock v1.5.2/go.mod h1:88MAG/4G7SMwSE3CeA0ZKzrT5CiOU3OJ+JlNzwDqpNU=
 github.com/aymanbagabas/go-osc52/v2 v2.0.1 h1:HwpRHbFMcZLEVr42D4p7XBqjyuxQH5SMiErDT4WkJ2k=
 github.com/aymanbagabas/go-osc52/v2 v2.0.1/go.mod h1:uYgXzlJ7ZpABp8OJ+exZzJJhRNQ2ASbcXHWsFqH8hp8=
 github.com/charmbracelet/lipgloss v0.12.1 h1:/gmzszl+pedQpjCOH+wFkZr/N90Snz40J/NR7A0zQcs=
@@ -18,6 +20,7 @@ github.com/gorilla/handlers v1.5.2 h1:cLTUSsNkgcwhgRqvCNmdbRWG0A3N4F+M2nWKdScwyE
 github.com/gorilla/handlers v1.5.2/go.mod h1:dX+xVpaxdSw+q0Qek8SSsl3dfMk3jNddUkMzo0GtH0w=
 github.com/gorilla/mux v1.8.1 h1:TuBL49tXwgrFYWhqrNgrUNEY92u81SPhu7sTdzQEiWY=
 github.com/gorilla/mux v1.8.1/go.mod h1:AKf9I4AEqPTmMytcMc0KkNouC66V3BtZ4qD5fmWSiMQ=
+github.com/kisielk/sqlstruct v0.0.0-20201105191214-5f3e10d3ab46/go.mod h1:yyMNCyc/Ib3bDTKd379tNMpB/7/H5TjM2Y9QJ5THLbE=
 github.com/lucasb-eyer/go-colorful v1.2.0 h1:1nnpGOrhyZZuNyfu1QjKiUICQ74+3FNCN69Aj6K7nkY=
 github.com/lucasb-eyer/go-colorful v1.2.0/go.mod h1:R4dSotOR9KMtayYi1e77YzuveK+i7ruzyGqttikkLy0=
 github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=

src/app/app.go

@@ -7,6 +7,7 @@ import (
 	"fmt"
 	"os"
 	"path/filepath"
+	fp "path/filepath"
 )
 
 type App struct {
@@ -15,10 +16,11 @@ type App struct {
 }
 
 type Config struct {
-	ServerConfig server.ServerConfig
-	Name         string
-	Version      string
-	Env          string
+	ServerConfig  server.ServerConfig
+	Name          string
+	Version       string
+	Env           string
+	BaseDirectory string
 }
 
 func (app *App) Start() {
@@ -32,6 +34,13 @@ func LoadConfig(filepath string) Config {
 		fmt.Println(err)
 		return Config{}
 	}
+	baseDir, err := os.Getwd()
+	if err != nil {
+		panic(err)
+	}
+	config.BaseDirectory = baseDir
+	config.ServerConfig.BaseDirectory = baseDir
+	config.ServerConfig.DB.DBBaseDirectory = fp.Join(baseDir, "src/db")
 	return config
 }
 

src/db/db.go

@@ -7,6 +7,7 @@ import (
 	"fmt"
 	"log"
 	"os"
+	"path/filepath"
 
 	"github.com/golang-migrate/migrate"
 	"github.com/golang-migrate/migrate/database/sqlite3"
@@ -16,10 +17,11 @@ import (
 )
 
 type DB struct {
-	DBLocation  string
-	Vacuum      string
-	ForeignKeys bool
-	Connection  *sql.DB
+	DBLocation      string
+	Vacuum          string
+	ForeignKeys     bool
+	Connection      *sql.DB
+	DBBaseDirectory string
 }
 
 func closeDB(new_db *sql.DB) {
@@ -57,7 +59,8 @@ func (db *DB) Initialize() error {
 		closeDB(new_db)
 		return err
 	}
-	err = runMigrations(new_db)
+	migrationFiles := filepath.Join(db.DBBaseDirectory, "migrations")
+	err = runMigrations(new_db, migrationFiles)
 	if err != nil {
 		log.Println("Error Running Migrations")
 		closeDB(new_db)
@@ -67,14 +70,15 @@ func (db *DB) Initialize() error {
 	return nil
 }
 
-func runMigrations(db *sql.DB) error {
+func runMigrations(db *sql.DB, migrationFiles string) error {
 	instance, err := sqlite3.WithInstance(db, &sqlite3.Config{})
 	if err != nil {
 		log.Printf("Error Connecting With SQLite Instance: %s", err)
 		return err
 	}
 
-	fSrc, err := (&file.File{}).Open("./src/db/migrations")
+	fSrc, err := (&file.File{}).Open(migrationFiles)
+	print()
 	if err != nil {
 		closeFile(fSrc)
 		log.Printf("Error Getting Migration Files: %s", err)

src/db/query.go

@@ -55,3 +55,10 @@ const updateLike = `INSERT INTO Likes(uri, count, domain_id)
 					ON CONFLICT(uri)
 					DO UPDATE
 					SET count = count + 1`
+
+const addDomainAndSettings = `BEGIN TRANSACTION;
+							  INSERT INTO DomainSettings (likes, comments, created_time)
+							  VALUES (?, ?, datetime());
+							  INSERT INTO Domain (settings_id, domain, created_time)
+							  VALUES (last_insert_rowid(), ?, datetime());
+							  COMMIT TRANSACTION;`

src/db/utils.go

@@ -65,7 +65,7 @@ func GetLikedIP(db *sql.DB, domain string, page string, ip string) LikedIPs {
 			log.Printf("DB: IP %s, Not Found Liked in %s for %s", ip, domain, page)
 			return LikedIPs{}
 		}
-		log.Printf("Error Getting Likes On %s For %s: %v", ip, page, err)
+		log.Printf("Error Getting LikedIPs On %s For %s: %v", ip, page, err)
 		return LikedIPs{}
 	}
 	return likedIP
@@ -83,3 +83,10 @@ func UpdateLikeCount(db *sql.DB, page string, doamin_id int) error {
 	_, err := db.Exec(updateLike, page, doamin_id)
 	return err
 }
+
+func AddDomainAndSettings(db *sql.DB, domain string, likes int, comments int) {
+	_, err := db.Exec(addDomainAndSettings, likes, comments, domain)
+	if err != nil {
+		log.Printf("Error Adding Domain %s: %v", domain, err)
+	}
+}

src/server/handlers/clap_counter.go

@@ -65,10 +65,7 @@ func get_likes(r *http.Request, clapCounter ClapCounter) db.Likes {
 func already_liked_IP(r *http.Request, clapCounter ClapCounter) bool {
 	db_connection := get_db_connection(r)
 	liked_ips := db.GetLikedIP(db_connection, clapCounter.URL, clapCounter.Page, clapCounter.remote_addr)
-	if liked_ips.IsEmpty() {
-		return false
-	}
-	return true
+	return !liked_ips.IsEmpty()
 }
 
 // add_like_to_page updates the like count for the given page and domain ID in the database.
@@ -105,6 +102,10 @@ func add_like_to_page(r *http.Request, clapCounter ClapCounter, doamin_id int) e
 func GetClaps(w http.ResponseWriter, r *http.Request) {
 	var clapCounter ClapCounter
 	var continue_counting bool = true
+	content_type_err := check_for_request_content_type(w, r)
+	if content_type_err != nil {
+		return
+	}
 	clapCounter.remote_addr = r.RemoteAddr
 	clapCounter.URL = get_domain(r.Referer())
 	clapCounter.Page = get_path(r, w)
@@ -127,10 +128,6 @@ func GetClaps(w http.ResponseWriter, r *http.Request) {
 		case http.MethodGet:
 			clapCounter.SetClapCounter(clapCounter.URL, ClapCount, likes.Count, true)
 		case http.MethodPost:
-			content_type_err := check_for_request_content_type(w, r)
-			if content_type_err != nil {
-				return
-			}
 			if already_liked_IP(r, clapCounter) {
 				clapCounter.SetClapCounter(clapCounter.URL, ClapAlreadyCounted, likes.Count, false)
 			} else {

src/server/handlers/ping.go

@@ -3,27 +3,35 @@ package handlers
 
 import (
 	"encoding/json"
-	"errors"
+	"log"
 	"net/http"
 )
 
 // Ping is an HTTP handler that responds with a "pong" message.
 // It also checks the connection to the database and returns an error if the connection fails.
 func Ping(w http.ResponseWriter, r *http.Request) {
+	msg := "pong"
 	db := get_db_connection(r)
 	if db != nil {
 		err := db.Ping()
 		if err != nil {
-			setHTTPError(w, err, "Database Error", http.StatusInternalServerError)
+			w.WriteHeader(http.StatusInternalServerError)
+			msg = "Database Ping Error"
 		}
 
 	} else {
-		msg := "Database Not Found"
-		setHTTPError(w, errors.New(msg), "Database Error", http.StatusInternalServerError)
+		w.WriteHeader(http.StatusInternalServerError)
+		msg = "Database Not Found"
 	}
-	p := map[string]string{"message": "pong"}
-	err := json.NewEncoder(w).Encode(p)
+	responseBody := map[string]string{"message": msg}
+	jsonData, err := json.Marshal(responseBody)
 	if err != nil {
-		setHTTPError(w, err, "JSON Error", http.StatusInternalServerError)
+		log.Printf("Error Encoding JSON: %v", err.Error())
+		return
+	}
+	w.Header().Set("Content-Type", "application/json")
+	_, err = w.Write(jsonData)
+	if err != nil {
+		log.Printf("Error writing response: %v", err)
 	}
 }