schema/main_test.go at main · Scorify/schema · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
package schema

import (
	"testing"

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

func ptr[v any](obj v) *v {
	return &obj
}

type exampleDnsSchema struct {
	Server         string `key:"server"`
	Port           int    `key:"port" default:"53"`
	LookupType     string `key:"lookup_type" default:"A" enum:"A,AAAA,CNAME,MX,NS,PTR,SOA,TXT"`
	Domain         string `key:"domain"`
	ExpectedOutput string `key:"expected_output"`
}

var exampleDnsSchemaFields = []*Field{
	{Key: "server", Type: FieldString},
	{Key: "port", Type: FieldInt, Default: ptr("53")},
	{Key: "lookup_type", Type: FieldString, Default: ptr("A"), Enum: &[]string{"A", "AAAA", "CNAME", "MX", "NS", "PTR", "SOA", "TXT"}},
	{Key: "domain", Type: FieldString},
	{Key: "expected_output", Type: FieldString},
}

var exampleDnsSchemaStruct = exampleDnsSchema{
	Server:         "1.1.1.1",
	Port:           53,
	LookupType:     "A",
	Domain:         "one.one.one.one",
	ExpectedOutput: "1.1.1.1",
}

const exampleDnsSchemaString = `{
	"server": "1.1.1.1",
	"port": 53,
	"lookup_type": "A",
	"domain": "one.one.one.one",
	"expected_output": "1.1.1.1"
}`

func TestMarshal(t *testing.T) {
	data, err := Marshal(exampleDnsSchemaStruct)
	assert.NoError(t, err)
	assert.JSONEq(t, exampleDnsSchemaString, string(data))
}

func TestUnmarshal(t *testing.T) {
	newDNSSchema := exampleDnsSchema{}
	err := Unmarshal([]byte(exampleDnsSchemaString), &newDNSSchema)
	assert.NoError(t, err)
	assert.Equal(t, exampleDnsSchemaStruct, newDNSSchema)
}

func TestDescribe(t *testing.T) {
	fields, err := Describe(exampleDnsSchemaStruct)
	assert.NoError(t, err)
	assert.Equal(t, exampleDnsSchemaFields, fields)
}