wuttamess/tests/test_apt.py

# -*- coding: utf-8; -*-

from unittest import TestCase
from unittest.mock import patch, MagicMock

from wuttamess import apt as mod


class TestDistUpgrade(TestCase):

    def test_basic(self):
        c = MagicMock()
        with patch.object(mod, 'update') as update:
            with patch.object(mod, 'upgrade') as upgrade:
                mod.dist_upgrade(c, frontend='whatever')
                update.assert_called_once_with(c)
                upgrade.assert_called_once_with(c, dist_upgrade=True, frontend='whatever')


class TestInstall(TestCase):

    def test_basic(self):
        c = MagicMock()
        mod.install(c, 'postfix')
        c.run.assert_called_once_with('DEBIAN_FRONTEND=noninteractive apt-get --assume-yes install postfix')


class TestIsInstalled(TestCase):

    def test_already_installed(self):
        c = MagicMock()
        c.run.return_value.ok = True
        self.assertTrue(mod.is_installed(c, 'postfix'))
        c.run.assert_called_once_with('dpkg-query -s postfix', warn=True)

    def test_not_installed(self):
        c = MagicMock()
        c.run.return_value.ok = False
        self.assertFalse(mod.is_installed(c, 'postfix'))
        c.run.assert_called_once_with('dpkg-query -s postfix', warn=True)


class TestUpdate(TestCase):

    def test_basic(self):
        c = MagicMock()
        mod.update(c)
        c.run.assert_called_once_with('apt-get update')


class TestUpgrade(TestCase):

    def test_basic(self):
        c = MagicMock()
        mod.upgrade(c)
        c.run.assert_called_once_with('DEBIAN_FRONTEND=noninteractive apt-get --assume-yes --option Dpkg::Options::="--force-confdef" --option Dpkg::Options::="--force-confold" upgrade')