1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
test_ebuild.py
~~~~~~~~~~~~~~~~~~
ebuild generator test suite
:copyright: (c) 2013 by Jauhien Piatlicki
:license: GPL-2, see LICENSE for more details.
"""
import os, tempfile, unittest
from g_sorcery import ebuild, package_db
package = package_db.Package("app-test", "test", "0.1")
package2 = package_db.Package("app-test", "tst", "1")
class DummyDB(package_db.PackageDB):
def __init__(self, directory, repo_uri="", db_uri=""):
super().__init__(directory, repo_uri, db_uri)
def generate_tree(self):
self.add_category("app-test")
self.add_package(package,
{"author" : "jauhien", "homepage" : "127.0.0.1"})
self.add_package(package2,
{"author" : "unknown", "homepage" : "example.com"})
class DummyEbuildGenerator(ebuild.EbuildGenerator):
def get_template(self, ebuild, description):
tmpl = ["test", "author: $author", "homepage: $homepage", "var: $$var"]
return tmpl
class TestEbuildGenerator(unittest.TestCase):
def setUp(self):
self.tempdir = tempfile.TemporaryDirectory()
def tearDown(self):
del self.tempdir
def test_process(self):
eg = DummyEbuildGenerator(None)
tst_dict = {"a" : "d", "b" : "e", "c" : "f"}
ebuild = ["$a", "$b", "$c"]
ebuild = eg.process(ebuild, tst_dict)
self.assertEqual(ebuild, ["d", "e", "f"])
def test_generate(self):
db = DummyDB(self.tempdir.name)
db.generate()
eg = DummyEbuildGenerator(db)
ebuild = eg.generate(package)
self.assertEqual(ebuild, ['test', 'author: jauhien',
'homepage: 127.0.0.1', 'var: $var'])
class DummyEbuildGeneratorFromFile(ebuild.EbuildGeneratorFromFile):
def __init__(self, db, path):
super().__init__(db)
self.path = path
def get_template_file(self, package, description):
return self.path
class TestEbuildGeneratorFromFile(unittest.TestCase):
def setUp(self):
self.tempdir = tempfile.TemporaryDirectory()
def tearDown(self):
del self.tempdir
def test_generate(self):
db = DummyDB(os.path.join(self.tempdir.name, 'tstdb'))
db.generate()
tmpl = os.path.join(self.tempdir.name, 'tst.tmpl')
with open(tmpl, 'w') as f:
f.write("""test
author: $author
homepage: $homepage
var: $$var""")
eg = DummyEbuildGeneratorFromFile(db, tmpl)
ebuild = eg.generate(package)
self.assertEqual(ebuild, ['test', 'author: jauhien',
'homepage: 127.0.0.1', 'var: $var'])
def suite():
suite = unittest.TestSuite()
suite.addTest(TestEbuildGenerator('test_process'))
suite.addTest(TestEbuildGenerator('test_generate'))
suite.addTest(TestEbuildGeneratorFromFile('test_generate'))
return suite
|
GitWeb