← Back to branch summary

~gandelman-a/software-properties/lp829109

« back to all changes in this revision

Viewing changes to tests/test_dbus.py

  • Committer: Michael Vogt
  • Date: 2011-07-12 08:20:09 UTC
  • Revision ID: michael.vogt@ubuntu.com-20110712082009-ytx1yjxoe54uh2q1
add tests for add/add-data/remove gpg keys

Show diffs side-by-side

added added

removed removed

Lines of Context:
tests/test_dbus.py
6
6
import copy
7
7
import dbus
8
8
import logging
 
9
import glob
9
10
import os
10
11
import signal
11
12
import subprocess
184
185
        self.iface.removeSource(s.replace("deb", "deb-src"))
185
186
        self.assertFalse(s in open("./aptroot/etc/apt/sources.list").read())
186
187
 
 
188
 
 
189
    def test_add_gpg_key(self):
 
190
        # clean
 
191
        for f in glob.glob("./aptroot/etc/apt/*.gpg"):
 
192
            os.remove(f)
 
193
        self.assertTrue(len(glob.glob("./aptroot/etc/apt/*.gpg")) == 0)
 
194
        # add key from file
 
195
        res = self.iface.addKey("./data/testkey.gpg")
 
196
        self.assertTrue(res)
 
197
        self.assertTrue(len(glob.glob("./aptroot/etc/apt/*.gpg")) == 3)
 
198
        self.assertNotEqual(os.path.getsize("./aptroot/etc/apt/trusted.gpg"), 0)
 
199
        # remove the key 
 
200
        res = self.iface.removeKey("D732CA59")
 
201
        self.assertTrue(res)
 
202
        self.assertEqual(os.path.getsize("./aptroot/etc/apt/trusted.gpg"), 0)
 
203
        # add from data
 
204
        data = open("./data/testkey.gpg").read()
 
205
        res = self.iface.addKeyFromData(data)
 
206
        self.assertTrue(res)
 
207
        self.assertTrue(len(glob.glob("./aptroot/etc/apt/*.gpg")) == 3)
 
208
        self.assertNotEqual(os.path.getsize("./aptroot/etc/apt/trusted.gpg"), 0)
 
209
        # remove the key 
 
210
        res = self.iface.removeKey("D732CA59")
 
211
        self.assertTrue(res)
 
212
        self.assertEqual(os.path.getsize("./aptroot/etc/apt/trusted.gpg"), 0)
 
213
        # test nonsense
 
214
        res = self.iface.addKeyFromData("nonsens")
 
215
        self.assertFalse(res)
 
216
        
 
217
 
187
218
if __name__ == "__main__":
188
219
    logging.basicConfig(level=logging.DEBUG)
189
220
    unittest.main()