handle top level map, float, int

Author: nshntarora

main_test.go

@@ -146,10 +146,7 @@ func TestAPIResponseConsistency(t *testing.T) {
 	createdUsers := []db.User{}
 	for i := 0; i < NUMBER_OF_USERS; i++ {
 		user, _, err := cacheClient.CreateRandomUser()
-		if err != nil {
-			fmt.Println("error ", err)
-			return
-		}
+		assert.Nil(t, err)
 		createdUsers = append(createdUsers, *user)
 	}
 
@@ -159,10 +156,7 @@ func TestAPIResponseConsistency(t *testing.T) {
 
 	for i := 0; i < NUMBER_OF_USERS; i++ {
 		users, tt, err := cacheClient.PaginateUsers()
-		if err != nil {
-			fmt.Println("error ", err)
-			return
-		}
+		assert.Nil(t, err)
 		cachedUsersResponses = append(cachedUsersResponses, users...)
 		timeTakenForCacheClient += tt
 	}
@@ -171,10 +165,7 @@ func TestAPIResponseConsistency(t *testing.T) {
 	timeTakenForDefaultClient := time.Duration(0)
 	for i := 0; i < NUMBER_OF_USERS; i++ {
 		users, tt, err := defaultClient.PaginateUsers()
-		if err != nil {
-			fmt.Println("error ", err)
-			return
-		}
+		assert.Nil(t, err)
 		defaultUserResponses = append(defaultUserResponses, users...)
 		timeTakenForDefaultClient += tt
 	}
@@ -184,6 +175,27 @@ func TestAPIResponseConsistency(t *testing.T) {
 
 	assert.Equal(t, len(cachedUsersResponses), len(defaultUserResponses))
 	assert.Equal(t, cachedUsersResponses, defaultUserResponses)
+
+	cachedSystemDetailsResponses := []map[string]interface{}{}
+	defaultSystemDetailsResponses := []map[string]interface{}{}
+	// get system details from cache client and default client
+	for i := 0; i < NUMBER_OF_USERS; i++ {
+		details, _, err := defaultClient.GetSystemDetails()
+		assert.NotNil(t, details)
+		assert.Nil(t, err)
+		defaultSystemDetailsResponses = append(defaultSystemDetailsResponses, details)
+	}
+
+	// get system details from cache client and default client
+	for i := 0; i < NUMBER_OF_USERS; i++ {
+		details, _, err := cacheClient.GetSystemDetails()
+		assert.NotNil(t, details)
+		assert.Nil(t, err)
+		cachedSystemDetailsResponses = append(cachedSystemDetailsResponses, details)
+	}
+
+	assert.Equal(t, len(cachedSystemDetailsResponses), len(defaultSystemDetailsResponses))
+	assert.Equal(t, cachedSystemDetailsResponses, defaultSystemDetailsResponses)
 }
 
 func TestWithTodoOperations(t *testing.T) {

test_api/todo/db/connection.go

@@ -100,6 +100,76 @@ func (d Connection) GetAllTodos(todos *[]Todo) error {
 	return d.DB.Find(todos).Error
 }
 
+func (d Connection) GetTotalTodos() int {
+	var count int64
+	d.DB.Model(&Todo{}).Count(&count)
+	return int(count)
+}
+
+func (d Connection) GetSystemActivityHistoryForLast7Days() ([]int, error) {
+	var todos []Todo
+	err := d.GetAllTodos(&todos)
+	if err != nil {
+		return nil, err
+	}
+	activityHistory := make([]int, 7)
+	for i := 0; i < 7; i++ {
+		var activity int
+		for _, todo := range todos {
+			if todo.CreatedAt.After(time.Now().AddDate(0, 0, -i)) {
+				activity++
+			}
+		}
+		activityHistory[i] = activity
+	}
+	return activityHistory, nil
+}
+
+func (d Connection) GetSystemCompletionRateForLast7Days() ([]float64, error) {
+	var todos []Todo
+	err := d.GetAllTodos(&todos)
+	if err != nil {
+		return nil, err
+	}
+	completionRates := make([]float64, 7)
+	for i := 0; i < 7; i++ {
+		var completed, total int
+		for _, todo := range todos {
+			if todo.CreatedAt.After(time.Now().AddDate(0, 0, -i)) {
+				total++
+				if todo.Done {
+					completed++
+				}
+			}
+		}
+		if total == 0 {
+			completionRates[i] = 0
+			continue
+		}
+		completionRates[i] = float64(completed) / float64(total)
+	}
+	return completionRates, nil
+}
+
+func (d Connection) GetSystemCompletionRate() (float64, error) {
+	var todos []Todo
+	err := d.GetAllTodos(&todos)
+	if err != nil {
+		return 0, err
+	}
+	var completed, total int
+	for _, todo := range todos {
+		total++
+		if todo.Done {
+			completed++
+		}
+	}
+	if total == 0 {
+		return 0, nil
+	}
+	return float64(completed) / float64(total), nil
+}
+
 func (d Connection) GetTodosByUserID(todos *[]Todo, userID string) error {
 	return d.DB.Where("user_id=?", userID).Find(todos).Error
 }