def setOs(self, osName):
"""Sets the host operating system.
:type osName: string
:param osName: os value to set host to
"""
self.stub.SetOs(host_pb2.HostSetOsRequest(host=self.data, os=osName),
timeout=Cuebot.Timeout)
def testSetOs(self, getStubMock):
stubMock = mock.Mock()
stubMock.SetOs.return_value = host_pb2.HostSetOsResponse()
getStubMock.return_value = stubMock
osName = 'linux'
host = opencue.wrappers.host.Host(host_pb2.Host(name=TEST_HOST_NAME))
host.setOs(osName)
stubMock.SetOs.assert_called_with(host_pb2.HostSetOsRequest(
host=host.data, os=osName),
timeout=mock.ANY)
def setOs(self, os):
"""Sets the host os
@type os: string
@param os: os value to set host to"""
self.stub.SetOs(host_pb2.HostSetOsRequest(host=self.data, os=os),
timeout=Cuebot.Timeout)
