RepoMirrors/ceph
1
0
Fork 0
mirror of https://github.com/ceph/ceph synced 2025-03-30 15:31:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #374 from andrewschoen/orchestra-test

Browse Source 
test improvements
This commit is contained in:
Zack Cerza 2014-12-03 13:47:30 -07:00
commit ff126cc98a
1 changed files with 27 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
teuthology/orchestra/test/test_cluster.py
View File

@ -1,7 +1,7 @@
import fudge
import pytest
from mock import patch
from mock import patch, Mock
from .. import cluster, remote
@ -208,20 +208,33 @@ class TestCluster(object):
c_foo = c.exclude('foo', lambda role: role.startswith('b'))
assert c_foo.remotes == {r2: ['bar'], r3: ['foo']}
@patch("teuthology.misc.write_file")
@patch("teuthology.misc.sudo_write_file")
def test_write_file(self, m_sudo_write_file, m_write_file):
r1 = remote.Remote('r1', ssh=fudge.Fake('SSH'))
c = cluster.Cluster(
class TestWriteFile(object):
""" Tests for cluster.write_file """
def setup(self):
self.r1 = remote.Remote('r1', ssh=Mock())
self.c = cluster.Cluster(
remotes=[
(r1, ['foo', 'bar']),
],
)
c.write_file("filename", "content", sudo=True)
m_sudo_write_file.assert_called_with(r1, "filename", "content", owner=None, perms=None)
(self.r1, ['foo', 'bar']),
],
)
@patch("teuthology.misc.write_file")
def test_write_file(self, m_write_file):
self.c.write_file("filename", "content")
m_write_file.assert_called_with(self.r1, "filename", "content")
@patch("teuthology.misc.write_file")
def test_fails_with_invalid_perms(self, m_write_file):
with pytest.raises(ValueError):
c.write_file("filename", "content", sudo=False, perms="perms")
c.write_file("filename", "content")
m_write_file.assert_called_with(r1, "filename", "content")
self.c.write_file("filename", "content", sudo=False, perms="invalid")
@patch("teuthology.misc.write_file")
def test_fails_with_invalid_owner(self, m_write_file):
with pytest.raises(ValueError):
self.c.write_file("filename", "content", sudo=False, owner="invalid")
@patch("teuthology.misc.sudo_write_file")
def test_with_sudo(self, m_sudo_write_file):
self.c.write_file("filename", "content", sudo=True)
m_sudo_write_file.assert_called_with(self.r1, "filename", "content", owner=None, perms=None)