Как выполнить модульное тестирование синхронизированного кода - PullRequest
9 голосов
/ 13 мая 2010

Я новичок в Java и junit. У меня есть следующий кусок кода, который я хочу проверить. Буду признателен, если вы отправите свои идеи о том, как лучше всего их протестировать.

По сути, следующий код касается выбора лидера из кластера. Лидер удерживает блокировку общего кэша, и службы лидера возобновляются и удаляются, если он каким-либо образом ослабляет блокировку кеша.

Как я могу убедиться, что лидер / поток по-прежнему удерживает блокировку кеша и что другой поток не может возобновить свои службы, пока выполняется первый?

public interface ContinuousService {

public void resume();
public void pause();
}


public abstract class ClusterServiceManager {
private volatile boolean leader = false;
private volatile boolean electable = true;
private List<ContinuousService> services;

protected synchronized void onElected() {
    if (!leader) {
        for (ContinuousService service : services) {
            service.resume();
        }
        leader = true;
    }
}

protected synchronized void onDeposed() {
    if (leader) {
        for (ContinuousService service : services) {
            service.pause();
        }
        leader = false;
    }
}

public void setServices(List<ContinuousService> services) {
    this.services = services;
}

@ManagedAttribute
public boolean isElectable() {
    return electable;
}

@ManagedAttribute
public boolean isLeader() {
    return leader;
}



public class TangosolLeaderElector extends ClusterServiceManager implements Runnable {
private static final Logger log = LoggerFactory.getLogger(TangosolLeaderElector.class);
private String election;
private long electionWaitTime= 5000L;

private NamedCache cache;

public void start() {
    log.info("Starting LeaderElector ({})",election);
    Thread t = new Thread(this, "LeaderElector ("+election+")");
    t.setDaemon(true);
    t.start();
}

public void run() {
    // Give the connection a chance to start itself up
    try {
        Thread.sleep(1000);
    } catch (InterruptedException e) {}

    boolean wasElectable = !isElectable();
    while (true) {
        if (isElectable()) {
            if (!wasElectable) {
                log.info("Leadership requested on election: {}",election);
                wasElectable = isElectable();
            }
            boolean elected = false;
            try {
                // Try and get the lock on the LeaderElectorCache for the current election
                if (!cache.lock(election, electionWaitTime)) {
                    // We didn't get the lock. cycle round again. 
                    // This code to ensure we check the electable flag every now & then
                    continue;
                }
                elected = true;
                log.info("Leadership taken on election: {}",election);
                onElected();

                // Wait here until the services fail in some way.
                while (true) {
                    try {
                        Thread.sleep(electionWaitTime);
                    } catch (InterruptedException e) {}
                    if (!cache.lock(election, 0)) {
                        log.warn("Cache lock no longer held for election: {}", election);
                        break;
                    } else if (!isElectable()) {
                        log.warn("Node is no longer electable for election: {}", election);
                        break;
                    }
                    // We're fine - loop round and go back to sleep.
                }
            } catch (Exception e) {
                if (log.isErrorEnabled()) {
                    log.error("Leadership election " + election + " failed (try bfmq logs for details)", e);
                }
            } finally {
                if (elected) {
                    cache.unlock(election);
                    log.info("Leadership resigned on election: {}",election);
                    onDeposed();
                }
                // On deposition, do not try and get re-elected for at least the standard wait time.
                try { Thread.sleep(electionWaitTime); } catch (InterruptedException e) {}
            }
        } else {
            // Not electable - wait a bit and check again.
            if (wasElectable) {
                log.info("Leadership NOT requested on election ({}) - node not electable",election);
                wasElectable = isElectable();
            }
            try {
                Thread.sleep(electionWaitTime);
            } catch (InterruptedException e) {}
        }
    }
}

public void setElection(String election) {
    this.election = election;
}

@ManagedAttribute
public String getElection() {
    return election;
}

public void setNamedCache(NamedCache nc) {
    this.cache = nc;
}

Ответы [ 3 ]

4 голосов
/ 13 мая 2010

Если вы не слишком внимательны к использованию JUnit, вы можете использовать среду TestNG. Они имеют многопоточную поддержку.

3 голосов
/ 14 мая 2010

В качестве альтернативы тестированию фреймворков (или дополнительным расширениям поверх JUnit, если вы его используете) является просто старый код:

  • Создайте несколько потоков и примените их к этой подпрограмме.
  • Перебирайте темы и проверяйте каждую, пока не выясните, кто является лидером.
  • Применить различные изменения среды (в том числе с течением времени) к состоянию вашей программы и повторить тест. Лидер по-прежнему лидер?
  • Теперь заставьте лидера отречься от престола (убить эту нить или что-то в этом роде). Другой поток вступил во владение?
0 голосов
/ 13 февраля 2014

Для JUnit есть MultithreadingTester из junit-toolbox .

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...