2 from basetest
import BaseTest
6 sys
.path
.insert(0, '..')
7 from zeroinstall
.injector
import model
, arch
, qdom
8 from zeroinstall
.injector
.namespaces
import XMLNS_IFACE
10 #from zeroinstall.injector.origsolver import DefaultSolver as Solver
11 #from zeroinstall.injector.pbsolver import PBSolver as Solver
12 #from zeroinstall.injector.sgsolver import DefaultSolver as Solver
13 from zeroinstall
.injector
.solver
import SATSolver
as Solver
16 logger
= logging
.getLogger()
19 def lookup_any(self
, digests
):
24 uri_prefix
= 'http://localhost/tests/'
27 def __init__(self
, n
):
31 def add_requires(self
, lib
, min_v
, max_v
):
32 self
.requires
.append((lib
, min_v
, max_v
))
35 def __init__(self
, name
):
39 def get_version(self
, version
):
40 if version
not in self
.versions
:
41 self
.versions
[version
] = Version(version
)
42 return self
.versions
[version
]
45 def child(parent
, name
, attrs
= None):
46 new
= qdom
.Element(XMLNS_IFACE
, name
, attrs
or {})
47 parent
.childNodes
.append(new
)
50 root
= qdom
.Element(XMLNS_IFACE
, 'interface', {'uri' : uri_prefix
+ self
.name
})
51 child(root
, 'name').content
= self
.name
52 child(root
, 'summary').content
= self
.name
55 for version
in self
.versions
.values():
56 impl
= child(root
, 'implementation', {
58 'version': str(version
.n
),
60 child(impl
, 'manifest-digest', {'sha1new': '1234'})
61 for lib
, min_v
, max_v
in version
.requires
:
62 req
= child(impl
, 'requires', {'interface': uri_prefix
+ lib
})
63 child(req
, 'version', {
64 'before': str(int(max_v
) + 1),
68 feed
= model
.ZeroInstallFeed(root
)
69 feed
.last_modified
= 1
77 def get_prog(self
, prog
):
78 if not prog
in self
.progs
:
79 self
.progs
[prog
] = Program(prog
)
80 return self
.progs
[prog
]
82 def get_interface(self
, uri
):
83 if uri
not in self
.interfaces
:
84 iface
= model
.Interface(uri
)
85 iface
._main
_feed
= self
.progs
[uri
.rsplit('/', 1)[1]].build_feed()
86 self
.interfaces
[uri
] = iface
87 return self
.interfaces
[uri
]
89 def assertSelection(expected
, repo
):
92 expected
= [tuple(e
.strip().split('-')) for e
in expected
.split(",")]
94 for line
in repo
.split('\n'):
98 prog
, versions
= line
.split(':')
100 for v
in versions
.split():
101 cache
.get_prog(prog
).get_version(v
)
103 prog
, requires
= line
.split('=>')
104 prog
, version_range
= prog
.strip().split('[')
105 lib
, min_v
, max_v
= requires
.split()
106 assert version_range
.endswith(']')
107 version_range
= version_range
[:-1]
108 if ',' in version_range
:
109 min_p
, max_p
= map(int, version_range
.split(','))
110 prog_versions
= range(min_p
, max_p
+ 1)
112 prog_versions
= [int(version_range
)]
113 for prog_version
in prog_versions
:
114 cache
.get_prog(prog
).get_version(str(prog_version
)).add_requires(lib
, min_v
, max_v
)
116 root
= uri_prefix
+ expected
[0][0]
117 s
= Solver(model
.network_offline
, cache
, stores
)
118 s
.solve(root
, arch
.get_architecture('Linux', 'x86_64'))
120 if expected
[0][1] == 'FAIL':
126 for iface
, impl
in s
.selections
.iteritems():
127 actual
.append(((iface
.uri
.rsplit('/', 1)[1]), impl
.get_version()))
131 if expected
!= actual
:
132 raise Exception("Solve failed:\nExpected: %s\n Actual: %s" % (expected
, actual
))
134 class TestSAT(BaseTest
):
135 def testSimple(self
):
136 assertSelection("prog-5, liba-5", """
144 def testBestImpossible(self
):
145 assertSelection("prog-1", """
152 assertSelection("prog-1", """
153 prog: 1 2 3 4 5 6 7 8 9
154 liba: 1 2 3 4 5 6 7 8 9
155 libb: 1 2 3 4 5 6 7 8 9
156 libc: 1 2 3 4 5 6 7 8 9
157 libd: 1 2 3 4 5 6 7 8 9
159 prog[2,9] => liba 1 9
160 liba[1,9] => libb 1 9
161 libb[1,9] => libc 1 9
162 libc[1,9] => libd 1 9
163 libd[1,9] => libe 0 0
166 def testNoSolution(self
):
167 assertSelection("prog-FAIL", """
170 prog[1,3] => liba 2 3
173 suite
= unittest
.makeSuite(TestSAT
)
174 if __name__
== '__main__':
175 sys
.argv
.append('-v')