Search in sources :

Example 11 with TreeNode

use of com.google.security.zynamics.zylib.types.trees.TreeNode in project binnavi by google.

the class CTagManagerTest method test_C_Constructors.

@Test
public void test_C_Constructors() {
    try {
        new CTagManager(null, TagType.VIEW_TAG, new MockSqlProvider());
        fail();
    } catch (final NullPointerException exception) {
    }
    try {
        new CTagManager(new Tree<CTag>(new TreeNode<CTag>(new CTag(1, "Foo", "Bar", TagType.VIEW_TAG, new MockSqlProvider()))), null, new MockSqlProvider());
        fail();
    } catch (final NullPointerException exception) {
    }
    try {
        new CTagManager(new Tree<CTag>(new TreeNode<CTag>(new CTag(1, "Foo", "Bar", TagType.VIEW_TAG, new MockSqlProvider()))), TagType.VIEW_TAG, null);
        fail();
    } catch (final NullPointerException exception) {
    }
    final MockSqlProvider sql = new MockSqlProvider();
    new CTagManager(new Tree<CTag>(new TreeNode<CTag>(new CTag(1, "Foo", "Bar", TagType.VIEW_TAG, sql))), TagType.VIEW_TAG, sql);
}
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)

Example 12 with TreeNode

use of com.google.security.zynamics.zylib.types.trees.TreeNode in project binnavi by google.

the class CTagManagerTest method testTags2.

@Test
public void testTags2() 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();
    manager.addListener(listener);
    // ----------------------------------------- ADDING TAGS ---------------------------------------
    final ITreeNode<CTag> tag1 = manager.addTag(manager.getRootTag(), "Tag I");
    final ITreeNode<CTag> tag2 = manager.addTag(manager.getRootTag(), "Tag II");
    final ITreeNode<CTag> tag3 = manager.addTag(tag2, "Tag III");
    // ------------------------------------------- INSERT TAGS
    // ---------------------------------------------------
    final ITreeNode<CTag> tag4 = manager.insertTag(manager.getRootTag(), "Tag IV");
    final ITreeNode<CTag> tag5 = manager.insertTag(tag3, "Tag V");
    try {
        manager.deleteTagSubTree(null);
        fail();
    } catch (final NullPointerException exception) {
    }
    try {
        manager.deleteTagSubTree(manager.getRootTag());
        fail();
    } catch (final IllegalArgumentException exception) {
    }
    manager.deleteTagSubTree(tag5);
    // Check listener events
    assertEquals("addedTag/addedTag/addedTag/insertedTag/insertedTag/deletedTag/", listener.eventList);
    // Check tag
    assertEquals(4, DepthFirstSorter.getSortedList(manager.getRootTag()).size());
    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.deleteTagSubTree(tag4);
    // Check listener events
    assertEquals("addedTag/addedTag/addedTag/insertedTag/insertedTag/deletedTag/deletedSubtree/", listener.eventList);
    // Check tag
    assertEquals(0, DepthFirstSorter.getSortedList(manager.getRootTag()).size());
}
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)

Example 13 with TreeNode

use of com.google.security.zynamics.zylib.types.trees.TreeNode 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)

Example 14 with TreeNode

use of com.google.security.zynamics.zylib.types.trees.TreeNode in project binnavi by google.

the class MockTagManager method addTag.

@Override
public ITreeNode<CTag> addTag(final ITreeNode<CTag> parent, final String name) {
    final TreeNode<CTag> node = new TreeNode<CTag>(new CTag(1, name, "", m_type, new MockSqlProvider()));
    parent.addChild(node);
    node.setParent(parent);
    return node;
}
Also used : TreeNode(com.google.security.zynamics.zylib.types.trees.TreeNode) ITreeNode(com.google.security.zynamics.zylib.types.trees.ITreeNode) MockSqlProvider(com.google.security.zynamics.binnavi.Database.MockClasses.MockSqlProvider) CTag(com.google.security.zynamics.binnavi.Tagging.CTag)

Aggregations

CTag (com.google.security.zynamics.binnavi.Tagging.CTag)14 TreeNode (com.google.security.zynamics.zylib.types.trees.TreeNode)14 ITreeNode (com.google.security.zynamics.zylib.types.trees.ITreeNode)11 CTagManager (com.google.security.zynamics.binnavi.Tagging.CTagManager)10 MockSqlProvider (com.google.security.zynamics.binnavi.Database.MockClasses.MockSqlProvider)9 CAddress (com.google.security.zynamics.zylib.disassembly.CAddress)5 Test (org.junit.Test)5 MockDatabase (com.google.security.zynamics.binnavi.Database.MockClasses.MockDatabase)4 CModule (com.google.security.zynamics.binnavi.disassembly.Modules.CModule)4 Date (java.util.Date)4 Before (org.junit.Before)4 TagManager (com.google.security.zynamics.binnavi.API.disassembly.TagManager)3 SQLProvider (com.google.security.zynamics.binnavi.Database.Interfaces.SQLProvider)3 MockTagManager (com.google.security.zynamics.binnavi.Tagging.MockTagManager)3 MockView (com.google.security.zynamics.binnavi.disassembly.MockView)3 Tree (com.google.security.zynamics.zylib.types.trees.Tree)3 Database (com.google.security.zynamics.binnavi.API.disassembly.Database)2 Module (com.google.security.zynamics.binnavi.API.disassembly.Module)2 MockDebugger (com.google.security.zynamics.binnavi.Debug.Debugger.MockDebugger)2 CProjectTreeNode (com.google.security.zynamics.binnavi.Gui.MainWindow.ProjectTree.Nodes.CProjectTreeNode)2