EVOLUTION-MANAGER
Edit File: test_tracer.cpython-36.pyc
3 " e� � @ s� d dl T ddlmZ ddlmZ ddlmZmZ ddlm Z ddl m Z m Z m Z mZ yddlmZ W n ddlmZ Y nX G d d � d ej�ZG d d � d e�ZG d d� de �ZG dd� de�ZG dd� de�ZG dd� de�Zddd�Ze dkr�ej � dS )� )�*� )�Tracer)�Rules)� Applications� Application)�AffectedProcess)�ProcessesCollection�PackagesCollection�ApplicationsCollection�AffectedProcessesCollection)�patchc @ s0 e Zd Zdd� Zeddd�dd� �Zdd � Zd S ) � TestRulesc C s> t � t_tt� tttd�| _d| j_tj ddd�� t t _ d S )N)�memoryi� ZkernelT)�name�ignore) r r Z_appsr �PackageManagerMockr �dump_memory_mock�tracerZ timestampZ_append_application� ProcessesMockr Zprocesses_factory)�self� r �!/usr/lib/python3.6/test_tracer.py�setUp s zTestRules.setUpz0tracer.resources.applications.System.init_systemZdummy)Z return_valuec C s> | j j� }| jt|�ttjd�tjd�g�� | j|t� d S )N�baz�qux)r Ztrace_affectedZassertSetEqual�setr �find�assertIsInstancer )r Z init_system�affectedr r r �test_trace_affected s $z TestRules.test_trace_affectedc C sV | j jtjd�t�}| j|t� | jt|�d� |d }| j|t � | j|j d� d S )Nr r r � ) r Ztrace_applicationr r �AffectedProcessMockr r Z assertEqual�lenr �pid)r r �processr r r �test_trace_application s z TestRules.test_trace_applicationN)�__name__� __module__� __qualname__r r r r& r r r r r s r c @ sX e Zd Zdd� Zdd� Zedd� �Zedd� �Zed d � �Zd d � Z d d� Z dd� Z dS )� ProcessMockc C s || _ || _|| _|| _d S )N)r$ �files�_name� _create_time)r r$ r � create_timer+ r r r �__init__+ s zProcessMock.__init__c C s | j S )N)r, )r r r r r 1 s zProcessMock.namec C s | j S )N)r, )r r r r � real_name4 s zProcessMock.real_namec C s dS )NFr )r r r r �is_interpreted8 s zProcessMock.is_interpretedc C s dS )NFr )r r r r � is_session< s zProcessMock.is_sessionc C s | j S )N)r- )r r r r r. @ s zProcessMock.create_timec C s g S )Nr )r r r r �childrenC s zProcessMock.childrenc C s d S )Nr )r r r r �parentF s zProcessMock.parentN) r' r( r) r/ r �propertyr0 r1 r2 r. r3 r4 r r r r r* * s r* c @ s e Zd Zddd�ZdS )r" Nc C s || _ t� | _t� | _d S )N)r$ r Zpackagesr+ )r r$ r r r r/ K s z AffectedProcessMock.__init__)N)r' r( r) r/ r r r r r" J s r" c @ s e Zd Zedd� �ZdS )r c C sN t tddddddg�tddd d d d g�td dddddg�tdddddg�g�S )N� ZfooiW �file1�file2�file3� Zbari' �file10�file11�file12r! r i �file7� r ia �file4�file9)r r* r r r r �allS s zProcessesMock.allN)r' r( r) � staticmethodrB r r r r r R s r c @ s e Zd Zdd� ZdS )� PackageMockc C s || _ || _|| _d S )N)r �modifiedr+ )r r rE r+ r r r r/ ^ s zPackageMock.__init__N)r' r( r) r/ r r r r rD ] s rD c @ s` e Zd Zedddddg�edddd d g�ed d d ddg�edddddg�gZdd� Zdd� ZdS )r �Ai r7 r8 r9 �Bi\ r@ Zfile5Zfile6�Cia r> Zfile8rA �Di�" r; r< r= c s t t� fdd�| j��S )Nc s | j � kS )N)rE )�p)� unix_timer r �<lambda>m s z8PackageManagerMock.packages_newer_than.<locals>.<lambda>)r �filter� _packages)r rK r )rK r �packages_newer_thanl s z&PackageManagerMock.packages_newer_thanc C s$ x | j D ]}|j|kr|jS qW d S )N)rN r r+ )r Zpkg_name�packager r r � package_fileso s z PackageManagerMock.package_filesN)r' r( r) rD rN rO rQ r r r r r d s r Nc C sL i }xBt j� D ]6}x0|jD ]&}||kr6|| j|� q|g||< qW qW |S )N)r rB r+ �append)�userr r% �filer r r r u s r �__main__)N) Z__meta__Ztracer.resources.tracerr Ztracer.resources.rulesr Z tracer.resources.applicationsr r Ztracer.resources.processesr Z tracer.resources.collectionsr r r r Z unittest.mockr ZmockZunittestZTestCaser �objectr* r" r rD r r r' �mainr r r r �<module> s$