RIC-769: Committing individual files rather than tar archive

[ric-plt/appmgr.git] / xapp_orchestrater / dev / xapp_onboarder / tests / test_trim_thread.py

diff --git a/xapp_orchestrater/dev/xapp_onboarder/tests/test_trim_thread.py b/xapp_orchestrater/dev/xapp_onboarder/tests/test_trim_thread.py
new file mode 100644 (file)
index 0000000..f4ca4e7
--- /dev/null
+++ b/xapp_orchestrater/dev/xapp_onboarder/tests/test_trim_thread.py
@@ -0,0 +1,41 @@
+################################################################################
+#   Copyright (c) 2020 AT&T Intellectual Property.                             #
+#                                                                              #
+#   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 time
+import os
+import shutil
+from xapp_onboarder.server import settings
+from xapp_onboarder.helm_controller.artifacts_manager import format_artifact_dir_size, get_dir_size, trim_artifact_dir
+def test_trim_thread():
+    path = settings.CHART_WORKSPACE_PATH
+    if os.path.exists(path):
+        shutil.rmtree(path)
+    os.makedirs(path)
+
+    size_limit = format_artifact_dir_size()
+    f = open(path+'/mock_size_file',"wb")
+    f.seek(int(size_limit)+100000000)
+    f.write(b"\0")
+    f.close()
+
+    artifact_dir_size = get_dir_size(start_path=path)
+    assert size_limit < artifact_dir_size, 'Fail to write large files'
+
+    trim_artifact_dir()
+    artifact_dir_size = get_dir_size(start_path=path)
+
+    assert size_limit > artifact_dir_size, 'Fail to trim'
+
+