import imp

import pytest

import requests

centos_cert = imp.load_source('centos_cert', 'centos-cert')

class MockResponse(object):

    def raise_for_status(self):

        pass

    @property

    def text(self):

        if self.url == 'https://accounts.centos.org/user/dogencert':

            return 'DUMMY-USER-CERT'

        if self.url == 'https://accounts.centos.org/ca/ca-cert.pem':

            return 'DUMMY-CA-CERT'

        raise NotImplementedError()

class RequestRecorder(object):

    """ Record args to requests.get() or requests.post() """

    def __call__(self, url, **kwargs):

        """ mocking requests.get() or requests.post() """

        self.response = MockResponse()

        self.response.url = url

        self.kwargs = kwargs

        return self.response

@pytest.fixture

def mock_get():

    return RequestRecorder()

@pytest.fixture

def mock_post():

    return RequestRecorder()

def test_download(monkeypatch, tmpdir, mock_get, mock_post):

    monkeypatch.delattr('requests.sessions.Session.request')

    monkeypatch.setenv('HOME', str(tmpdir))

    monkeypatch.setattr(requests, 'get', mock_get)

    monkeypatch.setattr(requests, 'post', mock_post)

    centos_cert.download_cert('myuser', 'mypass')

    assert mock_get.response.url == \

        'https://accounts.centos.org/ca/ca-cert.pem'

    assert mock_post.response.url == \

        'https://accounts.centos.org/user/dogencert'

    certfile = tmpdir.join('.centos.cert')

    assert certfile.check(file=1)

    assert certfile.read() == 'DUMMY-USER-CERT'

    assert oct(certfile.stat().mode & 0777) == '0600'

    cacertfile = tmpdir.join('.centos-server-ca.cert')

    assert cacertfile.check(file=1)

    assert cacertfile.read() == 'DUMMY-CA-CERT'

    uploadcafile = tmpdir.join('.centos-upload-ca.cert')

    assert uploadcafile.check(file=1)

    assert uploadcafile.read() == 'DUMMY-CA-CERT'