def test_validate_registries(self):
with self.assertRaises(PolyaxonBuildException):
DockerBuilder(context=".",
destination="image:tag",
registries="foo")
with self.assertRaises(PolyaxonBuildException):
DockerBuilder(
context=".",
destination="image:tag",
registries=["foo", V1UriType("user", "pwd", "host")],
)
builder = DockerBuilder(
context=".",
destination="image:tag",
registries=[V1UriType("user", "pwd", "host")],
)
assert builder.registries is not None
def test_get_uri(self):
value = self.config.get_uri("uri_key_1")
self.assertEqual(value, V1UriType("user", "pass", "siteweb.ca"))
value = self.config.get_uri("uri_key_2")
self.assertEqual(value, V1UriType("user2", "pass", "localhost:8080"))
value = self.config.get_uri("uri_key_3")
self.assertEqual(value, V1UriType("user2", "pass", "https://quay.io"))
value = self.config.get_uri("uri_list_key_1", is_list=True)
self.assertEqual(
value,
[
V1UriType("user", "pass", "siteweb.ca"),
V1UriType("user2", "pass", "localhost:8080"),
V1UriType("user2", "pass", "https://quay.io"),
],
)
with self.assertRaises(PolyaxonSchemaError):
self.config.get_uri("uri_error_key_1")
with self.assertRaises(PolyaxonSchemaError):
self.config.get_uri("uri_error_key_2")
with self.assertRaises(PolyaxonSchemaError):
self.config.get_uri("uri_error_key_3")
with self.assertRaises(PolyaxonSchemaError):
self.config.get_uri("uri_error_key_4")
with self.assertRaises(PolyaxonSchemaError):
self.config.get_uri("uri_list_key_1")
with self.assertRaises(PolyaxonSchemaError):
self.config.get_uri("uri_list_error_key_1", is_list=True)
with self.assertRaises(PolyaxonSchemaError):
self.config.get_uri("uri_list_error_key_2", is_list=True)
with self.assertRaises(PolyaxonSchemaError):
self.config.get_uri("uri_list_error_key_3", is_list=True)
with self.assertRaises(PolyaxonSchemaError):
self.config.get_uri("uri_list_error_key_4", is_list=True)
with self.assertRaises(PolyaxonSchemaError):
self.config.get_uri("uri_key_1", is_list=True)
with self.assertRaises(PolyaxonSchemaError):
self.config.get_uri("uri_non_existing_key")
with self.assertRaises(PolyaxonSchemaError):
self.config.get_uri("uri_non_existing_key", is_list=True)
self.assertEqual(
self.config.get_uri("uri_non_existing_key", is_optional=True), None
)
self.assertEqual(
self.config.get_uri(
"uri_non_existing_key",
is_optional=True,
default=V1UriType("user2", "pass", "localhost:8080"),
),
V1UriType("user2", "pass", "localhost:8080"),
)
self.assertEqual(
self.config.get_uri("uri_non_existing_key", is_list=True, is_optional=True),
None,
)
self.assertEqual(
self.config.get_uri(
"uri_non_existing_key",
is_list=True,
is_optional=True,
default=[
V1UriType("user", "pass", "siteweb.ca"),
V1UriType("user2", "pass", "localhost:8080"),
],
),
[
V1UriType("user", "pass", "siteweb.ca"),
V1UriType("user2", "pass", "localhost:8080"),
],
)