import os

from django.test import TestCase

from dockerhub.telemetry.client import get_django_statsd, get_env_statsd


class ClientTestCase(TestCase):
    def test_get_django_statsd_with_default_values(self):
        statsd = get_django_statsd()
        self.assertEquals(statsd.host, 'localhost')
        self.assertEquals(statsd.port, 8125)
        self.assertIsNone(statsd.namespace)

    def test_get_django_statsd_with_settings(self):
        with self.settings(STATSD_HOST='myhost', STATSD_PORT='8000',
                           STATSD_PREFIX='hello'):
            statsd = get_django_statsd()
            self.assertEquals(statsd.host, 'myhost')
            self.assertEquals(statsd.port, 8000)
            self.assertEquals(statsd.namespace, 'hello')

    def test_get_env_statsd_with_default_values(self):
        statsd = get_env_statsd()
        self.assertEquals(statsd.host, 'localhost')
        self.assertEquals(statsd.port, 8125)

    def test_get_env_statsd_with_env_values(self):
        env_vars = {
            'STATSD_HOST': 'envhost',
            'STATSD_PORT': '1000',
            'STATSD_PREFIX': 'foo'
        }

        os.environ.update(env_vars)

        statsd = get_env_statsd()
        self.assertEquals(statsd.host, 'envhost')
        self.assertEquals(statsd.port, 1000)
        self.assertEquals(statsd.namespace, 'foo')

        for k in env_vars:
            del os.environ[k]