from django_prometheus.testutils import PrometheusTestCaseMixin
from django.test import TestCase
from django.db import connections


class TestDbMetrics(PrometheusTestCaseMixin, TestCase):
    """Test django_prometheus.db metrics.

    Note regarding the values of metrics: many tests interact with the
    database, and the test runner itself does. As such, tests that
    require that a metric has a specific value are at best very
    fragile. Consider asserting that the value exceeds a certain
    threshold, or check by how much it increased during the test.
    """
    def setUp(self):
        pass

    def testConfigHasExpectedDatabases(self):
        """Not a real unit test: ensures that testapp.settings contains the
        databases this test expects."""
        self.assertTrue('default' in connections.databases.keys())
        self.assertTrue('test_db_1' in connections.databases.keys())
        self.assertTrue('test_db_2' in connections.databases.keys())

    def testCounters(self):
        cursor_db1 = connections['test_db_1'].cursor()
        cursor_db2 = connections['test_db_2'].cursor()
        cursor_db1.execute('SELECT 1')
        for _ in range(200):
            cursor_db2.execute('SELECT 2')
        cursor_db1.execute('SELECT 3')
        try:
            cursor_db1.execute('this is clearly not valid SQL')
        except:
            pass

        self.assertTrue(self.getMetric(
            'django_db_new_connections_total',
            alias='default', vendor='sqlite') > 0)
        self.assertTrue(self.getMetric(
            'django_db_new_connections_total',
            alias='test_db_1', vendor='sqlite') > 0)
        self.assertTrue(self.getMetric(
            'django_db_new_connections_total',
            alias='test_db_2', vendor='sqlite') > 0)

        self.assertMetricEquals(
            1, 'django_db_errors_total',
            alias='test_db_1', vendor='sqlite', type='OperationalError')

        self.assertTrue(self.getMetric(
            'django_db_execute_total',
            alias='test_db_1', vendor='sqlite') > 0)
        self.assertTrue(self.getMetric(
            'django_db_execute_total',
            alias='test_db_2', vendor='sqlite') > 200)

    def testExecuteMany(self):
        cursor_db1 = connections['test_db_1'].cursor()
        cursor_db1.executemany(
            'INSERT INTO testapp_lawn(location) VALUES (?)', [
                ('Paris',), ('New York',), ('Berlin',), ('San Francisco',)])
        self.assertTrue(self.getMetric(
            'django_db_execute_many_total',
            alias='test_db_1', vendor='sqlite') >= 4)

from django_prometheus.testutils import PrometheusTestCaseMixin
from django.test import TestCase
from django.db import connections


class TestDbMetrics(PrometheusTestCaseMixin, TestCase):
    """Test django_prometheus.db metrics.

    Note regarding the values of metrics: many tests interact with the
    database, and the test runner itself does. As such, tests that
    require that a metric has a specific value are at best very
    fragile. Consider asserting that the value exceeds a certain
    threshold, or check by how much it increased during the test.
    """
    def setUp(self):
        pass

    def testConfigHasExpectedDatabases(self):
        """Not a real unit test: ensures that testapp.settings contains the
        databases this test expects."""
        self.assertTrue('default' in connections.databases.keys())
        self.assertTrue('test_db_1' in connections.databases.keys())
        self.assertTrue('test_db_2' in connections.databases.keys())

    def testCounters(self):
        cursor_db1 = connections['test_db_1'].cursor()
        cursor_db2 = connections['test_db_2'].cursor()
        cursor_db1.execute('SELECT 1')
        for _ in range(200):
            cursor_db2.execute('SELECT 2')
        cursor_db1.execute('SELECT 3')
        try:
            cursor_db1.execute('this is clearly not valid SQL')
        except:
            pass

        self.assertMetricEquals(
            1, 'django_db_errors_total',
            alias='test_db_1', vendor='sqlite', type='OperationalError')

        self.assertTrue(self.getMetric(
            'django_db_execute_total',
            alias='test_db_1', vendor='sqlite') > 0)
        self.assertTrue(self.getMetric(
            'django_db_execute_total',
            alias='test_db_2', vendor='sqlite') >= 200)

    def testExecuteMany(self):
        cursor_db1 = connections['test_db_1'].cursor()
        cursor_db1.executemany(
            'INSERT INTO testapp_lawn(location) VALUES (?)', [
                ('Paris',), ('New York',), ('Berlin',), ('San Francisco',)])
        self.assertTrue(self.getMetric(
            'django_db_execute_many_total',
            alias='test_db_1', vendor='sqlite') >= 4)

Remove the test for the db_new_connections_total metric.