Mocks classes for testing the storage calls
(final boolean default_get,
final boolean default_put)
| 738 | * Mocks classes for testing the storage calls |
| 739 | */ |
| 740 | private void setupStorage(final boolean default_get, |
| 741 | final boolean default_put) throws Exception { |
| 742 | storage = new MockBase(tsdb, client, default_get, default_put, true, true); |
| 743 | final List<byte[]> families = new ArrayList<byte[]>(1); |
| 744 | families.add(Tree.TREE_FAMILY()); |
| 745 | storage.addTable(TREE_TABLE, families); |
| 746 | |
| 747 | byte[] key = new byte[] { 0, 1 }; |
| 748 | // set pre-test values |
| 749 | storage.addColumn(TREE_TABLE, key, Tree.TREE_FAMILY(), |
| 750 | "tree".getBytes(MockBase.ASCII()), |
| 751 | (byte[])TreetoStorageJson.invoke(buildTestTree())); |
| 752 | |
| 753 | TreeRule rule = new TreeRule(1); |
| 754 | rule.setField("host"); |
| 755 | rule.setType(TreeRuleType.TAGK); |
| 756 | storage.addColumn(TREE_TABLE, key, Tree.TREE_FAMILY(), |
| 757 | "tree_rule:0:0".getBytes(MockBase.ASCII()), JSON.serializeToBytes(rule)); |
| 758 | |
| 759 | rule = new TreeRule(1); |
| 760 | rule.setField(""); |
| 761 | rule.setLevel(1); |
| 762 | rule.setType(TreeRuleType.METRIC); |
| 763 | storage.addColumn(TREE_TABLE, key, Tree.TREE_FAMILY(), |
| 764 | "tree_rule:1:0".getBytes(MockBase.ASCII()), JSON.serializeToBytes(rule)); |
| 765 | |
| 766 | Branch root = new Branch(1); |
| 767 | root.setDisplayName("ROOT"); |
| 768 | TreeMap<Integer, String> root_path = new TreeMap<Integer, String>(); |
| 769 | root_path.put(0, "ROOT"); |
| 770 | root.prependParentPath(root_path); |
| 771 | // TODO - static |
| 772 | Method branch_json = Branch.class.getDeclaredMethod("toStorageJson"); |
| 773 | branch_json.setAccessible(true); |
| 774 | storage.addColumn(TREE_TABLE, key, Tree.TREE_FAMILY(), |
| 775 | "branch".getBytes(MockBase.ASCII()), (byte[])branch_json.invoke(root)); |
| 776 | |
| 777 | // tree 2 |
| 778 | key = new byte[] { 0, 2 }; |
| 779 | |
| 780 | Tree tree2 = new Tree(); |
| 781 | tree2.setTreeId(2); |
| 782 | tree2.setName("2nd Tree"); |
| 783 | tree2.setDescription("Other Tree"); |
| 784 | storage.addColumn(TREE_TABLE, key, Tree.TREE_FAMILY(), |
| 785 | "tree".getBytes(MockBase.ASCII()), |
| 786 | (byte[])TreetoStorageJson.invoke(tree2)); |
| 787 | |
| 788 | rule = new TreeRule(2); |
| 789 | rule.setField("host"); |
| 790 | rule.setType(TreeRuleType.TAGK); |
| 791 | storage.addColumn(TREE_TABLE, key, Tree.TREE_FAMILY(), |
| 792 | "tree_rule:0:0".getBytes(MockBase.ASCII()), JSON.serializeToBytes(rule)); |
| 793 | |
| 794 | rule = new TreeRule(2); |
| 795 | rule.setField(""); |
| 796 | rule.setLevel(1); |
| 797 | rule.setType(TreeRuleType.METRIC); |
no test coverage detected