def test_is_bootstrapping_stat_file(tmpdir):
agent.CERT_PATH = str(tmpdir)
agent.CLIENT_CERT_PATH = str(tmpdir / 'client.crt')
with mock.patch('agent.logger') as prn:
assert agent.is_bootstrapping()
assert mock.call(
'No certificate found on disk.') in prn.warning.mock_calls
def test_is_bootstrapping_false_on_valid_cert(tmpdir):
crt = tmpdir / 'client.crt'
agent.CERT_PATH = str(tmpdir)
agent.CLIENT_CERT_PATH = str(crt)
with mock.patch('builtins.print') as prn:
Path(crt).write_text('nonzero')
assert not agent.is_bootstrapping()
assert not prn.mock_calls
def test_is_bootstrapping_check_filesize(tmpdir):
crt = tmpdir / 'client.crt'
agent.CERT_PATH = str(tmpdir)
agent.CLIENT_CERT_PATH = str(crt)
with mock.patch('agent.logger') as prn:
Path(agent.CLIENT_CERT_PATH).touch()
assert agent.is_bootstrapping()
assert mock.call('Certificate found but it is broken') in prn.warning.mock_calls
def test_is_bootstrapping_create_dir(tmpdir):
notexistent_dir = str(tmpdir / 'notexistent')
agent.CERT_PATH = notexistent_dir
with mock.patch('os.makedirs') as md, \
mock.patch('os.chmod') as chm, \
mock.patch('builtins.print') as prn:
assert agent.is_bootstrapping()
assert md.called_with(notexistent_dir)
assert chm.called_with(notexistent_dir, 0o700)
assert mock.call('No certificate found on disk.') in prn.mock_calls
def test_is_bootstrapping_create_dir(tmpdir):
notexistent_dir = tmpdir / 'notexistent'
agent.CERT_PATH = str(notexistent_dir)
agent.CLIENT_CERT_PATH = str(notexistent_dir / 'client.crt')
with mock.patch('os.makedirs') as md, \
mock.patch('os.chmod') as chm, \
mock.patch('agent.logger') as prn:
assert agent.is_bootstrapping()
assert md.called_with(notexistent_dir)
assert chm.called_with(notexistent_dir, 0o700)
assert mock.call('No certificate found on disk.') in prn.warning.mock_calls
def test_is_bootstrapping_stat_file(tmpdir):
agent.CERT_PATH = str(tmpdir)
with mock.patch('builtins.print') as prn:
assert agent.is_bootstrapping()
assert mock.call('No certificate found on disk.') in prn.mock_calls
