device_posture_test.go

// Copyright (c) David Bond, Tailscale Inc, & Contributors
// SPDX-License-Identifier: MIT

package tailscale

import (
	"context"
	"encoding/json"
	"net/http"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestClient_DevicePosture_CreateIntegration(t *testing.T) {
	t.Parallel()

	client, server := NewTestHarness(t)
	server.ResponseCode = http.StatusOK

	req := CreatePostureIntegrationRequest{
		Provider:     PostureIntegrationProviderIntune,
		CloudID:      "cloudid",
		ClientID:     "clientid",
		TenantID:     "tenantid",
		ClientSecret: "clientsecret",
	}

	resp := &PostureIntegration{
		ID:       "1",
		Provider: PostureIntegrationProviderIntune,
		CloudID:  "cloudid",
		ClientID: "clientid",
		TenantID: "tenantid",
	}
	server.ResponseBody = resp

	integration, err := client.DevicePosture().CreateIntegration(context.Background(), req)
	require.NoError(t, err)
	assert.Equal(t, http.MethodPost, server.Method)
	assert.Equal(t, "/api/v2/tailnet/example.com/posture/integrations", server.Path)
	assert.Equal(t, resp, integration)

	var actualRequest CreatePostureIntegrationRequest
	err = json.Unmarshal(server.Body.Bytes(), &actualRequest)
	require.NoError(t, err)
	assert.Equal(t, req, actualRequest)
}

func TestClient_DevicePosture_UpdateIntegration(t *testing.T) {
	t.Parallel()

	client, server := NewTestHarness(t)
	server.ResponseCode = http.StatusOK

	req := UpdatePostureIntegrationRequest{
		CloudID:      "cloudid",
		ClientID:     "clientid",
		TenantID:     "tenantid",
		ClientSecret: PointerTo("clientsecret"),
	}

	resp := &PostureIntegration{
		ID:       "1",
		Provider: PostureIntegrationProviderIntune,
		CloudID:  "cloudid",
		ClientID: "clientid",
		TenantID: "tenantid",
	}
	server.ResponseBody = resp

	actualResp, err := client.DevicePosture().UpdateIntegration(context.Background(), "1", req)
	assert.NoError(t, err)
	assert.Equal(t, http.MethodPatch, server.Method)
	assert.Equal(t, "/api/v2/posture/integrations/1", server.Path)
	assert.Equal(t, resp, actualResp)

	var actualRequest UpdatePostureIntegrationRequest
	err = json.Unmarshal(server.Body.Bytes(), &actualRequest)
	require.NoError(t, err)
	assert.Equal(t, req, actualRequest)
}

func TestClient_DevicePosture_DeleteIntegration(t *testing.T) {
	t.Parallel()

	client, server := NewTestHarness(t)
	server.ResponseCode = http.StatusOK

	err := client.DevicePosture().DeleteIntegration(context.Background(), "1")
	assert.NoError(t, err)
	assert.Equal(t, http.MethodDelete, server.Method)
	assert.Equal(t, "/api/v2/posture/integrations/1", server.Path)
}

func TestClient_DevicePosture_GetIntegration(t *testing.T) {
	t.Parallel()

	client, server := NewTestHarness(t)
	server.ResponseCode = http.StatusOK

	resp := &PostureIntegration{
		ID:       "1",
		Provider: PostureIntegrationProviderIntune,
		CloudID:  "cloudid1",
		ClientID: "clientid1",
		TenantID: "tenantid1",
	}
	server.ResponseBody = resp

	actualResp, err := client.DevicePosture().GetIntegration(context.Background(), "1")
	assert.NoError(t, err)
	assert.Equal(t, http.MethodGet, server.Method)
	assert.Equal(t, "/api/v2/posture/integrations/1", server.Path)
	assert.Equal(t, resp, actualResp)
}

func TestClient_DevicePosture_ListIntegrations(t *testing.T) {
	t.Parallel()

	client, server := NewTestHarness(t)
	server.ResponseCode = http.StatusOK

	resp := []PostureIntegration{
		{
			ID:       "1",
			Provider: PostureIntegrationProviderIntune,
			CloudID:  "cloudid1",
			ClientID: "clientid1",
			TenantID: "tenantid1",
		},
		{
			ID:       "2",
			Provider: PostureIntegrationProviderJamfPro,
			CloudID:  "cloudid2",
			ClientID: "clientid2",
			TenantID: "tenantid2",
		},
	}
	server.ResponseBody = map[string][]PostureIntegration{
		"integrations": resp,
	}

	actualResp, err := client.DevicePosture().ListIntegrations(context.Background())
	assert.NoError(t, err)
	assert.Equal(t, http.MethodGet, server.Method)
	assert.Equal(t, "/api/v2/tailnet/example.com/posture/integrations", server.Path)
	assert.Equal(t, resp, actualResp)
}