From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: Received: from lists.gentoo.org (pigeon.gentoo.org [208.92.234.80]) by finch.gentoo.org (Postfix) with ESMTP id B2DFD1381F3 for ; Wed, 18 Sep 2013 00:57:27 +0000 (UTC) Received: from pigeon.gentoo.org (localhost [127.0.0.1]) by pigeon.gentoo.org (Postfix) with SMTP id A660BE0AE0; Wed, 18 Sep 2013 00:57:26 +0000 (UTC) Received: from smtp.gentoo.org (smtp.gentoo.org [140.211.166.183]) (using TLSv1 with cipher AECDH-AES256-SHA (256/256 bits)) (No client certificate requested) by pigeon.gentoo.org (Postfix) with ESMTPS id 2D09CE0AE0 for ; Wed, 18 Sep 2013 00:57:25 +0000 (UTC) Received: from hornbill.gentoo.org (hornbill.gentoo.org [94.100.119.163]) (using TLSv1 with cipher AECDH-AES256-SHA (256/256 bits)) (No client certificate requested) by smtp.gentoo.org (Postfix) with ESMTPS id BEE4733ED52 for ; Wed, 18 Sep 2013 00:57:24 +0000 (UTC) Received: from localhost.localdomain (localhost [127.0.0.1]) by hornbill.gentoo.org (Postfix) with ESMTP id 6FFE5E530A for ; Wed, 18 Sep 2013 00:57:23 +0000 (UTC) From: "Jauhien Piatlicki" To: gentoo-commits@lists.gentoo.org Content-Transfer-Encoding: 8bit Content-type: text/plain; charset=UTF-8 Reply-To: gentoo-dev@lists.gentoo.org, "Jauhien Piatlicki" Message-ID: <1379465832.b3e47360bd5ccb871e0860c24220259d4fbb3da0.jauhien@gentoo> Subject: [gentoo-commits] proj/g-sorcery:master commit in: tests/ X-VCS-Repository: proj/g-sorcery X-VCS-Files: tests/test_DBGenerator.py X-VCS-Directories: tests/ X-VCS-Committer: jauhien X-VCS-Committer-Name: Jauhien Piatlicki X-VCS-Revision: b3e47360bd5ccb871e0860c24220259d4fbb3da0 X-VCS-Branch: master Date: Wed, 18 Sep 2013 00:57:23 +0000 (UTC) Precedence: bulk List-Post: List-Help: List-Unsubscribe: List-Subscribe: List-Id: Gentoo Linux mail X-BeenThere: gentoo-commits@lists.gentoo.org X-Archives-Salt: fc5ebee0-6ea4-4574-ae08-6b3200f6dbec X-Archives-Hash: 20d8b4d162a838cefcc3bac20c43fa72 commit: b3e47360bd5ccb871e0860c24220259d4fbb3da0 Author: Jauhien Piatlicki (jauhien) gmail com> AuthorDate: Wed Sep 18 00:57:12 2013 +0000 Commit: Jauhien Piatlicki gmail com> CommitDate: Wed Sep 18 00:57:12 2013 +0000 URL: http://git.overlays.gentoo.org/gitweb/?p=proj/g-sorcery.git;a=commit;h=b3e47360 tests/test_DBGenerator: initial commit --- tests/test_DBGenerator.py | 110 ++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 110 insertions(+) diff --git a/tests/test_DBGenerator.py b/tests/test_DBGenerator.py new file mode 100644 index 0000000..b5e40f8 --- /dev/null +++ b/tests/test_DBGenerator.py @@ -0,0 +1,110 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- + +""" + test_DBGenerator.py + ~~~~~~~~~~~~~~~~~~~ + + DBGenerator test suite + + :copyright: (c) 2013 by Jauhien Piatlicki + :license: GPL-2, see LICENSE for more details. +""" + +import os +import unittest + +from g_sorcery.compatibility import TemporaryDirectory +from g_sorcery.exceptions import InvalidKeyError +from g_sorcery.g_collections import Package +from g_sorcery.package_db import DBGenerator + +from tests.base import BaseTest +from tests.server import Server + + +class TestingDBGenerator(DBGenerator): + def get_download_uries(self, common_config, config): + return [config["repo_uri"] + "/repo.data"] + + def parse_data(self, data_f): + content = data_f.read() + content = content.split("packages\n") + ebuild_data_lines = content[0].split("\n") + packages_lines = content[1].split("\n") + ebuild_data = {} + packages = [] + for line in ebuild_data_lines: + if line: + data = line.split(" ") + ebuild_data[data[0]] = data[1] + for line in packages_lines: + if line: + data = line.split(" ") + packages.append(Package(data[0], data[1], data[2])) + return {"ebuild_data": ebuild_data, "packages": packages} + + + def process_data(self, pkg_db, data, common_config, config): + data = data["repo.data"] + ebuild_data = data["ebuild_data"] + for package in data["packages"]: + pkg_db.add_category(package.category) + pkg_db.add_package(package, ebuild_data) + + +class TestDBGenerator(BaseTest): + + def test_functionality(self): + db_generator = TestingDBGenerator() + common_config = {} + config = {"repo_uri": "127.0.0.1:8080"} + + packages = [Package("app-test1", "test", "1"), Package("app-test1", "test", "2"), + Package("app-test1", "test1", "1"), Package("app-test2", "test2", "1")] + ebuild_data = {"test1": "test1", "test2": "test2"} + + orig_tempdir = TemporaryDirectory() + with open(os.path.join(orig_tempdir.name, "repo.data"), "w") as f: + for key, value in ebuild_data.items(): + f.write(key + " " + value + "\n") + f.write("packages\n") + for package in packages: + f.write(package.category + " " + package.name + " " + package.version + "\n") + + srv = Server(orig_tempdir.name) + srv.start() + + pkg_db = db_generator(self.tempdir.name, "test_repo", + common_config = common_config, config = config) + + srv.shutdown() + srv.join() + + self.assertEqual(set(pkg_db.list_categories()), set(["app-test1", "app-test2"])) + self.assertTrue(pkg_db.in_category("app-test1", "test")) + self.assertFalse(pkg_db.in_category("app-test2", "test")) + self.assertRaises(InvalidKeyError, pkg_db.in_category, "app-test3", "test") + self.assertEqual(set(pkg_db.list_package_names("app-test1")), set(['test', 'test1'])) + self.assertEqual(set(pkg_db.list_catpkg_names()),set(['app-test1/test', 'app-test1/test1', 'app-test2/test2'])) + self.assertRaises(InvalidKeyError, pkg_db.list_package_versions, "invalid", "test") + self.assertRaises(InvalidKeyError, pkg_db.list_package_versions, "app-test1", "invalid") + self.assertEqual(set(pkg_db.list_package_versions("app-test1", "test")), set(['1', '2'])) + self.assertEqual(set(pkg_db.list_all_packages()), set(packages)) + self.assertEqual(pkg_db.get_package_description(packages[0]), ebuild_data) + self.assertRaises(KeyError, pkg_db.get_package_description, Package("invalid", "invalid", "1")) + self.assertEqual(pkg_db.get_max_version("app-test1", "test"), "2") + self.assertEqual(pkg_db.get_max_version("app-test1", "test1"), "1") + self.assertRaises(InvalidKeyError, pkg_db.get_max_version, "invalid", "invalid") + pkg_set = set(packages) + for package, data in pkg_db: + self.assertTrue(package in pkg_set) + self.assertEqual(data, ebuild_data) + pkg_set.remove(package) + self.assertTrue(not pkg_set) + + +def suite(): + suite = unittest.TestSuite() + suite.addTest(TestDBGenerator('test_functionality')) + return suite