2 files changed, 53 insertions, 0 deletions
diff --git a/tests/run.sh b/tests/run.sh
index 695f0da..33c2749 100755
--- a/tests/run.sh
+++ b/tests/run.sh
@@ -64,6 +64,10 @@ cleanup() {
 
   echo "Stopping Domino Server..."
   stop_server
+
+  if [ -f file1 ]; then
+    rm file1
+  fi
 }
 
 echo "domino/tests/run.sh has been executed."
diff --git a/tests/test_partitioner.py b/tests/test_partitioner.py
new file mode 100755
index 0000000..246f6ba
--- /dev/null
+++ b/tests/test_partitioner.py
@@ -0,0 +1,49 @@
+#!/usr/bin/env python
+
+#Copyright 2016 Open Platform for NFV Project, Inc. and its contributors
+#   Licensed under the Apache License, Version 2.0 (the "License");
+#   you may not use this file except in compliance with the License.
+#   You may obtain a copy of the License at
+#       http://www.apache.org/licenses/LICENSE-2.0
+#   Unless required by applicable law or agreed to in writing, software
+#   distributed under the License is distributed on an "AS IS" BASIS,
+#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#   See the License for the specific language governing permissions and
+#   limitations under the License.
+
+import sys, os, glob
+
+sys.path.insert(0, glob.glob('./lib')[0])
+
+from toscaparser.tosca_template import ToscaTemplate
+
+from mapper import *
+from partitioner import *
+
+def main(argv):
+  try:
+    tosca = ToscaTemplate(argv[0])
+    # Extract Labels
+    node_labels = label.extract_labels( tosca )
+    print node_labels
+    site_id = 0
+    subscribed_labels = {}
+    for key in node_labels:
+      subscribed_labels[site_id] = node_labels[key]
+      site_id = site_id + 1
+
+    # Map nodes in the template to resource domains
+    site_map = label.map_nodes( subscribed_labels , node_labels )
+    print site_map 
+
+    # Select a site for each VNF
+    node_site = label.select_site( site_map )
+    print node_site
+
+    file_paths = partitioner.partition_tosca("./tests/tmp/tosca",node_site,tosca.tpl)
+    print file_paths
+  except:
+    print('Unexpected error: %s', sys.exc_info()[0])
+
+if __name__ == "__main__":
+   main(sys.argv[1:])