kafka_schemas_test.go

package aiven

import (
	"context"
	"encoding/json"
	"net/http"
	"net/http/httptest"
	"reflect"
	"testing"
)

func setupKafkaSchemasTestCase(t *testing.T) (*Client, func(t *testing.T)) {
	t.Log("setup Kafka Schemas test case")

	const (
		UserName     = "test@aiven.io"
		UserPassword = "testabcd"
		AccessToken  = "some-random-token"
	)

	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		// auth
		if r.URL.Path == "/userauth" {
			w.Header().Set("Content-Type", "application/json")
			w.WriteHeader(http.StatusOK)
			err := json.NewEncoder(w).Encode(authResponse{
				Token: AccessToken,
				State: "active",
			})

			if err != nil {
				t.Error(err)
			}
			return
		}

		// config
		if r.URL.Path == "/project/test-pr/service/test-sr/kafka/schema/config" {
			if r.Method == "PUT" {
				w.Header().Set("Content-Type", "application/json")
				w.WriteHeader(http.StatusOK)
				err := json.NewEncoder(w).Encode(KafkaSchemaConfigUpdateResponse{
					APIResponse{},
					KafkaSchemaConfig{CompatibilityLevel: "FULL"},
				})

				if err != nil {
					t.Error(err)
				}
			}

			if r.Method == "GET" {
				w.Header().Set("Content-Type", "application/json")
				w.WriteHeader(http.StatusOK)
				err := json.NewEncoder(w).Encode(KafkaSchemaConfigResponse{
					CompatibilityLevel: "FULL",
				})

				if err != nil {
					t.Error(err)
				}
			}

			return
		}

		// config
		if r.URL.Path == "/project/test-pr/service/test-sr/kafka/schema/config" {
			w.Header().Set("Content-Type", "application/json")
			w.WriteHeader(http.StatusOK)
			err := json.NewEncoder(w).Encode(KafkaSchemaConfigResponse{
				CompatibilityLevel: "FULL",
			})

			if err != nil {
				t.Error(err)
			}

			return
		}

		// subjects
		if r.URL.Path == "/project/test-pr/service/test-sr/kafka/schema/subjects" {
			w.Header().Set("Content-Type", "application/json")
			w.WriteHeader(http.StatusOK)
			err := json.NewEncoder(w).Encode(KafkaSchemaSubjectsResponse{
				APIResponse{},
				KafkaSchemaSubjects{Subjects: []string{"testSb1", "testSb2"}},
			})

			if err != nil {
				t.Error(err)
			}

			return
		}

		// add subject no versions
		if r.URL.Path == "/project/test-pr/service/test-sr/kafka/schema/subjects/test-schema-no-versions/versions" {
			if r.Method == "GET" {
				w.Header().Set("Content-Type", "application/json")
				w.WriteHeader(http.StatusNotFound)

				return
			}

			if r.Method == "POST" {
				w.Header().Set("Content-Type", "application/json")
				w.WriteHeader(http.StatusOK)

				err := json.NewEncoder(w).Encode(KafkaSchemaSubjectResponse{
					APIResponse{},
					1,
				})

				if err != nil {
					t.Error(err)
				}

				return
			}
		}

		// add subject has versions
		if r.URL.Path == "/project/test-pr/service/test-sr/kafka/schema/subjects/test-schema/versions" {
			if r.Method == "GET" {
				w.Header().Set("Content-Type", "application/json")
				w.WriteHeader(http.StatusOK)

				err := json.NewEncoder(w).Encode(KafkaSchemaSubjectVersionsResponse{
					APIResponse{},
					KafkaSchemaSubjectVersions{Versions: []int{1, 2, 3, 4}},
				})

				if err != nil {
					t.Error(err)
				}

				return
			}

			if r.Method == "POST" {
				w.Header().Set("Content-Type", "application/json")
				w.WriteHeader(http.StatusOK)

				err := json.NewEncoder(w).Encode(KafkaSchemaSubjectResponse{
					APIResponse{},
					5,
				})

				if err != nil {
					t.Error(err)
				}

				return
			}
		}

		// validate against version 4
		if r.URL.Path == "/project/test-pr/service/test-sr/kafka/schema/compatibility/subjects/test-schema/versions/4" {
			w.Header().Set("Content-Type", "application/json")
			w.WriteHeader(http.StatusOK)
			err := json.NewEncoder(w).Encode(KafkaSchemaValidateResponse{
				APIResponse{},
				true,
			})

			if err != nil {
				t.Error(err)
			}

			return
		}

		if r.URL.Path == "/project/test-pr/service/test-sr/kafka/schema/subjects/test-schema/versions/5" {
			if r.Method == "DELETE" {
				w.Header().Set("Content-Type", "application/json")
				w.WriteHeader(http.StatusOK)

				err := json.NewEncoder(w).Encode(APIResponse{})

				if err != nil {
					t.Error(err)
				}

				return
			}

			if r.Method == "GET" {
				w.Header().Set("Content-Type", "application/json")
				w.WriteHeader(http.StatusOK)

				err := json.NewEncoder(w).Encode(KafkaSchemaSubjectVersionResponse{
					APIResponse: APIResponse{},
					Version: KafkaSchemaSubjectVersion{
						Id:      5,
						Schema:  "",
						Subject: "est-schema",
						Version: 5,
					},
				})

				if err != nil {
					t.Error(err)
				}

				return
			}
		}

	}))

	apiUrl = ts.URL
	c, err := NewUserClient(UserName, UserPassword, "aiven-go-client-test/"+Version())
	if err != nil {
		t.Fatalf("user authentication error: %s", err)
	}

	return c, func(t *testing.T) {
		t.Log("teardown ElasticsearchACLs test case")
		ts.Close()
	}
}

func TestKafkaGlobalSchemaConfigHandler_Update(t *testing.T) {
	c, tearDown := setupKafkaSchemasTestCase(t)
	defer tearDown(t)

	ctx := context.Background()

	type fields struct {
		client *Client
	}
	type args struct {
		project string
		service string
		c       KafkaSchemaConfig
	}
	tests := []struct {
		name    string
		fields  fields
		args    args
		want    *KafkaSchemaConfigUpdateResponse
		wantErr bool
	}{
		{
			"",
			fields{client: c},
			args{
				project: "test-pr",
				service: "test-sr",
				c: KafkaSchemaConfig{
					CompatibilityLevel: "FULL",
				},
			},
			&KafkaSchemaConfigUpdateResponse{
				APIResponse: APIResponse{},
				KafkaSchemaConfig: KafkaSchemaConfig{
					CompatibilityLevel: "FULL",
				},
			},
			false,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			h := &KafkaGlobalSchemaConfigHandler{
				client: tt.fields.client,
			}
			got, err := h.Update(ctx, tt.args.project, tt.args.service, tt.args.c)
			if (err != nil) != tt.wantErr {
				t.Errorf("UpdateConfig() error = %v, wantErr %v", err, tt.wantErr)
				return
			}
			if !reflect.DeepEqual(got, tt.want) {
				t.Errorf("UpdateConfig() got = %v, want %v", got, tt.want)
			}
		})
	}
}

func TestKafkaSchemaHandler_Add(t *testing.T) {
	c, tearDown := setupKafkaSchemasTestCase(t)
	defer tearDown(t)

	ctx := context.Background()

	schema := `
		{
				"doc": "example",
				"fields": [{
					"default": 5,
					"doc": "my test number",
					"name": "test",
					"namespace": "test",
					"type": "int"
				}],
				"name": "example",
				"namespace": "example",
				"type": "record"
			}`

	type fields struct {
		client *Client
	}
	type args struct {
		project string
		service string
		name    string
		subject KafkaSchemaSubject
	}
	tests := []struct {
		name    string
		fields  fields
		args    args
		want    *KafkaSchemaSubjectResponse
		wantErr bool
	}{
		{
			"no-versions",
			fields{client: c},
			args{
				project: "test-pr",
				service: "test-sr",
				name:    "test-schema-no-versions",
				subject: KafkaSchemaSubject{Schema: schema},
			},
			&KafkaSchemaSubjectResponse{
				Id: 1,
			},
			false,
		},
		{
			"has-versions",
			fields{client: c},
			args{
				project: "test-pr",
				service: "test-sr",
				name:    "test-schema",
				subject: KafkaSchemaSubject{Schema: schema},
			},
			&KafkaSchemaSubjectResponse{
				Id: 5,
			},
			false,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			h := &KafkaSubjectSchemasHandler{
				client: tt.fields.client,
			}
			got, err := h.Add(ctx, tt.args.project, tt.args.service, tt.args.name, tt.args.subject)
			if (err != nil) != tt.wantErr {
				t.Errorf("Add() error = %v, wantErr %v", err, tt.wantErr)
				return
			}
			if !reflect.DeepEqual(got, tt.want) {
				t.Errorf("Add() got = %v, want %v", got, tt.want)
			}
		})
	}
}

func TestKafkaSchemaHandler_Delete(t *testing.T) {
	c, tearDown := setupKafkaSchemasTestCase(t)
	defer tearDown(t)

	ctx := context.Background()

	type fields struct {
		client *Client
	}
	type args struct {
		project string
		service string
		name    string
		version int
	}
	tests := []struct {
		name    string
		fields  fields
		args    args
		wantErr bool
	}{
		{
			"",
			fields{client: c},
			args{
				project: "test-pr",
				service: "test-sr",
				name:    "test-schema",
				version: 5,
			},
			false,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			h := &KafkaSubjectSchemasHandler{
				client: tt.fields.client,
			}
			if err := h.Delete(ctx, tt.args.project, tt.args.service, tt.args.name, tt.args.version); (err != nil) != tt.wantErr {
				t.Errorf("Delete() error = %v, wantErr %v", err, tt.wantErr)
			}
		})
	}
}

func TestKafkaGlobalSchemaConfigHandler_Get(t *testing.T) {
	c, tearDown := setupKafkaSchemasTestCase(t)
	defer tearDown(t)

	ctx := context.Background()

	type fields struct {
		client *Client
	}
	type args struct {
		project string
		service string
	}
	tests := []struct {
		name    string
		fields  fields
		args    args
		want    *KafkaSchemaConfigResponse
		wantErr bool
	}{
		{
			"",
			fields{client: c},
			args{
				project: "test-pr",
				service: "test-sr",
			},
			&KafkaSchemaConfigResponse{
				APIResponse:        APIResponse{},
				CompatibilityLevel: "FULL",
			},
			false,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			h := &KafkaGlobalSchemaConfigHandler{
				client: tt.fields.client,
			}
			got, err := h.Get(ctx, tt.args.project, tt.args.service)
			if (err != nil) != tt.wantErr {
				t.Errorf("GetConfig() error = %v, wantErr %v", err, tt.wantErr)
				return
			}
			if !reflect.DeepEqual(got, tt.want) {
				t.Errorf("GetConfig() got = %v, want %v", got, tt.want)
			}
		})
	}
}

func TestKafkaSchemaHandler_Get(t *testing.T) {
	c, tearDown := setupKafkaSchemasTestCase(t)
	defer tearDown(t)

	ctx := context.Background()

	type fields struct {
		client *Client
	}
	type args struct {
		project string
		service string
		name    string
		version int
	}
	tests := []struct {
		name    string
		fields  fields
		args    args
		want    *KafkaSchemaSubjectVersionResponse
		wantErr bool
	}{
		{
			"",
			fields{client: c},
			args{
				project: "test-pr",
				service: "test-sr",
				name:    "test-schema",
				version: 5,
			},
			&KafkaSchemaSubjectVersionResponse{
				APIResponse: APIResponse{},
				Version: KafkaSchemaSubjectVersion{
					Id:      5,
					Schema:  "",
					Subject: "est-schema",
					Version: 5,
				},
			},
			false,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			h := &KafkaSubjectSchemasHandler{
				client: tt.fields.client,
			}
			got, err := h.Get(ctx, tt.args.project, tt.args.service, tt.args.name, tt.args.version)
			if (err != nil) != tt.wantErr {
				t.Errorf("Get() error = %v, wantErr %v", err, tt.wantErr)
				return
			}
			if !reflect.DeepEqual(got, tt.want) {
				t.Errorf("Get() got = %v, want %v", got, tt.want)
			}
		})
	}
}

func TestKafkaSchemaHandler_GetVersions(t *testing.T) {
	c, tearDown := setupKafkaSchemasTestCase(t)
	defer tearDown(t)

	ctx := context.Background()

	type fields struct {
		client *Client
	}
	type args struct {
		project string
		service string
		name    string
	}
	tests := []struct {
		name    string
		fields  fields
		args    args
		want    *KafkaSchemaSubjectVersionsResponse
		wantErr bool
	}{
		{
			"",
			fields{client: c},
			args{
				project: "test-pr",
				service: "test-sr",
				name:    "test-schema",
			},
			&KafkaSchemaSubjectVersionsResponse{
				APIResponse: APIResponse{},
				KafkaSchemaSubjectVersions: KafkaSchemaSubjectVersions{
					Versions: []int{1, 2, 3, 4},
				},
			},
			false,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			h := &KafkaSubjectSchemasHandler{
				client: tt.fields.client,
			}
			got, err := h.GetVersions(ctx, tt.args.project, tt.args.service, tt.args.name)
			if (err != nil) != tt.wantErr {
				t.Errorf("GetVersions() error = %v, wantErr %v", err, tt.wantErr)
				return
			}
			if !reflect.DeepEqual(got, tt.want) {
				t.Errorf("GetVersions() got = %v, want %v", got, tt.want)
			}
		})
	}
}

func TestKafkaSchemaHandler_List(t *testing.T) {
	c, tearDown := setupKafkaSchemasTestCase(t)
	defer tearDown(t)

	ctx := context.Background()

	type fields struct {
		client *Client
	}
	type args struct {
		project string
		service string
	}
	tests := []struct {
		name    string
		fields  fields
		args    args
		want    *KafkaSchemaSubjectsResponse
		wantErr bool
	}{
		{
			"",
			fields{client: c},
			args{
				project: "test-pr",
				service: "test-sr",
			},
			&KafkaSchemaSubjectsResponse{
				APIResponse: APIResponse{},
				KafkaSchemaSubjects: KafkaSchemaSubjects{
					Subjects: []string{"testSb1", "testSb2"},
				},
			},
			false,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			h := &KafkaSubjectSchemasHandler{
				client: tt.fields.client,
			}
			got, err := h.List(ctx, tt.args.project, tt.args.service)
			if (err != nil) != tt.wantErr {
				t.Errorf("List() error = %v, wantErr %v", err, tt.wantErr)
				return
			}
			if !reflect.DeepEqual(got, tt.want) {
				t.Errorf("List() got = %v, want %v", got, tt.want)
			}
		})
	}
}

func TestKafkaSchemaHandler_Validate(t *testing.T) {
	c, tearDown := setupKafkaSchemasTestCase(t)
	defer tearDown(t)

	ctx := context.Background()

	type fields struct {
		client *Client
	}
	type args struct {
		project string
		service string
		name    string
		version int
		subject KafkaSchemaSubject
	}
	tests := []struct {
		name    string
		fields  fields
		args    args
		want    bool
		wantErr bool
	}{
		{
			"",
			fields{client: c},
			args{
				project: "test-pr",
				service: "test-sr",
				name:    "test-schema",
				version: 4,
				subject: KafkaSchemaSubject{Schema: `
				{
					"doc": "example",
					"fields": [{
						"default": 5,
						"doc": "my test number",
						"name": "test",
						"namespace": "test",
						"type": "int"
					}],
					"name": "example",
					"namespace": "example",
					"type": "record"
				}`},
			},
			true,
			false,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			h := &KafkaSubjectSchemasHandler{
				client: tt.fields.client,
			}
			got, err := h.Validate(ctx, tt.args.project, tt.args.service, tt.args.name, tt.args.version, tt.args.subject)
			if (err != nil) != tt.wantErr {
				t.Errorf("Validate() error = %v, wantErr %v", err, tt.wantErr)
				return
			}
			if got != tt.want {
				t.Errorf("Validate() got = %v, want %v", got, tt.want)
			}
		})
	}
}