package system

import (
	"context"
	"io"
	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/model"
	"net/http"
	"net/http/httptest"
	"testing"
	"time"
)

func TestQrcodeAPISuccessAndBodyTimeout(t *testing.T) {
	for _, scenario := range []string{"success", "header-timeout", "body-timeout", "partial-body-timeout"} {
		t.Run(scenario, func(t *testing.T) {
			release := make(chan struct{})
			server := httptest.NewServer(http.HandlerFunc(func(writer http.ResponseWriter, request *http.Request) {
				_, _ = io.Copy(io.Discard, request.Body)
				if scenario != "success" {
					if scenario != "header-timeout" {
						writer.Header().Set("Content-Length", "1024")
						writer.WriteHeader(http.StatusOK)
						if scenario == "partial-body-timeout" {
							_, _ = writer.Write([]byte(`{"code":0,"data":{"ticket":"partial"}}`))
						}
						writer.(http.Flusher).Flush()
					}
					select {
					case <-request.Context().Done():
					case <-release:
					}
					return
				}
				_, _ = writer.Write([]byte(`{"code":0,"data":{"ticket":"test-ticket","expireTime":300,"qrUrl":"test-url"}}`))
			}))
			defer server.Close()
			defer close(release)
			started := time.Now()
			result, err := callQrcodeApi(context.Background(), &SysQrcodeConfig{
				ServerApi: server.URL, Timeout: "70ms", Token: "test-token", SceneIdLogin: 1,
			}, true)
			if scenario != "success" {
				if err == nil || result != nil || time.Since(started) > time.Second {
					t.Fatalf("stalled response was not rejected: result=%v error=%v", result, err)
				}
			} else if err != nil || result == nil || result.Ticket != "test-ticket" {
				t.Fatalf("successful response changed: result=%v error=%v", result, err)
			}
		})
	}
}

func TestUserCodeHTTPTimeoutAndStatus(t *testing.T) {
	for _, scenario := range []string{"success", "rejected", "timeout"} {
		t.Run(scenario, func(t *testing.T) {
			release := make(chan struct{})
			server := httptest.NewServer(http.HandlerFunc(func(writer http.ResponseWriter, request *http.Request) {
				_, _ = io.Copy(io.Discard, request.Body)
				if scenario == "timeout" {
					select {
					case <-request.Context().Done():
					case <-release:
					}
					return
				}
				if scenario == "rejected" {
					writer.WriteHeader(http.StatusServiceUnavailable)
					return
				}
				writer.WriteHeader(http.StatusAccepted)
			}))
			defer server.Close()
			defer close(release)
			started := time.Now()
			err := callUserCodeHTTP(context.Background(), model.UserCodeHttpSenderConfig{
				Url: server.URL, Timeout: "70ms", SuccessStatus: http.StatusAccepted,
			}, "xxx@example.com", "123456")
			if (err != nil) != (scenario != "success") || time.Since(started) > time.Second {
				t.Fatalf("unexpected sender result: %v", err)
			}
		})
	}
}