1 package liquibase.changelog.visitor;
2
3 import liquibase.changelog.ChangeSet;
4 import liquibase.changelog.DatabaseChangeLog;
5 import liquibase.database.Database;
6 import static org.easymock.EasyMock.expectLastCall;
7 import static org.easymock.classextension.EasyMock.*;
8 import org.junit.Test;
9
10 public class RollbackVisitorTest {
11 @Test
12 public void visit() throws Exception {
13 Database database = createMock(Database.class);
14
15 ChangeSet changeSet = createMock(ChangeSet.class);
16 changeSet.rollback(database);
17 expectLastCall();
18
19
20 database.removeRanStatus(changeSet);
21 expectLastCall();
22
23 database.commit();
24 expectLastCall();
25
26
27 replay(changeSet);
28 replay(database);
29
30 RollbackVisitor visitor = new RollbackVisitor(database);
31 visitor.visit(changeSet, new DatabaseChangeLog(), database);
32
33 verify(database);
34 verify(changeSet);
35 }
36 }