Skip to content

Commit 5abcc08

Browse files
chore: first wave of unit tests
1 parent e9cd929 commit 5abcc08

File tree

1 file changed

+34
-0
lines changed

1 file changed

+34
-0
lines changed

tests/unit/test_resolver.py

Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,34 @@
1+
# Copyright 2024 Google LLC
2+
#
3+
# Licensed under the Apache License, Version 2.0 (the "License");
4+
# you may not use this file except in compliance with the License.
5+
# You may obtain a copy of the License at
6+
#
7+
# http://www.apache.org/licenses/LICENSE-2.0
8+
#
9+
# Unless required by applicable law or agreed to in writing, software
10+
# distributed under the License is distributed on an "AS IS" BASIS,
11+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
# See the License for the specific language governing permissions and
13+
# limitations under the License.
14+
15+
from google.cloud.sql.connector.connection_name import ConnectionName
16+
from google.cloud.sql.connector.resolver import DefaultResolver
17+
from google.cloud.sql.connector.resolver import DnsResolver
18+
19+
conn_str = "test-project:test-region:test-instance"
20+
conn_name = ConnectionName("test-project", "test-region", "test-instance")
21+
22+
23+
async def test_DefaultResolver() -> None:
24+
"""Test DefaultResolver just parses instance connection string."""
25+
resolver = DefaultResolver()
26+
result = await resolver.resolve(conn_str)
27+
assert result == conn_name
28+
29+
30+
async def test_DnsResolver_with_conn_str() -> None:
31+
"""Test DnsResolver with instance connection name just parses connection string."""
32+
resolver = DnsResolver()
33+
result = await resolver.resolve(conn_str)
34+
assert result == conn_name

0 commit comments

Comments
 (0)