def test_multiple_computers(self):
"""Verify a user can use multiple computers."""
infos = [("pc1", "name"), ("pc2", "name"), ("pc2", "name")]
root = tempfile.mkdtemp()
downloads = tempfile.mkdtemp()
try:
with patch("dtb.user.get_info", Mock(side_effect=infos)):
user = User.new(root, "name", downloads=downloads)
user2 = User.add(root, "name")
user3 = User.add(root, "name")
self.assertEqual(user, user2)
self.assertEqual(user, user3)
finally:
shutil.rmtree(root)
def test_add_old_format(self):
"""Verify a user can be added to an old format."""
root = tempfile.mkdtemp()
try:
with patch("dtb.user.get_info", Mock(return_value=self.INFOS[0])):
user = User.new(root, "TempUser")
open(user.path_info, "w").close() # blank the file
with patch("dtb.user.get_info", Mock(return_value=self.INFOS[1])):
user2 = User.add(root, "TempUser")
self.assertEqual(1, len(user2.info))
finally:
shutil.rmtree(root)
