JSON-RPC-M1/ResponseMetaData_test.go at main · vault-thirteen/JSON-RPC-M1 · 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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
package jrm1

import (
	"fmt"
	"testing"

	"github.com/vault-thirteen/auxie/tester"
)

func Test_ResponseMetaData_AddField(t *testing.T) {
	aTest := tester.New(t)
	var md ResponseMetaData
	var err error

	// Test #1. Normal addition.
	md = make(ResponseMetaData)
	err = md.AddField("key", "value")
	aTest.MustBeNoError(err)
	aTest.MustBeEqual(md["key"], "value")

	// Test #2. Duplicate key.
	md = make(ResponseMetaData)
	err = md.AddField("key", "value-1")
	aTest.MustBeNoError(err)
	err = md.AddField("key", "value-2")
	aTest.MustBeAnError(err)
}

// _newRpcErrorFastWrapper runs the 'NewRpcErrorFast' and stops panic.
func _addFieldFastWrapper(f func(key string, value any), key string, value any) (hasException bool) {
	defer func() {
		x := recover()
		if x != nil {
			hasException = true
			fmt.Println(fmt.Sprintf("An exception was captured: %v", x))
		}
	}()

	f(key, value)

	return false
}

func Test_ResponseMetaData_AddFieldFast(t *testing.T) {
	aTest := tester.New(t)
	var hasException bool
	var md ResponseMetaData

	// Test #1. No panic.
	md = make(ResponseMetaData)
	hasException = _addFieldFastWrapper(md.AddFieldFast, "a", 123)
	aTest.MustBeEqual(hasException, false)

	// Test #2. Panic.
	md = make(ResponseMetaData)
	hasException = _addFieldFastWrapper(md.AddFieldFast, "a", 123)
	aTest.MustBeEqual(hasException, false)
	hasException = _addFieldFastWrapper(md.AddFieldFast, "a", 456)
	aTest.MustBeEqual(hasException, true)
}

func Test_ResponseMetaData_GetField(t *testing.T) {
	aTest := tester.New(t)
	var md ResponseMetaData
	var err error

	// Test.
	md = make(ResponseMetaData)
	err = md.AddField("key", "value")
	aTest.MustBeNoError(err)
	aTest.MustBeEqual(md.GetField("key"), "value")
}

func Test_ResponseMetaData_RemoveField(t *testing.T) {
	aTest := tester.New(t)
	var md ResponseMetaData
	var err error

	// Test #1. Normal deletion.
	md = make(ResponseMetaData)
	err = md.AddField("key", "value")
	aTest.MustBeNoError(err)
	aTest.MustBeEqual(md["key"], "value")
	err = md.RemoveField("key")
	aTest.MustBeNoError(err)
	aTest.MustBeEqual(md["key"], nil)

	// Test #2. Absent key.
	md = make(ResponseMetaData)
	err = md.AddField("key", "value")
	aTest.MustBeNoError(err)
	err = md.RemoveField("key")
	aTest.MustBeNoError(err)
	err = md.RemoveField("key")
	aTest.MustBeAnError(err)
}