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
|
package tooltest
import (
"context"
"os"
"strconv"
"testing"
"github.com/stretchr/testify/require"
"github.com/ydb-platform/ydb/library/go/core/log/nop"
"github.com/ydb-platform/ydb/library/go/yandex/tvm"
"github.com/ydb-platform/ydb/library/go/yandex/tvm/tvmauth"
)
const serviceTicketStr = "3:serv:CBAQ__________9_IgYIKhCWkQY:DnbhBOAMpunP9TuhCvXV8Hg9MEUHSFbRETf710eHVS7plghVsdM-JlLR6XtGeiofX3yiCFMs4Nq7aFJqZwX75HFgGiQymyWWKm2pWTyF0pp8QnaTivIM-Q6xmMqfInUlYrozhkVPmIxT4fqsdrKEACq-Zh8VtuNQYrTLZgsUfWo"
func recipeToolOptions(t *testing.T) tvmauth.TvmToolSettings {
var portStr, token []byte
portStr, err := os.ReadFile("tvmtool.port")
require.NoError(t, err)
var port int
port, err = strconv.Atoi(string(portStr))
require.NoError(t, err)
token, err = os.ReadFile("tvmtool.authtoken")
require.NoError(t, err)
return tvmauth.TvmToolSettings{Alias: "me", Port: port, AuthToken: string(token)}
}
func disableDstCheckOptions(t *testing.T) tvmauth.TvmToolSettings {
s := recipeToolOptions(t)
s.DisableDstCheck = true
return s
}
func TestToolClient(t *testing.T) {
c, err := tvmauth.NewToolClient(recipeToolOptions(t), &nop.Logger{})
require.NoError(t, err)
defer c.Destroy()
t.Run("GetServiceTicketForID", func(t *testing.T) {
_, err := c.GetServiceTicketForID(context.Background(), 100500)
require.NoError(t, err)
})
t.Run("GetInvalidTicket", func(t *testing.T) {
_, err := c.GetServiceTicketForID(context.Background(), 100999)
require.Error(t, err)
require.IsType(t, &tvm.Error{}, err)
require.Equal(t, tvm.ErrorBrokenTvmClientSettings, err.(*tvm.Error).Code)
})
t.Run("ClientStatus", func(t *testing.T) {
status, err := c.GetStatus(context.Background())
require.NoError(t, err)
t.Logf("Got client status: %v", status)
require.Equal(t, tvm.ClientStatus(0), status.Status)
require.Equal(t, "OK", status.LastError)
})
}
func TestDisableDstCheck(t *testing.T) {
c, err := tvmauth.NewToolClient(disableDstCheckOptions(t), &nop.Logger{})
require.NoError(t, err)
defer c.Destroy()
ticketS, err := c.CheckServiceTicket(context.Background(), serviceTicketStr)
require.NoError(t, err)
require.Equal(t, 100502, int(ticketS.DstID))
}
|
