← Back to branch summary

~ubuntu-core-dev/unattended-upgrades/ubuntu

« back to all changes in this revision

Viewing changes to test/test_substitute.py

  • Committer: Michael Vogt
  • Date: 2010-08-02 09:04:20 UTC
  • mfrom: (136 ubuntu)
  • mto: This revision was merged to the branch mainline in revision 139.
  • Revision ID: michael.vogt@ubuntu.com-20100802090420-b1vaha9f359q9vto
merged from ubuntu

Show diffs side-by-side

added added

removed removed

Lines of Context:
test/test_substitute.py
 
1
#!/usr/bin/python
 
2
 
 
3
import apt
 
4
import apt_pkg
 
5
import os
 
6
import logging
 
7
import unittest
 
8
import sys
 
9
 
 
10
from StringIO import StringIO
 
11
 
 
12
import unattended_upgrade
 
13
from unattended_upgrade import substitute, get_allowed_origins
 
14
 
 
15
class TestSubstitude(unittest.TestCase):
 
16
 
 
17
    def setUp(self):
 
18
        # monkey patch DISTRO_{CODENAME, ID}
 
19
        unattended_upgrade.DISTRO_CODENAME = "nacked"
 
20
        unattended_upgrade.DISTRO_ID = "MyDistroID"
 
21
 
 
22
    def testSubstitute(self):
 
23
        """ test if the substitute function works """
 
24
        self.assertTrue(substitute("${distro_codename}-updates"),
 
25
                        "nacked-updates")
 
26
        self.assertTrue(substitute("${distro_id}"), "MyDistroID")
 
27
 
 
28
    def test_get_allowed_origins_with_substitute(self):
 
29
        """ test if substitute for get_allowed_origins works """
 
30
        apt_pkg.config.clear("Unattended-Upgrade::Allowed-Origins")
 
31
        apt_pkg.config.set("Unattended-Upgrade::Allowed-Origins::",
 
32
                           "${distro_id} ${distro_codename}-security")
 
33
        l = get_allowed_origins()
 
34
        self.assertTrue(("MyDistroID","nacked-security") in l)
 
35
 
 
36
if __name__ == "__main__":
 
37
    logging.basicConfig(level=logging.DEBUG)
 
38
    unittest.main()
 
39