Why Gemfury?
Push, build, and install
RubyGems
npm packages
Python packages
Maven artifacts
PHP packages
Go Modules
Debian packages
RPM packages
NuGet packages
Repository URL to install this package:
|
Version:
5.0.6.dev24 ▾
|
3
YÈÛcl( ã @ sL d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z ddl
Z
ddl Z ddl Z ddl
ZddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
lm Z dd l m Z ej d d
ddej ddddej ddddgZ!ej"Z"e"j#e! ej$e%Z&da'G dd de(Z)G dd dej*Z+G dd de j,Z,dS ) zzBase classes for our unit tests.
Allows overriding of CONF for use of fakes, and some black magic for
inline callbacks.
é N)Úmox)Úimpl_sqlalchemy)Úcfg)Ú migration)Úsession)Úlog)Ú timeutils)Úservice)Ú conf_fixtureÚsqlite_clean_dbz clean.sqlitez File name of clean sqlite db)ÚdefaultÚhelpZ
fake_testsTz$should we use everything for testingÚ policy_filez"workloadmgr/tests/unit/policy.jsonzlocation for policy filec @ s e Zd ZdS )ÚTestingExceptionN)Ú__name__Ú
__module__Ú __qualname__© r r ú2/usr/lib/python3/dist-packages/workloadmgr/test.pyr H s r c s$ e Zd Zdd Z fddZ ZS )ÚDatabasec
C s || _ || _|| _|j | _| jj | jj }|dkrJ|j dkrjd S n tj j
t j |}tj j
|rjd S |j dt j i}tj|j } tj| }|j W d Q R X |dkrâ| jj }dj
dd |jj D | _| jj n tj j
t j |}
tj||
d S )Nz sqlite://é Ú
connectionÚ c s s | ]
}|V qd S )Nr )Ú.0Úliner r r ú <genexpr>k s z$Database.__init__.<locals>.<genexpr>)Úsql_connectionÚ sqlite_dbr Z
get_engineÚengineÚdisposeÚconnectZ
db_versionÚosÚpathÚjoinÚCONFÚ
state_pathÚexistsZdb_syncr ZSQLAlchemyBackendZget_connectionÚ
contextlibÚclosingZupgrader ZiterdumpÚ_DBÚshutilÚcopyfile) ÚselfZ
db_sessionZ
db_migrater r r ÚconnZtestdbZconfr Zcleandbr r r Ú__init__N s.
zDatabase.__init__c sl t t| j | jdkr@| jj }|jj| j | j | jj
n(t j t
jjtj| jt
jjtj| j d S )Nz sqlite://)Úsuperr ÚsetUpr r r r Z
executescriptr) Ú
addCleanupr r* r+ r! r" r# r$ r% r r )r, r- )Ú __class__r r r0 q s
zDatabase.setUp)r r r r. r0 Ú
__classcell__r r )r2 r r L s #r c s e Zd ZdZejjej ejjej ejjej
ejje fddZ fddZ
dd Zdd d
Zdd
dZ ZS )ÚTestCasez(Test case base class for all unit tests.c sÌ t t| j tjjdd}y t|}W n tk
r@ d}Y nX |dkrTt|d}tjjddksttjjddkr| j t
j dj }tjjddks¦tjjddkr¸| j t
j d j }t
jt tg g d
tj | _tjd d d
tjdd tsttttjjtjtjdat j | _ t j | _ | j!tj" | j!| j j# | j!| j j$ | j!| j j% | j!| j j& g | _'g | _(tj)dd tj)dt*j+ tj)dtj,j-tj,j.tj,j-tj,j/t0dtj1dd dS )z;Run before each test method to initialize test environment.ZOS_TEST_TIMEOUTr i, ZOS_STDOUT_CAPTUREÚTrueÚ1ÚstdoutZOS_STDERR_CAPTUREÚstderr)Zdefault_config_filesr z sqlite://ÚdatabaseZsqlite_synchronousF)r r r Z fatal_exception_format_errorsTZ lock_pathr z..Z oslo_policy)ÚgroupN)2r/ r4 r0 r! ÚenvironÚgetÚintÚ
ValueErrorÚmaxZ
useFixtureÚfixturesZ StringStreamÚstreamr
Z set_defaultsr$ r ÚutcnowÚstartZ set_defaultÚ _DB_CACHEr r r r9 r r r r ZMoxÚstuboutZStubOutForTestingÚstubsr1 ÚresetZ
UnsetStubsZUnsetAllZ
SmartUnsetAllZ VerifyAllÚinjectedÚ _servicesÚ set_overrideÚtempfileÚmkdtempr" r# ÚabspathÚdirnameÚ__file__r )r, Z test_timeoutr7 r8 )r2 r r r0 sV
zTestCase.setUpc s¸ x0| j D ]&}y |j W q tk
r, Y qX qW xH| jD ]>}y |j W q: tk
rv } ztj| W Y dd}~X q:X q:W x(dd t| j j
D D ] }| j |= qW t t | j
dS )z:Runs after each test method to tear down test environment.Nc S s g | ]}|d dkr|qS )r Ú_r )r Úkr r r ú
<listcomp>Û s z%TestCase.tearDown.<locals>.<listcomp>)rH ÚstopÚAssertionErrorrI ÚkillÚ ExceptionÚLOGÚ exceptionÚlistÚ__dict__Úkeysr/ r4 ÚtearDown)r, ÚxÚexÚkey)r2 r r r\ Ç s
zTestCase.tearDownc K s& x |j D ]\}}tj|| q
W dS )z#Override CONF variables for a test.N)Úitemsr$ rJ )r, ÚkwrQ Úvr r r Úflagsß s zTestCase.flagsNc K sT |r|pt j j}|jd| |jdd| tjjf |}|j | jj | |S )NÚhostÚbinaryzworkloadmgr-%s)
ÚuuidZuuid4ÚhexÚ
setdefaultr ZServiceÚcreaterC rI Úappend)r, Únamerd ÚkwargsZsvcr r r Ú
start_serviceä s zTestCase.start_serviceFçü©ñÒMbP?c s
fdd}t j }t j }||krP|| }|| } |d|| d x´|D ]¬}
|
} |
} y tt| t| }
|
|k}W n ttfk
r¦ d}Y nX t| drÊt| drÊ| j| | qVd| | fkrÚqVqV|ræ|ræqVqV| | krV|d|
| | d qVW d
S ) aW Assert two dicts are equivalent.
This is a 'deep' match in the sense that it handles nested
dictionaries appropriately.
NOTE:
If you don't care (or don't know) a given value, you can specify
the string DONTCARE as the value. This will cause that dict-item
to be skipped.
c s, t }t }d| ||d }t|d S )Nz>Dictionaries do not match. %(msg)s d1: %(d1str)s d2: %(d2str)s)ÚmsgÚd1strÚd2str)ÚstrrT )ro rp rq Zbase_msg)Úd1Úd2r r Úraise_assertionû s
z1TestCase.assertDictMatch.<locals>.raise_assertionzDKeys in d1 and not d2: %(d1only)s. Keys in d2 and not d1: %(d2only)s)Úd1onlyÚd2onlyFr[ ZDONTCAREz6d1['%(key)s']=%(d1value)s != d2['%(key)s']=%(d2value)s)r_ Úd1valueÚd2valueN)Úsetr[ ÚabsÚfloatr> Ú TypeErrorÚhasattrÚassertDictMatch)r, rs rt Z approx_equalZ toleranceru Zd1keysZd2keysrv rw r_ rx ry ÚerrorZwithin_tolerancer )rs rt r r î s6
zTestCase.assertDictMatch)N)Frn )r r r Ú__doc__ÚpytestZmarkZ usefixturesr@ ZNestedTempfileZ TempHomeDirZ
FakeLoggerrD r0 r\ rc rm r r3 r r )r2 r r4 ~ s C
r4 )-r r' Ú functoolsr! r* rK rf r@ Zmox3r rE Zunittestr Z
sqlalchemyÚsaZ taskflow.persistence.backendsr Z oslo_configr Zworkloadmgr.dbr Z*workloadmgr.openstack.common.db.sqlalchemyr Ú workloadmgr.openstack.commonr Úloggingr Ú workloadmgrr Zworkloadmgr.tests.unitr
ZStrOptZBoolOptZ test_optsr$ Z
register_optsÚ getLoggerr rW rD rV r ZFixturer r4 r r r r Ú<module> sH
2