Added logic for finding devices

controllers/CompaniesController.go

@@ -1,19 +0,0 @@
-package controllers
-
-// import (
-// 	"net/http"
-// 	models "gitlab.com/pactual1/backend/models"
-
-// 	"gitlab.com/pactual1/backend/models"
-// 	"github.com/gin-gonic/gin"
-// )
-
-// func Companies(c *gin.Context) {
-// 	var companies []models.Company
-// 	// Fetch companies from DB here
-// 	if err := models.FetchCompanies(&companies); err != nil {
-// 		c.JSON(http.StatusInternalServerError, gin.H{"error": "Error fetching companies"})
-// 		return
-// 	}
-// 	c.JSON(http.StatusOK, gin.H{"companies": companies})
-// }

controllers/DevicesController.go

@@ -2,6 +2,7 @@ package controllers
 
 import (
 	"encoding/json"
+	"log"
 	"net/http"
 
 	"github.com/gin-gonic/gin"
@@ -20,12 +21,30 @@ func SaveDeviceInfofunc(c *gin.Context) {
 		return
 	}
 
-	// Convert raw JSON bytes to string
+	// Unmarshal to the deviceInfo structure
+	err = json.Unmarshal(rawData, &deviceInfo)
+	if err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid JSON payload"})
+		return
+	}
+		
+
 	deviceInfo.RawJSON = string(rawData)
 
-	// Save deviceInfo to your database here
-	db :=shared.GetDb()
-	db.Create(&deviceInfo)
+	// Attempt to find the device by IMEI; if found, set the DeviceID
+	var device models.Device
+	if err := shared.GetDb().Where("imei = ?", deviceInfo.IMEI).First(&device).Error; err == nil {
+		deviceInfo.DeviceID = device.ID
+	} else {
+		log.Printf("Could not find device with imei %v", deviceInfo.IMEI)
+	}
 
-	c.JSON(http.StatusOK, gin.H{"message": "Successfully received device info", "data": deviceInfo})
+
+	// Save deviceInfo to your database
+	if err := shared.GetDb().Create(&deviceInfo).Error; err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "Could not save device info"})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{"message": "Successfully received and saved device info", "data": deviceInfo})
 }

controllers/DevicesController_test.go

@@ -0,0 +1,81 @@
+package controllers
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"log"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"github.com/gin-gonic/gin"
+	"github.com/stretchr/testify/assert"
+
+	"gitlab.com/pactual1/backend/config"
+	"gitlab.com/pactual1/backend/shared"
+)
+
+// Assume setupServer() returns your gin.Engine instance
+func runTestServer() *httptest.Server {
+	// Initialize your database or any other setup
+	// For the sake of this example, I'll just return the Gin engine
+
+	// LOAD APPLICATION CONFIGURATION
+	err := config.Load()
+	if err != nil {
+		log.Fatal(err)
+	}
+	
+	shared.Init()
+	r := gin.Default()
+	r.POST("/device_info", SaveDeviceInfofunc)
+	return httptest.NewServer(r)
+}
+
+func TestSaveDeviceInfofunc(t *testing.T) {
+	ts := runTestServer()
+	defer ts.Close()
+
+	t.Run("it should return 400 when JSON payload is invalid", func(t *testing.T) {
+		payload := []byte(`{"InvalidPayload": `)
+		resp, err := http.Post(fmt.Sprintf("%s/device_info", ts.URL), "application/json", bytes.NewBuffer(payload))
+
+		if err != nil {
+			t.Fatalf("Expected no error, got %v", err)
+		}
+		
+
+		assert.Equal(t, 400, resp.StatusCode)
+	})
+
+	t.Run("it should log a message when IMEI is not found", func(t *testing.T) {
+		// Add a valid payload but with an unknown IMEI
+		payload := []byte(`{"IMEI": "UNKNOWN", "SomeOtherField": "Value"}`)
+		resp, err := http.Post(fmt.Sprintf("%s/device_info", ts.URL), "application/json", bytes.NewBuffer(payload))
+
+		if err != nil {
+			t.Fatalf("Expected no error, got %v", err)
+		}
+
+		// Here, ideally you'd want to check your logs to ensure the message was logged
+		assert.Equal(t, 200, resp.StatusCode)
+	})
+
+	t.Run("it should return 200 when all conditions are correct", func(t *testing.T) {
+		// Add a valid payload
+		deviceInfo := map[string]interface{}{
+			"IMEI":          "SOME_VALID_IMEI",
+			"SomeOtherField": "Value",
+		}
+
+		payload, _ := json.Marshal(deviceInfo)
+		resp, err := http.Post(fmt.Sprintf("%s/device_info", ts.URL), "application/json", bytes.NewBuffer(payload))
+
+		if err != nil {
+			t.Fatalf("Expected no error, got %v", err)
+		}
+
+		assert.Equal(t, 200, resp.StatusCode)
+	})
+}