status-go/centralizedmetrics/providers/mixpanel_test.go

69 lines
2.3 KiB
Go

package providers
import (
"encoding/json"
"net/http"
"net/http/httptest"
"testing"
"github.com/stretchr/testify/require"
"github.com/status-im/status-go/centralizedmetrics/common"
)
func TestMixpanelMetricProcessor(t *testing.T) {
var t1 int64 = 1719933294765
var t2 int64 = 1719933294766
// Create a test server to mock Mixpanel API
testServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
// Check the request method and URL
require.Equal(t, http.MethodPost, r.Method)
expectedPath := "/track"
require.Equal(t, expectedPath, r.URL.Path)
queryParams := r.URL.Query()
require.Equal(t, "1", queryParams.Get("strict"))
require.Equal(t, "testAppID", queryParams.Get("project_id"))
// Check headers
require.Equal(t, "application/json", r.Header.Get("accept"))
require.Equal(t, "application/json", r.Header.Get("content-type"))
// Check request body
var metrics []mixpanelMetric
err := json.NewDecoder(r.Body).Decode(&metrics)
require.NoError(t, err)
require.Len(t, metrics, 2)
metric1 := metrics[0]
require.Equal(t, "user123", metric1.Properties.UserID)
require.Equal(t, "testSecret", metric1.Properties.Token)
require.Equal(t, "purchase", metric1.Event)
require.Equal(t, t1, metric1.Properties.Time)
require.Equal(t, map[string]interface{}{"price": 10.0}, metric1.Properties.AdditionalProperties)
metric2 := metrics[1]
require.Equal(t, "user123", metric2.Properties.UserID)
require.Equal(t, "testSecret", metric2.Properties.Token)
require.Equal(t, "purchase", metric2.Event)
require.Equal(t, t2, metric2.Properties.Time)
require.Equal(t, map[string]interface{}{"price": 11.0}, metric2.Properties.AdditionalProperties)
// Respond with 200 OK
w.WriteHeader(http.StatusOK)
}))
defer testServer.Close()
// Initialize the MixpanelMetricProcessor with the test server URL
processor := NewMixpanelMetricProcessor("testAppID", "testSecret", testServer.URL)
// Example metrics
metrics := []common.Metric{
{UserID: "user123", EventName: "purchase", EventValue: map[string]interface{}{"price": 10.0}, Timestamp: t1},
{UserID: "user123", EventName: "purchase", EventValue: map[string]interface{}{"price": 11.0}, Timestamp: t2},
}
// Process metrics
require.NoError(t, processor.Process(metrics))
}