Gecko [ thearyasamaj.org ]

Name	Size	Permission
__pycache__	[ DIR ]	drwxr-xr-x
__init__.py	0 B	-rw-r--r--
__meta__.py	310 B	-rw-r--r--
test_FilenameCleaner.py	942 B	-rw-r--r--
test_alpm.py	1.29 KB	-rw-r--r--
test_applications.py	3.91 KB	-rw-r--r--
test_collections.py	2.4 KB	-rw-r--r--
test_dnf.py	1.29 KB	-rw-r--r--
test_dpkg.py	1.26 KB	-rw-r--r--
test_hooks.py	1.01 KB	-rw-r--r--
test_lang.py	238 B	-rw-r--r--
test_package.py	371 B	-rw-r--r--
test_portage.py	1.28 KB	-rw-r--r--
test_processes.py	2.03 KB	-rw-r--r--
test_query.py	574 B	-rw-r--r--
test_rules.py	2.19 KB	-rw-r--r--
test_tracer.py	3.4 KB	-rw-r--r--
test_views.py	9.78 KB	-rw-r--r--
test_yum.py	1.26 KB	-rw-r--r--

Code Editor : test_tracer.py

from .__meta__ import *
from tracer.resources.tracer import Tracer
from tracer.resources.rules import Rules
from tracer.resources.applications import Applications, Application
from tracer.resources.processes import AffectedProcess
from tracer.resources.collections import \
	ProcessesCollection, \
	PackagesCollection, \
	ApplicationsCollection, \
	AffectedProcessesCollection

try:
	from unittest.mock import patch
except:
	from mock import patch

class TestRules(unittest.TestCase):
	def setUp(self):
		Applications._apps = ApplicationsCollection()
		self.tracer = Tracer(PackageManagerMock(), Rules, Applications, memory=dump_memory_mock)
		self.tracer.timestamp = 5555  # Sure, it should be a UNIX timestamp value
		Applications._append_application({"name": "kernel", "ignore": True})
		Application.processes_factory = ProcessesMock

@patch('tracer.resources.applications.System.init_system', return_value="dummy")
	def test_trace_affected(self, init_system):
		affected = self.tracer.trace_affected()
		self.assertSetEqual(set(affected), set([Applications.find("baz"), Applications.find("qux")]))
		self.assertIsInstance(affected, ApplicationsCollection)

def test_trace_application(self):
		affected = self.tracer.trace_application(Applications.find("baz"), AffectedProcessMock)
		self.assertIsInstance(affected, AffectedProcessesCollection)
		self.assertEqual(len(affected), 1)

process = affected[0]
		self.assertIsInstance(process, AffectedProcess)
		self.assertEqual(process.pid, 4)  # pid of "baz" in our mock

class ProcessMock(object):
	def __init__(self, pid, name, create_time, files):
		self.pid = pid
		self.files = files
		self._name = name
		self._create_time = create_time

def name(self):
		return self._name

@property
	def real_name(self):
		return self._name

@property
	def is_interpreted(self):
		return False

@property
	def is_session(self):
                return False

def create_time(self):
		return self._create_time

def children(self):
		return []

def parent(self):
		return None

class AffectedProcessMock(AffectedProcess):
	def __init__(self, pid=None):
		# Do not run the parent __init__
		self.pid = pid
		self.packages = set()
		self.files = set()

class ProcessesMock(object):
	@staticmethod
	def all():
		return ProcessesCollection([
			ProcessMock(2, "foo", 1111, ["file1", "file2", "file3"]),
			ProcessMock(3, "bar", 9999, ["file10", "file11", "file12"]),
			ProcessMock(4, "baz", 6666, ["file7", "file1", "file3"]),
			ProcessMock(5, "qux", 7777, ["file4", "file9"]),
		])

class PackageMock(object):
	def __init__(self, name, modified, files):
		self.name = name
		self.modified = modified
		self.files = files

class PackageManagerMock(object):
	_packages = [
		PackageMock("A", 3333, ["file1", "file2", "file3"]),
		PackageMock("B", 4444, ["file4", "file5", "file6"]),
		PackageMock("C", 7777, ["file7", "file8", "file9"]),
		PackageMock("D", 8888, ["file10", "file11", "file12"]),
	]

def packages_newer_than(self, unix_time):
		return PackagesCollection(filter(lambda p: p.modified >= unix_time, self._packages))

def package_files(self, pkg_name):
		for package in self._packages:
			if package.name == pkg_name:
				return package.files

def dump_memory_mock(user=None):
	memory = {}
	for process in ProcessesMock.all():
			for file in process.files:
				if file in memory:
					memory[file].append(process)
				else:
					memory[file] = [process]
	return memory

if __name__ == '__main__':
	unittest.main()

${this.title}

Code Editor : test_tracer.py