RepoMirrors/ceph
1
0
Fork 0
mirror of https://github.com/ceph/ceph synced 2025-03-25 11:48:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

cephadm/test: support context class

Browse Source 
Signed-off-by: Joao Eduardo Luis <joao@suse.com>
This commit is contained in:
Joao Eduardo Luis 2021-01-14 14:35:33 -01:00
parent c785dc5679
commit 5178403833
No known key found for this signature in database
GPG Key ID: D3DCCB5DB5229660
2 changed files with 28 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/cephadm/tests/fixtures.py
View File

@ -36,6 +36,7 @@ def exporter():
mock.patch('cephadm.CephadmDaemon.run', _mock_run), \
mock.patch('cephadm.CephadmDaemon._scrape_host_facts', _mock_scrape_host):
exporter = cd.CephadmDaemon(fsid='foobar', daemon_id='test')
ctx = cd.CephadmContext()
exporter = cd.CephadmDaemon(ctx, fsid='foobar', daemon_id='test')
assert exporter.token == 'MyAccessToken'
yield exporter

42
src/cephadm/tests/test_cephadm.py
View File

@ -1,4 +1,5 @@
# type: ignore
from typing import List, Optional
import mock
from mock import patch
import os
@ -204,32 +205,38 @@ default via fe80::2480:28ec:5097:3fe2 dev wlp2s0 proto ra metric 20600 pref medi
# test normal valid login with url, username and password specified
call_throws.return_value = '', '', 0
args = cd._parse_args(['registry-login', '--registry-url', 'sample-url', '--registry-username', 'sample-user', '--registry-password', 'sample-pass'])
cd.args = args
retval = cd.command_registry_login()
ctx: Optional[cd.CephadmContext] = cd.cephadm_init_ctx(
['registry-login', '--registry-url', 'sample-url',
'--registry-username', 'sample-user', '--registry-password',
'sample-pass'])
assert ctx
retval = cd.command_registry_login(ctx)
assert retval == 0
# test bad login attempt with invalid arguments given
args = cd._parse_args(['registry-login', '--registry-url', 'bad-args-url'])
cd.args = args
ctx: Optional[cd.CephadmContext] = cd.cephadm_init_ctx(
['registry-login', '--registry-url', 'bad-args-url'])
assert ctx
with pytest.raises(Exception) as e:
assert cd.command_registry_login()
assert cd.command_registry_login(ctx)
assert str(e.value) == ('Invalid custom registry arguments received. To login to a custom registry include '
'--registry-url, --registry-username and --registry-password options or --registry-json option')
# test normal valid login with json file
get_parm.return_value = {"url": "sample-url", "username": "sample-username", "password": "sample-password"}
args = cd._parse_args(['registry-login', '--registry-json', 'sample-json'])
cd.args = args
retval = cd.command_registry_login()
ctx: Optional[cd.CephadmContext] = cd.cephadm_init_ctx(
['registry-login', '--registry-json', 'sample-json'])
assert ctx
retval = cd.command_registry_login(ctx)
assert retval == 0
# test bad login attempt with bad json file
get_parm.return_value = {"bad-json": "bad-json"}
args = cd._parse_args(['registry-login', '--registry-json', 'sample-json'])
cd.args = args
ctx: Optional[cd.CephadmContext] = cd.cephadm_init_ctx(
['registry-login', '--registry-json', 'sample-json'])
assert ctx
with pytest.raises(Exception) as e:
assert cd.command_registry_login()
assert cd.command_registry_login(ctx)
assert str(e.value) == ("json provided for custom registry login did not include all necessary fields. "
"Please setup json file as\n"
"{\n"
@ -240,10 +247,13 @@ default via fe80::2480:28ec:5097:3fe2 dev wlp2s0 proto ra metric 20600 pref medi
# test login attempt with valid arguments where login command fails
call_throws.side_effect = Exception
args = cd._parse_args(['registry-login', '--registry-url', 'sample-url', '--registry-username', 'sample-user', '--registry-password', 'sample-pass'])
cd.args = args
ctx: Optional[cd.CephadmContext] = cd.cephadm_init_ctx(
['registry-login', '--registry-url', 'sample-url',
'--registry-username', 'sample-user', '--registry-password',
'sample-pass'])
assert ctx
with pytest.raises(Exception) as e:
cd.command_registry_login()
cd.command_registry_login(ctx)
assert str(e.value) == "Failed to login to custom registry @ sample-url as sample-user with given password"
def test_get_image_info_from_inspect(self):
@ -381,7 +391,7 @@ class TestCustomContainer(unittest.TestCase):
class TestCephadmExporter(object):
exporter: cd.CephadmDaemon
files_created = []
files_created: List[str] = []
crt = """-----BEGIN CERTIFICATE-----
MIIC1zCCAb8CEFHoZE2MfUVzo53fzzBKAT0wDQYJKoZIhvcNAQENBQAwKjENMAsG
A1UECgwEQ2VwaDEZMBcGA1UECwwQY2VwaGFkbS1leHBvcnRlcjAeFw0yMDExMjUy