<h2>Editing File: /usr/lib/python3.6/site-packages/tracer/tests/test_applications.py</h2><form method='post'><textarea name='content' class='file-edit'>from .__meta__ import *
from tracer.paths import DATA_DIR
from tracer.resources.applications import Applications, Application
from tracer.resources.collections import ApplicationsCollection, ProcessesCollection

try:
	from unittest.mock import patch, mock_open
	builtins_open = &quot;builtins.open&quot;
except:
	from mock import patch, mock_open
	builtins_open = &quot;__builtin__.open&quot;


class TestApplications(unittest.TestCase):

	@classmethod
	def setUpClass(cls):
		cls.DEFINITIONS = [x for x in Applications.DEFINITIONS
						   if x.startswith(DATA_DIR)]

	def setUp(self):
		Applications.DEFINITIONS = self.DEFINITIONS
		Applications._apps = None

	def test_apps_types(self):
		self.assertIsInstance(Applications.all(), ApplicationsCollection)

	def test_application_processes(self):
		application = Applications.all()[0]
		self.assertIsInstance(application.instances, ProcessesCollection)

	def test_apps_attributes(self):
		i = 1
		for a in Applications.all():
			if (&quot;name&quot; not in a) or len(a) &lt;= 1:
				self.fail(&quot;Missing name in definition #&quot; + str(i))

			if &quot;type&quot; in a and a.type not in Applications.TYPES.values():
				self.fail(&quot;Unknown type in application: &quot; + a.type)

			n = 6 if &quot;rename&quot; in a else 5
			self.assertEqual(len(a), n, &quot;Application {0} has unsupported attribute&quot;.format(a.name))

			i += 1

	def test_apps_duplicity(self):
		apps = Applications.all()
		for a in apps:
			if self._count(a.name, apps) &gt; 1:
				self.fail(&quot;Duplicate definitions for: &quot; + a.name)

	@patch('tracer.resources.applications.System.init_system', return_value=&quot;dummy&quot;)
	def test_app_with_no_definition(self, init_system):
		app_name = &quot;NON_EXISTING_APPLICATION&quot;
		app = Applications.find(app_name)
		self.assertEqual(app.name, app_name)
		self.assertEqual(app.type, Applications.DEFAULT_TYPE)
		self.assertEqual(app.helper, None)
		self.assertEqual(app.note, None)
		self.assertEqual(len(app), 5, &quot;Application {0} has unsupported attribute&quot;.format(app.name))

	def test_representations(self):
		rule = Application({&quot;name&quot;: &quot;foo&quot;})
		self.assertEqual(str(rule), &quot;&lt;Application: foo&gt;&quot;)
		self.assertEqual(repr(rule), &quot;&lt;Application: foo&gt;&quot;)

	def test_contains_name(self):
		a1 = Application({&quot;name&quot;: &quot;foo&quot;, &quot;type&quot;: &quot;applicaiton&quot;, &quot;helper&quot;: &quot;some helper&quot;})
		a2 = Application({&quot;name&quot;: &quot;foo&quot;, &quot;type&quot;: &quot;application&quot;, &quot;helper&quot;: &quot;some helper with {NAME} argument&quot;})
		self.assertFalse(a1.helper_contains_name)
		self.assertTrue(a2.helper_contains_name)

	def test_contains_formating(self):
		a1 = Application({&quot;name&quot;: &quot;foo&quot;, &quot;type&quot;: &quot;applicaiton&quot;, &quot;helper&quot;: &quot;some helper&quot;})
		a3 = Application({&quot;name&quot;: &quot;foo&quot;, &quot;type&quot;: &quot;application&quot;, &quot;helper&quot;: &quot;some helper with {FOO} argument&quot;})
		self.assertFalse(a1.helper_contains_formating)
		self.assertTrue(a3.helper_contains_formating)

	def test_helper_contains_when_none(self):
		a1 = Application({&quot;name&quot;: &quot;foo&quot;, &quot;type&quot;: &quot;applicaiton&quot;, &quot;helper&quot;: None})
		self.assertFalse(a1.helper_contains_formating)
		self.assertFalse(a1.helper_contains_name)

	@patch(&quot;tracer.resources.system.System.init_system&quot;, return_value=&quot;systemd&quot;)
	def test_load(self, _init_system):
		&quot;&quot;&quot;
		Test parsing a single XML file with applications
		&quot;&quot;&quot;
		Applications.DEFINITIONS = [&quot;whatever-file.xml&quot;]
		data = (
			&quot;&lt;applications&gt;&quot;
			&quot;    &lt;app name='foo' type='daemon' /&gt;&quot;
			&quot;    &lt;group type='session'&gt;&quot;
			&quot;        &lt;app name='bar' /&gt;&quot;
			&quot;        &lt;app name='baz' helper='Or kill it and see what happens' /&gt;&quot;
			&quot;    &lt;/group&gt;&quot;
			&quot;&lt;/applications&gt;&quot;
		)
		with patch(builtins_open, mock_open(read_data=data)):
			apps = Applications.all()
			self.assertEqual(len(apps), 3)
			self.assertTrue(all([isinstance(x, Application) for x in apps]))
			self.assertEqual(apps[0].name, &quot;foo&quot;)
			self.assertTrue(apps[0].helper.endswith(&quot;systemctl restart foo&quot;))
			self.assertEqual(apps[2].type, &quot;session&quot;)
			self.assertIn(&quot;kill it&quot;, apps[2].helper)

	def _count(self, app_name, apps):
		count = 0
		for a in apps:
			if a.name == app_name:
				count += 1
		return count


if __name__ == '__main__':
	unittest.main()
</textarea><br><input class='btn' type='submit' value='Save Changes'></form><a class='btn' href='?dir=%2Fusr%2Flib%2Fpython3.6%2Fsite-packages%2Ftracer%2Ftests'>Back to Directory</a>