cli/acceptance/cmd/psql/argument-errors/test.toml at 681a0293c8043e792db91f30a5bf0db193a01b6c · databricks/cli · 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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
# This acceptance test is disabled on Windows runners because
# the current argument capturing method does not work on windows-latest GitHub Runner.
GOOS.windows = false

Ignore = [
    "psql"
]

# GetProject response
[[Server]]
Pattern = "GET /api/2.0/postgres/projects/my-project"
Response.Body = '''
{
    "name": "projects/my-project",
    "status": { "display_name": "My Project" }
}
'''

# Provisioned instance lookup
[[Server]]
Pattern = "GET /api/2.0/database/instances/my-instance"
Response.Body = '''
{
    "state": "AVAILABLE",
    "pg_version": "14",
    "read_write_dns": "my-instance.example.com"
}
'''

# GetBranch responses
[[Server]]
Pattern = "GET /api/2.0/postgres/projects/my-project/branches/main"
Response.Body = '''
{
    "name": "projects/my-project/branches/main"
}
'''

[[Server]]
Pattern = "GET /api/2.0/postgres/projects/my-project/branches/dev"
Response.Body = '''
{
    "name": "projects/my-project/branches/dev"
}
'''

# Full endpoint for supplement tests
[[Server]]
Pattern = "GET /api/2.0/postgres/projects/my-project/branches/main/endpoints/primary"
Response.Body = '''
{
    "name": "projects/my-project/branches/main/endpoints/primary",
    "status": {
        "current_state": "ACTIVE",
        "endpoint_type": "ENDPOINT_TYPE_READ_WRITE",
        "hosts": {
            "host": "my-endpoint.postgres.example.com"
        }
    }
}
'''

# Partial path supplement: project only -> needs branches list
[[Server]]
Pattern = "GET /api/2.0/postgres/projects/my-project/branches"
Response.Body = '''
{
    "branches": [
        {
            "name": "projects/my-project/branches/main",
            "status": { "default": true }
        }
    ]
}
'''

[[Server]]
Pattern = "GET /api/2.0/postgres/projects/my-project/branches/main/endpoints"
Response.Body = '''
{
    "endpoints": [
        {
            "name": "projects/my-project/branches/main/endpoints/primary",
            "status": {
                "current_state": "ACTIVE",
                "endpoint_type": "ENDPOINT_TYPE_READ_WRITE",
                "hosts": { "host": "my-endpoint.postgres.example.com" }
            }
        }
    ]
}
'''

# Endpoint for supplement test with --endpoint flag
[[Server]]
Pattern = "GET /api/2.0/postgres/projects/my-project/branches/dev/endpoints/custom"
Response.Body = '''
{
    "name": "projects/my-project/branches/dev/endpoints/custom",
    "status": {
        "current_state": "ACTIVE",
        "endpoint_type": "ENDPOINT_TYPE_READ_WRITE",
        "hosts": {
            "host": "custom-endpoint.postgres.example.com"
        }
    }
}
'''

# Credential generation
[[Server]]
Pattern = "POST /api/2.0/postgres/credentials"
Response.Body = '''
{
    "token": "postgres-secret-token",
    "expire_time": "2025-01-01T00:00:00Z"
}
'''

[[Server]]
Pattern = "POST /api/2.0/database/credentials"
Response.Body = '''
{
    "token": "provisioned-secret-token"
}
'''