def test_get_arch_success(self, m_lock): m_lock.list_locks.return_value = [{"arch": "arch"}] result = suite.get_arch('magna') m_lock.list_locks.assert_called_with( machine_type="magna", count=1 ) assert result == "arch"
def test_get_arch_fail(self, m_lock): m_lock.list_locks.return_value = False suite.get_arch('magna') m_lock.list_locks.assert_called_with(machine_type="magna", count=1)