| # SPDX-License-Identifier: Apache-2.0 |
| # |
| # Modifications by Apache Solr contributors; see git log for details. |
| # Licensed under the Apache License, Version 2.0. |
| # |
| # The OpenSearch Contributors require contributions made to |
| # this file be licensed under the Apache-2.0 license or a |
| # compatible open source license. |
| # Modifications Copyright OpenSearch Contributors. See |
| # GitHub history for details. |
| # Licensed to Elasticsearch B.V. under one or more contributor |
| # license agreements. See the NOTICE file distributed with |
| # this work for additional information regarding copyright |
| # ownership. Elasticsearch B.V. licenses this file to you 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 os |
| import unittest.mock as mock |
| from unittest import TestCase |
| |
| import psutil |
| |
| from solrorbit.utils import process |
| |
| |
| class ProcessTests(TestCase): |
| class Process: |
| def __init__(self, pid, name, cmdline): |
| self.pid = pid |
| self._name = name |
| self._cmdline = cmdline |
| self.killed = False |
| |
| def name(self): |
| return self._name |
| |
| def cmdline(self): |
| return self._cmdline |
| |
| def kill(self): |
| self.killed = True |
| |
| def status(self): |
| if self.killed: |
| raise psutil.NoSuchProcess(self.pid) |
| else: |
| return "running" |
| |
| @mock.patch("psutil.process_iter") |
| def test_find_other_benchmark_processes(self, process_iter): |
| benchmark_os_5_process = ProcessTests.Process(100, "java", |
| ["/usr/lib/jvm/java-8-oracle/bin/java", "-Xms2g", "-Xmx2g", "-Enode.name=benchmark-node0", |
| "org.elasticsearch.bootstrap.Elasticsearch"]) |
| benchmark_os_1_process = ProcessTests.Process(101, "java", |
| ["/usr/lib/jvm/java-8-oracle/bin/java", |
| "-Xms2g", "-Xmx2g", |
| "-Des.node.name=benchmark-node0", |
| "org.elasticsearch.bootstrap.Elasticsearch"]) |
| metrics_store_process = ProcessTests.Process(102, "java", ["/usr/lib/jvm/java-8-oracle/bin/java", "-Xms2g", "-Xmx2g", |
| "-Des.path.home=~/benchmark/metrics/", |
| "org.elasticsearch.bootstrap.Elasticsearch"]) |
| random_python = ProcessTests.Process(103, "python3", ["/some/django/app"]) |
| other_process = ProcessTests.Process(104, "init", ["/usr/sbin/init"]) |
| benchmark_process_p = ProcessTests.Process(105, "python3", ["/usr/bin/python3", "~/.local/bin/solr-orbit"]) |
| benchmark_process_e = ProcessTests.Process(107, "solr-orbit", ["/usr/bin/python3", "~/.local/bin/solr-orbit"]) |
| benchmark_process_mac = ProcessTests.Process(108, "Python", ["/Python.app/Contents/MacOS/Python", |
| "~/.local/bin/solr-orbit"]) |
| # fake own process by determining our pid |
| own_benchmark_process = ProcessTests.Process( |
| os.getpid(), "Python", |
| ["/Python.app/Contents/MacOS/Python", |
| "~/.local/bin/solr-orbit"]) |
| night_benchmark_process = ProcessTests.Process(110, "Python", ["/Python.app/Contents/MacOS/Python", "~/.local/bin/night_benchmark"]) |
| |
| process_iter.return_value = [ |
| benchmark_os_1_process, |
| benchmark_os_5_process, |
| metrics_store_process, |
| random_python, |
| other_process, |
| benchmark_process_p, |
| benchmark_process_e, |
| benchmark_process_mac, |
| own_benchmark_process, |
| night_benchmark_process, |
| ] |
| |
| self.assertEqual([benchmark_process_p, benchmark_process_e, benchmark_process_mac], |
| process.find_all_other_benchmark_processes()) |
| |
| @mock.patch("psutil.process_iter") |
| def test_find_no_other_benchmark_process_running(self, process_iter): |
| metrics_store_process = ProcessTests.Process(102, "java", ["/usr/lib/jvm/java-8-oracle/bin/java", "-Xms2g", "-Xmx2g", |
| "-Des.path.home=~/benchmark/metrics/", |
| "org.elasticsearch.bootstrap.Elasticsearch"]) |
| random_python = ProcessTests.Process(103, "python3", ["/some/django/app"]) |
| |
| process_iter.return_value = [ metrics_store_process, random_python] |
| |
| self.assertEqual(0, len(process.find_all_other_benchmark_processes())) |
| |
| @mock.patch("psutil.process_iter") |
| def test_kills_only_benchmark_processes(self, process_iter): |
| benchmark_os_5_process = ProcessTests.Process(100, "java", |
| ["/usr/lib/jvm/java-8-oracle/bin/java", "-Xms2g", "-Xmx2g", "-Enode.name=benchmark-node0", |
| "org.elasticsearch.bootstrap.Elasticsearch"]) |
| benchmark_os_1_process = ProcessTests.Process(101, "java", |
| ["/usr/lib/jvm/java-8-oracle/bin/java", |
| "-Xms2g", "-Xmx2g", |
| "-Des.node.name=benchmark-node0", |
| "org.elasticsearch.bootstrap.Elasticsearch"]) |
| metrics_store_process = ProcessTests.Process(102, "java", ["/usr/lib/jvm/java-8-oracle/bin/java", "-Xms2g", "-Xmx2g", |
| "-Des.path.home=~/benchmark/metrics/", |
| "org.elasticsearch.bootstrap.Elasticsearch"]) |
| random_python = ProcessTests.Process(103, "python3", ["/some/django/app"]) |
| other_process = ProcessTests.Process(104, "init", ["/usr/sbin/init"]) |
| benchmark_process_p = ProcessTests.Process(105, "python3", ["/usr/bin/python3", "~/.local/bin/solr-orbit"]) |
| # On Linux, process names are truncated to 15 chars; "solr-orbit" (9 chars) fits within that limit. |
| benchmark_process_l = ProcessTests.Process(106, "solr-orbit", ["/usr/bin/python3", "~/.local/bin/solr-orbit"]) |
| benchmark_process_e = ProcessTests.Process(107, "solr-orbit", ["/usr/bin/python3", "~/.local/bin/solr-orbit"]) |
| benchmark_process_mac = ProcessTests.Process(108, "Python", ["/Python.app/Contents/MacOS/Python", |
| "~/.local/bin/solr-orbit"]) |
| # fake own process by determining our pid |
| own_benchmark_process = ProcessTests.Process( |
| os.getpid(), "Python", |
| ["/Python.app/Contents/MacOS/Python", "~/.local/bin/solr-orbit"]) |
| night_benchmark_process = ProcessTests.Process(110, "Python", ["/Python.app/Contents/MacOS/Python", "~/.local/bin/night_benchmark"]) |
| |
| process_iter.return_value = [ |
| benchmark_os_1_process, |
| benchmark_os_5_process, |
| metrics_store_process, |
| random_python, |
| other_process, |
| benchmark_process_p, |
| benchmark_process_l, |
| benchmark_process_e, |
| benchmark_process_mac, |
| own_benchmark_process, |
| night_benchmark_process, |
| ] |
| |
| process.kill_running_benchmark_instances() |
| |
| self.assertFalse(benchmark_os_5_process.killed) |
| self.assertFalse(benchmark_os_1_process.killed) |
| self.assertFalse(metrics_store_process.killed) |
| self.assertFalse(random_python.killed) |
| self.assertFalse(other_process.killed) |
| self.assertTrue(benchmark_process_p.killed) |
| self.assertTrue(benchmark_process_l.killed) |
| self.assertTrue(benchmark_process_e.killed) |
| self.assertTrue(benchmark_process_mac.killed) |
| self.assertFalse(own_benchmark_process.killed) |
| self.assertFalse(night_benchmark_process.killed) |
