import kojihub
from .utils import DBQueryTestCase


class TestGetMavenArchive(DBQueryTestCase):

    def setUp(self):
        super(TestGetMavenArchive, self).setUp()
        self.maxDiff = None

    def test_valid(self):
        kojihub.get_maven_archive(123)
        self.assertEqual(len(self.queries), 1)
        query = self.queries[0]
        self.assertEqual(query.tables, ['maven_archives'])
        self.assertEqual(query.joins, None)
        self.assertEqual(query.clauses, ['archive_id = %(archive_id)i'])
        self.assertEqual(query.values, {'archive_id': 123})
        self.assertEqual(query.columns, ['archive_id', 'artifact_id', 'group_id', 'version'])