Search in sources :

Example 46 with CTagManager

use of com.google.security.zynamics.binnavi.Tagging.CTagManager in project binnavi by google.

the class CTagManagerTest method testTags.

@Test
public void testTags() throws CouldntSaveDataException, CouldntDeleteException {
    final MockSqlProvider sql = new MockSqlProvider();
    final CTagManager manager = new CTagManager(new Tree<CTag>(new TreeNode<CTag>(new CTag(1, "Foo", "Bar", TagType.VIEW_TAG, sql))), TagType.VIEW_TAG, sql);
    final MockTagManagerListener listener = new MockTagManagerListener();
    // ----------------------------------------- ADDING TAGS ---------------------------------------
    manager.addListener(listener);
    try {
        manager.addTag(null, "Tag I");
        fail();
    } catch (final NullPointerException exception) {
    }
    assertEquals("", listener.eventList);
    try {
        manager.addTag(manager.getRootTag(), null);
        fail();
    } catch (final NullPointerException exception) {
    }
    assertEquals("", listener.eventList);
    final ITreeNode<CTag> tag1 = manager.addTag(manager.getRootTag(), "Tag I");
    // Check listener events
    assertEquals("addedTag/", listener.eventList);
    // Check tag
    assertEquals(manager.getRootTag(), tag1.getParent());
    assertTrue(manager.getRootTag().getChildren().contains(tag1));
    final ITreeNode<CTag> tag2 = manager.addTag(manager.getRootTag(), "Tag II");
    // Check listener events
    assertEquals("addedTag/addedTag/", listener.eventList);
    // Check tag
    assertEquals(manager.getRootTag(), tag1.getParent());
    assertTrue(manager.getRootTag().getChildren().contains(tag1));
    assertEquals(manager.getRootTag(), tag2.getParent());
    assertTrue(manager.getRootTag().getChildren().contains(tag2));
    final ITreeNode<CTag> tag3 = manager.addTag(tag2, "Tag III");
    // Check listener events
    assertEquals("addedTag/addedTag/addedTag/", listener.eventList);
    // Check tag
    assertEquals(manager.getRootTag(), tag1.getParent());
    assertTrue(manager.getRootTag().getChildren().contains(tag1));
    assertEquals(manager.getRootTag(), tag2.getParent());
    assertTrue(manager.getRootTag().getChildren().contains(tag2));
    assertEquals(tag2, tag3.getParent());
    assertTrue(tag2.getChildren().contains(tag3));
    try {
        manager.insertTag(null, "Foo");
        fail();
    } catch (final NullPointerException exception) {
    }
    try {
        manager.insertTag(manager.getRootTag(), null);
        fail();
    } catch (final NullPointerException exception) {
    }
    final ITreeNode<CTag> tag4 = manager.insertTag(manager.getRootTag(), "Tag IV");
    // Check listener events
    assertEquals("addedTag/addedTag/addedTag/insertedTag/", listener.eventList);
    // Check tag
    assertEquals(tag4, tag1.getParent());
    assertFalse(manager.getRootTag().getChildren().contains(tag1));
    assertTrue(tag4.getChildren().contains(tag1));
    assertEquals(tag4, tag2.getParent());
    assertFalse(manager.getRootTag().getChildren().contains(tag2));
    assertTrue(tag4.getChildren().contains(tag2));
    assertEquals(tag2, tag3.getParent());
    assertTrue(tag2.getChildren().contains(tag3));
    final ITreeNode<CTag> tag5 = manager.insertTag(tag3, "Tag V");
    // Check listener events
    assertEquals("addedTag/addedTag/addedTag/insertedTag/insertedTag/", listener.eventList);
    // Check tag
    assertEquals(tag4, tag1.getParent());
    assertFalse(manager.getRootTag().getChildren().contains(tag1));
    assertTrue(tag4.getChildren().contains(tag1));
    assertEquals(tag4, tag2.getParent());
    assertFalse(manager.getRootTag().getChildren().contains(tag2));
    assertTrue(tag4.getChildren().contains(tag2));
    assertEquals(tag2, tag3.getParent());
    assertTrue(tag2.getChildren().contains(tag3));
    assertEquals(tag3, tag5.getParent());
    assertTrue(tag3.getChildren().contains(tag5));
    try {
        manager.deleteTag(null);
        fail();
    } catch (final NullPointerException exception) {
    }
    try {
        manager.deleteTag(manager.getRootTag());
        fail();
    } catch (final IllegalArgumentException exception) {
    }
    manager.deleteTag(tag5);
    // Check listener events
    assertEquals("addedTag/addedTag/addedTag/insertedTag/insertedTag/deletedTag/", listener.eventList);
    // Check tag
    assertEquals(tag4, tag1.getParent());
    assertFalse(manager.getRootTag().getChildren().contains(tag1));
    assertTrue(tag4.getChildren().contains(tag1));
    assertEquals(tag4, tag2.getParent());
    assertFalse(manager.getRootTag().getChildren().contains(tag2));
    assertTrue(tag4.getChildren().contains(tag2));
    assertEquals(tag2, tag3.getParent());
    assertTrue(tag2.getChildren().contains(tag3));
    assertNull(tag5.getParent());
    assertFalse(tag3.getChildren().contains(tag5));
    assertNull(CTagHelpers.findTag(manager.getRootTag(), 5));
    manager.deleteTag(tag4);
    // Check listener events
    assertEquals("addedTag/addedTag/addedTag/insertedTag/insertedTag/deletedTag/deletedTag/", listener.eventList);
    // Check tag
    assertEquals(manager.getRootTag(), tag1.getParent());
    assertTrue(manager.getRootTag().getChildren().contains(tag1));
    assertEquals(manager.getRootTag(), tag2.getParent());
    assertTrue(manager.getRootTag().getChildren().contains(tag2));
    assertEquals(tag2, tag3.getParent());
    assertTrue(tag2.getChildren().contains(tag3));
    assertNull(tag4.getParent());
    assertFalse(manager.getRootTag().getChildren().contains(tag4));
    assertNull(CTagHelpers.findTag(manager.getRootTag(), 4));
}
Also used : MockSqlProvider(com.google.security.zynamics.binnavi.Database.MockClasses.MockSqlProvider) TreeNode(com.google.security.zynamics.zylib.types.trees.TreeNode) ITreeNode(com.google.security.zynamics.zylib.types.trees.ITreeNode) CTag(com.google.security.zynamics.binnavi.Tagging.CTag) CTagManager(com.google.security.zynamics.binnavi.Tagging.CTagManager) Test(org.junit.Test)

Aggregations

CTagManager (com.google.security.zynamics.binnavi.Tagging.CTagManager)46 CTag (com.google.security.zynamics.binnavi.Tagging.CTag)34 Test (org.junit.Test)31 ExpensiveBaseTest (com.google.security.zynamics.binnavi.disassembly.types.ExpensiveBaseTest)23 MockSqlProvider (com.google.security.zynamics.binnavi.Database.MockClasses.MockSqlProvider)17 Tree (com.google.security.zynamics.zylib.types.trees.Tree)15 MockDatabase (com.google.security.zynamics.binnavi.Database.MockClasses.MockDatabase)13 CModule (com.google.security.zynamics.binnavi.disassembly.Modules.CModule)13 INaviModule (com.google.security.zynamics.binnavi.disassembly.INaviModule)12 CAddress (com.google.security.zynamics.zylib.disassembly.CAddress)12 Date (java.util.Date)12 TreeNode (com.google.security.zynamics.zylib.types.trees.TreeNode)10 Before (org.junit.Before)9 TagManager (com.google.security.zynamics.binnavi.API.disassembly.TagManager)8 ITreeNode (com.google.security.zynamics.zylib.types.trees.ITreeNode)8 Database (com.google.security.zynamics.binnavi.API.disassembly.Database)7 CView (com.google.security.zynamics.binnavi.disassembly.views.CView)7 Module (com.google.security.zynamics.binnavi.API.disassembly.Module)6 MockTagManager (com.google.security.zynamics.binnavi.Tagging.MockTagManager)5 MockView (com.google.security.zynamics.binnavi.disassembly.MockView)5