from unittest import TestCase

from dvc.remote.azure import RemoteAzure


class TestRemoteAzure(TestCase):
    container_name = "container-name"
    connection_string = (
        "DefaultEndpointsProtocol=http;AccountName=devstoreaccount1;"
        "AccountKey=Eby8vdM02xNOcqFlqUwJPLlmEtlCDXJ1OUzFT50uSRZ6IFsu"
        "Fq2UVErCz4I6tq/K1SZFPTOtr/KBHBeksoGMGw==;"
        "BlobEndpoint=http://127.0.0.1:10000/devstoreaccount1;"
    )

    def test_init_compat(self):
        url = (
            "azure://ContainerName={container_name};{connection_string}"
        ).format(
            container_name=self.container_name,
            connection_string=self.connection_string,
        )
        config = {"url": url}
        remote = RemoteAzure(None, config)
        self.assertEqual(remote.url, url)
        self.assertEqual(remote.prefix, "")
        self.assertEqual(remote.bucket, self.container_name)
        self.assertEqual(remote.connection_string, self.connection_string)

    def test_init(self):
        prefix = "some/prefix"
        url = "azure://{}/{}".format(self.container_name, prefix)
        config = {"url": url, "connection_string": self.connection_string}
        remote = RemoteAzure(None, config)
        self.assertEqual(remote.url, url)
        self.assertEqual(remote.prefix, prefix)
        self.assertEqual(remote.bucket, self.container_name)
        self.assertEqual(remote.connection_string, self.connection_string)