thefuck/tests/test_history.py

import pytest
from mock import Mock
from thefuck.history import History


class TestHistory(object):
    @pytest.fixture(autouse=True)
    def process(self, monkeypatch):
        Process = Mock()
        Process.return_value.parent.return_value.pid = 1
        monkeypatch.setattr('thefuck.history.Process', Process)
        return Process

    @pytest.fixture(autouse=True)
    def db(self, monkeypatch):
        class DBMock(dict):
            def __init__(self):
                super(DBMock, self).__init__()
                self.sync = Mock()

            def __call__(self, *args, **kwargs):
                return self

        db = DBMock()
        monkeypatch.setattr('thefuck.history.shelve.open', db)
        return db

    def test_set(self, db):
        history = History()
        history.update(last_command='ls',
                       last_fixed_command=None)
        assert db == {'1-last_command': 'ls',
                      '1-last_fixed_command': None}

    def test_get(self, db):
        history = History()
        db['1-last_command'] = 'cd ..'
        assert history.last_command == 'cd ..'

    def test_get_without_value(self):
        history = History()
        assert history.last_command is None