def test_get_node_tokens_vnodes(self):
     with open("tests/resources/restore_node_tokenmap_vnodes.json",
               'r') as f:
         tokens = restore_node.get_node_tokens('node3.mydomain.net', f)
         self.assertEqual(tokens, ['2', '3'])
 def test_get_node_tokens(self):
     with open("tests/resources/restore_node_tokenmap.json", 'r') as f:
         tokens = restore_node.get_node_tokens('node3.mydomain.net', f)
         self.assertEqual(tokens, ['3074457345618258400'])