From 9a44349bfb8fcbbe25974f6650fef9ea0e9fd13f Mon Sep 17 00:00:00 2001
From: Siddharth Ravikumar <sravik@bgsu.edu>
Date: Thu, 12 Feb 2015 21:54:27 -0500
Subject: tests/utils.py: copied the utility functions in tests/events_test.py
to this module.
These utility functions will be deleted from tests/events_test.py
---
tests/utils.py | 83 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++--
1 file changed, 81 insertions(+), 2 deletions(-)
diff --git a/tests/utils.py b/tests/utils.py
index b8ba456..8b974ed 100644
--- a/tests/utils.py
+++ b/tests/utils.py
@@ -21,8 +21,8 @@ import yaml
from nose.tools import *
from os import path, remove, rmdir
-from combox.config import config_cb
-
+from combox.config import config_cb, get_nodedirs
+from combox.file import relative_path
def get_input_func():
"""Returns the input function
@@ -55,3 +55,82 @@ def get_config():
input_func = get_input_func()
return config_cb(config_dir, pass_func, input_func, write=False)
+
+
+def shardedp(f):
+ """Checks if file's shards exists in the node directories"""
+
+ config = get_config()
+ nodes = get_nodedirs(config)
+ i = 0
+ for node in nodes:
+ rel_path = relative_path(f, config)
+ shard = "%s.shard%s" % (path.join(node, rel_path), i)
+ i += 1
+ assert path.isfile(shard)
+
+
+def dirp(d):
+ """
+ Checks if the directory was created under node directories
+ """
+ config = get_config()
+ nodes = get_nodedirs(config)
+ for node in nodes:
+ rel_path = relative_path(d, config)
+ directory = path.join(node, rel_path)
+ assert path.isdir(directory)
+
+
+def renamedp(old_p, new_p):
+ """
+ Checks if the file shards or directory were/was renamed in the under the node directories.
+
+ old_p: old path to directory or file under combox directory.
+ new_p: new (present) path to the directory or file under combox directory.
+ """
+
+ config = get_config()
+ nodes = get_nodedirs(config)
+
+ is_dir = True if path.isdir(new_p) else False
+ i = 0
+
+ for node in nodes:
+ old_rel_path = relative_path(old_p, config)
+ new_rel_path = relative_path(new_p, config)
+
+ if is_dir:
+ old_path = path.join(node, old_rel_path)
+ new_path = path.join(node, new_rel_path)
+ else:
+ old_path = "%s.shard%s" % (path.join(node, old_rel_path), i)
+ new_path = "%s.shard%s" % (path.join(node, new_rel_path), i)
+ i += 1
+
+ assert not path.exists(old_path)
+ assert path.exists(new_path)
+
+
+def path_deletedp(p, is_dir=False):
+ """
+ Checks if the directory or respective file shards is deleted under node directories.
+
+ p: path to the directory or file, under the combox directory, that was deleted.
+ is_dir: set to True if `p' denotes a deleted directory. Default value is False.
+ """
+
+ config = get_config()
+ nodes = get_nodedirs(config)
+
+ i = 0
+ for node in nodes:
+ rel_path = relative_path(p, config)
+
+ if is_dir:
+ path_ = path.join(node, rel_path)
+ else:
+ path_ = "%s.shard%s" % (path.join(node, rel_path), i)
+ i += 1
+
+ assert not path.exists(path_)
--
cgit v1.2.3