Sindbad~EG File Manager
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$
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists