From ff9f5aa9e348345a828bf5eb5b36dad399cf5a56 Mon Sep 17 00:00:00 2001
From: Bernhard Mueller <bernhard.mueller11@gmail.com>
Date: Mon, 2 Oct 2017 17:00:14 +0700
Subject: [PATCH] Add unit tests

---
 tests/__init__.py                          |   0
 tests/asm_test.py                          |   1 +
 tests/contractstorage_test.py              |  26 +++++++++++++++++++++
 tests/ethcontract_test.py                  |   2 ++
 tests/evm_test.py                          |   2 ++
 tests/rpc_test.py                          |   2 ++
 tests/teststorage/contractstorage.fs       | Bin 0 -> 16295 bytes
 tests/teststorage/contractstorage.fs.index | Bin 0 -> 31 bytes
 tests/teststorage/contractstorage.fs.lock  |   1 +
 tests/teststorage/contractstorage.fs.tmp   |   0
 tests/util_test.py                         |   2 ++
 11 files changed, 36 insertions(+)
 create mode 100644 tests/__init__.py
 create mode 100644 tests/asm_test.py
 create mode 100644 tests/contractstorage_test.py
 create mode 100644 tests/ethcontract_test.py
 create mode 100644 tests/evm_test.py
 create mode 100644 tests/rpc_test.py
 create mode 100644 tests/teststorage/contractstorage.fs
 create mode 100644 tests/teststorage/contractstorage.fs.index
 create mode 100644 tests/teststorage/contractstorage.fs.lock
 create mode 100644 tests/teststorage/contractstorage.fs.tmp
 create mode 100644 tests/util_test.py

diff --git a/tests/__init__.py b/tests/__init__.py
new file mode 100644
index 00000000..e69de29b
diff --git a/tests/asm_test.py b/tests/asm_test.py
new file mode 100644
index 00000000..10e61596
--- /dev/null
+++ b/tests/asm_test.py
@@ -0,0 +1 @@
+import unittest
diff --git a/tests/contractstorage_test.py b/tests/contractstorage_test.py
new file mode 100644
index 00000000..61823670
--- /dev/null
+++ b/tests/contractstorage_test.py
@@ -0,0 +1,26 @@
+import unittest
+from mythril.ether.contractstorage import get_persistent_storage
+import os
+
+
+class GetAndSearchContractTestCase(unittest.TestCase):
+
+    def mockCallback(self, code_hash, code, addresses, balances):
+        self.isFound = True
+        pass
+
+    def runTest(self):
+
+        script_path = os.path.dirname(os.path.realpath(__file__))
+        storage_dir = os.path.join(script_path, 'teststorage')
+        storage = get_persistent_storage(storage_dir)    
+
+        contract = storage.get_contract_by_hash(bytes.fromhex("ea061445eacbe86b7ffed2bb9e52075e"))
+
+        self.assertTrue("0x60606040" in contract.code,'error reading contract code from database')
+
+        self.isFound = False
+
+        storage.search("code#PUSH1#", self.mockCallback)
+
+        self.assertTrue(self.isFound,'storage search error')      
diff --git a/tests/ethcontract_test.py b/tests/ethcontract_test.py
new file mode 100644
index 00000000..c5c2eee9
--- /dev/null
+++ b/tests/ethcontract_test.py
@@ -0,0 +1,2 @@
+import unittest
+
diff --git a/tests/evm_test.py b/tests/evm_test.py
new file mode 100644
index 00000000..c5c2eee9
--- /dev/null
+++ b/tests/evm_test.py
@@ -0,0 +1,2 @@
+import unittest
+
diff --git a/tests/rpc_test.py b/tests/rpc_test.py
new file mode 100644
index 00000000..c5c2eee9
--- /dev/null
+++ b/tests/rpc_test.py
@@ -0,0 +1,2 @@
+import unittest
+
diff --git a/tests/teststorage/contractstorage.fs b/tests/teststorage/contractstorage.fs
new file mode 100644
index 0000000000000000000000000000000000000000..a3c113189b8dfe443dd7f0fb554fe26643b88afe
GIT binary patch
literal 16295
zcmd6uORpr!RmXd#dme5HFcKS9%o4F^WoG0PBnx?D1PEltwoq@3m5)eI%k7@2z8A?x
zGWKRH#1{a0!4f2v7_ni;7FK)!#1i=%cw_PVpNPoH%&O{J-8~~uRqJ*>;&I|U{^t=<
zfADpe-T#jt{>%5i_FI2*FMNLT9{>O5y?aldJ-L0-KYefNZ~LLY@$Ze#eSiDpqi3<*
zy<NZb_(8h3_xnG(KYr|=-#odw_0Mh}eYgMk<0sF){qWD0M}Hch9PT~((fuDB-gvCO
zn(y%D4-WTl9>tM<!c71A=l}E9{~YeW^z--Pi}>I6^}RAa{>P=&-@hIqkAIDk#*d!e
zKJUlduit+3y#Ka8Jb3@zIQ|=lx5n>&|MtV@Po6&Vw;%fFk5b#4(Dvc`+rAGEZ~yr4
z&IdQk0^iuJAQ66C!B1JiOU@B_`rD7cu%iCv@a97In}_@3m%s7c`<q8!{i<C&O#dGq
z{P^(J2akW1o==|L-1g5#|E;Hp@#gUM(*6G7oeydD?gx(_((Gw}bNj8~=||&t4qy1=
zUwrY)_x>_T@BW>O`zscAi4d*g{V)FP?ceyy&;IlOzVpLh{P(~A!`I&Vi^CfW84nI`
z?l^-Ck^mn5QKbG)ApYUu{*Mn2K44{^9Nzxq@XjZPcN2{d?xgYWP6TjynhodsX?pH&
zZakd-=HdS1-+@=z_oi{|+quuW!H;>7RlYB}vL33rFY9`0io9?`lg;hnLH}54HuO&=
zgPX%!-#oniw>+8t>8JnpzrWHiOn&pu1$_T?nEciYnf#S+{9)4Y7x4J4yGZ=KUw<Kw
zH?yn(WBh3Hhc_Sp*YAq6b*68bRfVhbs?IaMRa3h;e(KEgJ}a`GznQDDGOue_6#blK
zbKmB9sXjS<+KlIJ$miCV;b~XKr_D4LUE{;kp-WGF;VSR!X^}>7Ra0j9T!p7D3vFFH
z40B&);b~dNr{g$JbvA^jlMheJa_+maufx+hQdW+x=3k(}<!xwNWkWZXMQd$c5l5)H
zsUBPv#&u;Jp|0D$Z;P-tHx6Nhx^aEu(+G)%dd$Xx1&48cq_=U?khNtTAyL*&)7*4L
zn61nr&-$^>{qi*L!)(Lc)>GXDK9mk-iR)80WK}y%Hd~pvGtF5&cUj;=S&gCXJZ8Qr
zr$AX*&s9?mTB1pwHj;9i%~LfTuhRQ``yRO$YF|3_F59B2)eBw9+pLwol;{V7)!447
zn%V39Jm+<rR|W6$I_sl$Wv=Vm9^FFK+5#O!VY;kq^`wNNmPL0tPpoI1wJrB>qN_C$
z?;Gw*?n~YGSzB{oa9`+t%vzL3R~`ZADzD<2v)0)gttQj>>Q`}9**ds)T{fc)i3=S(
zJMu<r>l>|qs*mgMvarI!))+W)I?{-&HzO6<Y+S?65L|{Rv!ZP6t`%KItE-SXTw;N<
zIMwo8oaPby)Lfc9UZa?z1WxeOc+2Wr#}_WQB~pQ7@DMd&1!GsS>f;xAX4$gx;#B^u
zd?!WG?<%C0GfXdzM7{wH=T~dmQ!)0cEyDSwb4I*H+m75oH{fiPjU>yR_`iKZ3rR-&
zN0X&bWrp4zHQ^Pd)0EYzbP{hpn0lf~xy)kC)p>8fE)#mU4I)UEi<!avE3_A^KQm`S
zgO(AF@^o%bU$B|JT5E_aPw1;eA4qbfFK`vCL9+uV(Pinl>3QI@q}G}p#CtIUofn6y
zLG3XQp%RuYR5ZMiRUl1agvpAxjd3q&fDj?fC`-zPVsVsVilbmvs6LvZxflhPia*VO
zBZn$Q)HBVIUJZ-~(&(E|^8_OtsIrx8<Q!NknaS#B{tjiex=&IVt8y)c6Zwa+M*P!J
zR%N!ww}?t+r=vGJjq`?iz$o5<R{C4pUyubYG9FKbQ5H3|bokhfrUo7KeYd9JlKmSE
zelSmyMy&BHn-RmR{Jd$^;6XR70^j^>GUg3|1c4>e&rHg?D!GykqlJ>Gv_n^gc??}S
z^Ch|}&A;fXlIO8@wS1CY$&<zFc}CyxYF!(2gEuW!xI<Ctnc8#>f9sCx*7!rax&rH?
ztWg|L+-Oc3RG~tQT+V1Vn$=YUW1->Li{XF5BCW>ypn<dpT{Na?qx34v{3U)Y^D9!2
zwl{hri!ymv#a4gR3X3z&E<6sLm7UUGkpmo1Twr%8W~ihEiWzczAT(Aj_<&`@Qzu(y
z?p5?)99wu=C{9(sT%L64t6o8lmqrk;bK)Up6+%?_#<kv{9EsNOzSnALM`$Q(pNkL$
ziKB*EFGH=3SPM;{Q!+ctl?g5cu7viLqCnF#I0(v!hj~M9tPx!`tvy|>wOcg(C~Zkb
z$Qtxs@ss=<lG12~s?v)(&qtzvaGjauC5g0zeCtSuU~b!E$~QfPkttePP<KB9$_zh(
zMk2Yh9I5MS%aINa?Kn~_99gg7bt%GQoTsP<<<>GXnc7-5T5`2EqFK@>>BU7ZcygN6
zoDE6To(t|%<U-W%pkrsb(745ZO;f`PY$a<h1P>SyjnqU|G)0fUweh?@r^m8Z!-u9Z
zd@!v`h%+V>3y72GT<p7P_x3G(NeI;V5aW>NnrDah-YJ&WsIm*}fcQ2br)1O<k&7cd
z&%DJp+*MUo>Zz$|+xlWIx*?}vG1TNZeLg4-CxV+uCEIQij*|BR`DBr#g>%u)zH6(o
znake1(y65d<Yo?>32V$}vU6bpR%v!Cn>GaA@gw-QfE#L?ENzXf&Rxa|d6tM2LV#Dp
zhz*WwHoS?jw<T9K?<SOBZA+Ad|5eOawS0g)iW1=)JrHhpjoG{l_#x!Bcue`yl81(O
z<|CP7>&tMu>ygCYqXvn>e*d2AGIDz5>sn=&P}#cFn~~%*<2uI$zf6x?Z@$~wT<D=X
zL$Gz?S^2j%_<Q;&K9F}*mRmAGc;6MoKTrT(o!U@{HWF_@{H^bdeHoLuR-;}TVdpzL
zEMx0CiNV#Mk;pxD)s~SXN53T{l+CCWCO|9Q6N;-n!Hs3-L3*Yo-58_|281!u!_s0P
z3JM^g5PBu`J)$u71uq!%S`?dR=M*1@WFN~qHoAM+PLe-OqgmwM=P9P!(oO)9b+c+l
zpC0YUxuOhN_p8&s_3OJxAb9ncj0KSPU9mv2@+gCL6^Hgqj4qr|1ZOoT$Slab6cK?=
zSR4p)b!JZFWBL9nuO!w^xxgLs*!;c8(B?FK>{T-ix`?h~*({k{g4l`CLRQFI^#YR=
zw}qMw#sKQ8-lOH9i-Tetp^WlS)ob;?`fF4zKI<pdShZ%5z%o@EEynkTmh_LvNBhB1
zcT$yYTRUQO^P|m-Q5NM00k8NGeHMwz4<b4IV^LY=#d=r~e<L+c)=C~?oUj}>qT~Se
zYc!$N+?}IZQVw(nuDm|=;<YlBSSg3!tNI#GovIxsqbzUcx^G-P*VKeb0874E<6Mu^
zCr@v(x+{F$)T|N#`C2WfVx~>ORoPT^j9etoW~R3*W^%KVBxynlt}JO?k`%gn03r%X
zd)|8MNwpVGqCE245oIKQ7xg;ZL<Gmk-w-ySa-NgF8Vc{C3uHufBfPa7AtOYYy&rNa
z<pa(cvZ=I|*+8YG?7SO!SxjYHd)IbuYRYlwy2|@<AX%SV(%q4num)j))y}BU^s1FC
zR3qE1j~;jV|K3NB&9*)Oi!SA(L2t~@SY?I^12?R4gvzz<$=0Yp$>PjULGdvYRor#z
zAu&ti;*v4Y>eGtGBQgdap(~4R4|n-$);7V<_1Zv+wx)iL*~R+Slw*15e?fdpB%@qE
zX2&VgYM01y8y`(>6CV-hCXP;Jv6&TK3gMCukZ<DJY8J?Av95D`J-$VLjIsA?$M+cI
zG7{0_@jAwAhM_C_(|D^^wr8Mwt6C7>*=(kb$@){At&#LTptd2O><uFpRxf$F)d^RT
z;4~7YDXz=NS7mdWDI%x!->*&{LQz++#H?t_v~PFU{}9-PzG$JvSl~ahw@BM+B$ssi
zga+2s733O$mWUPX$=GOCA<`95oU#SYV9yi@@(fKoK2uD{Gj?l^&lLCZjD4Hqvk(;;
z`h=&@)O0JVV@5;xOZ{g>ms#rzDgdD(m9#KP^$Z?K&!`*+>kkp)#tIiZfVaUeudzix
zrrJ-~fpgy114gj6M$po%v1E5|EL@LQKCwvE^m&Sxtrx@(dU`!AbH1(br`&R-y-`Om
zV>8=k8!MMRMf(c%)?_cX!{<W8y|NG0u}u2JNrK{wnFm?CA`6cu;&H&{DvTu!57OZ+
zGiwa4OjF1v$jZnz>CxG4!qNv#Q_qUpp(^zBSnJF#-aY5Bc0FF9&a}}c&2DsW{0vd9
z<$m)V@EIy;MfGIHvbD->YqX(Z4;^W-Fsp9jys4O|RuOjht%b#0%pUiD&&|u|&b8x6
zq6wdZb%a<;QZV0*3rq;4_&c2;kQ_}?Sh{*7%OWpAI>t4uibDieOUBP^^@rV#v98!n
zna#3AImN$~_X=D~qed({$GaHWtT}`ZpbgS#ML`swRXt&)BQ#HWM?GA4uho4vYqjja
zG|hJ3&#YCI-=JA)v0tmO%l56KJ&HTCBJCmUmni!Wz0oFi!Gyyy2wK?QY%@D%f1MAt
z56RcYWz;;?|AMTU3eItWL2j{+HJEs(T?w^HZ^M3>s0%Sjp7+EH74anj;cI>NG{T4A
zag$V6R9r<Pni#E#?RZsOjidRfQzK|`xYL=J$mb{>^S<~t@Jg0OWt<$<u*Ru|wBaLx
z|4Y!Acqr<zD%CGEuB6QNnIxa~OO=%pcHvJf=6FR8U=7$tnw7}KPLtSd(pK#b1T;aM
z%$I>ykq2yn1#T6kc2M#1gaZpFq~YLi1&1tSwZEz7Ih&4nm@B7Nyo#O(54ks#l76kC
zbYi9xJmgSWkW*)tVl6&Tcvwu^_}~;H3cuN34V?RQ@v!hSQO7u<rSeJVG11+P|3riI
z-N)RPS7~2`3SrXI6o*NZz%G-f`hnDnCsge;L|gF8X(wQbHh25_xvOKvK5$1UFU;3?
zv04&P(Xl5TMtS*t{iP;-g_m<9xS=(~b62NQ<%HL|#a^o9*2{d0ZK3<;^*%%mSSg-!
zWhI<SyE1Ah3uTO)AY-dj@|PWbo~(q5%bHt6MEj^g`QD1-dw#O2i<!ONt#Ra1)Ua4d
zZMgP%w313EJuX^FIa7>EQHI48=dGl2$<{twE5XaIWp+lYReeQ#R5#h=vw`!SoLwtX
z-;7N+>aa*0XJp=59>=spc!PFxXf#??%pTVAC%T0J`9SIs_Wn5CVV}6Wy^9rTeOFwJ
zMAo&5gpuk0U9CbM9;~FY5hAab#>$#X>}4(Bg!NTgz_MNyD;1Bh3L!r0e$`bp#7Gx)
zxvuOCL9p`|=yI(n(?)rDU8eZD{+{B#?N=Ui8%`r{^|_!By2)+oDe7VH3u*MDH;<>s
z6eqDO9pA?Y>0I6xk-t%9ze4^dzcxN<iVz=BcK%!<e^cKTA2rQq9v^Y^yq|JU44<;a
zl>HuUzs36=b76};kO6o@pTctE6jz~*=X2w9HsNyAKrw!;VM%{v&u+Ejq8b8b=Q*x1
zxuIYKqHo5rn~QO9Q^BEZH}wtq+EBwg)W&l{V~LytVsfY<&2&cZ$hEH@JJgg!QA^$6
z?4FeNzY6wQ_@w~OmHm@ihvYn8{dVB$k1W)xhdQl6iX3px{JZX#JqOX!b2ekdQ_vJB
z@iV8s=8`?yj6)o3!?e!#e&kmu^TZ*xJ$Df6UEH~L>?^o4Zn>j<CG9}LtBP%xt5f`Q
z+=({NzJ9~$l|E~RaV`dJZ8hMUHV7#K56GtNI?iBq{n+?ZZs>HDMmk&5GF`cL>?^pT
z^J!~c8Q=z)U+u3}6KnV)ZVUt34R<M~&1`rKMd3@G5}G~xu&`wEc~8#KdTo5|b)!F?
zqP}Q3SB`xJH+GhTZpdy>;JTpfb@k^3+~`RT*wf@V?AQ)76>1Lr_85lm9KYacN6*O{
z4l?EApuNr$ZZu0ya#=gB9X;|x@%rALmDo{j_Ic5(Xw$L(jR9BlP*8tx#hABq)pJ~n
zn3F}EP|43EqW`>1<+Y=~h)S=M`MJ=hgqv>c`k8}-{m{+jFpMpBR`x^95$3t)=T19J
z9`tLC;N>q<dF|-usEiTBp#2qmONcwDwJ;kT=Mh`on6qX0wBSTTqQfe^I;e2=l!j%Y
zyE6~Ra~rB^nZLA?VVrG{_ZmX<XZ1xp14~Sa+}S-qgnCb02YsSI*_+Nx=uOLxUVIyO
zkE~6~1Jp}raM&$iCm^jP#!6N#VB|JtNj45Y*oryBz;An)(PB;GOsaK<N`plP7DEZ2
z;Evkp6u+H0feJiR%ti*CB{jUnjRe{?7dwSEuGP3~RqNVTjP&%gr*)(!fF_UFX0X~l
zTy!VZGDprPDaRjko7-A^o^t%LMmENkuv2!L<B!<Rszr!?j<K`2ko0kr&mrhNhc4y#
zb=yv0K00zuq}P!6T@A5k8!AFH>WD2VQN3cUhyn4Ni@tQ6RnOarpB~ozOpKTF%iF5s
zH;S1bJV#N0@)o~U`F3i@VkAxOyN)wA1E+!dnV`b;!;sZYKXJ%~pW%AqO`IIjBav-A
zfux?mvA{Ol(amG;_^D?Lh&3Z8!{u>^Dw~3nt}KR=A^cjg?~BQ^p*e8E1$3*L-WL^p
l+mY@bU@phH%qP!51Rav1d{J^>qy))z!0Y$}_%HqZ{{eGUR`UP=

literal 0
HcmV?d00001

diff --git a/tests/teststorage/contractstorage.fs.index b/tests/teststorage/contractstorage.fs.index
new file mode 100644
index 0000000000000000000000000000000000000000..cb464437599f88b1191c517f24653536af58c484
GIT binary patch
literal 31
dcmZo*_Fksf!0gP%00zz+5SqcH4JhiT2LMor1fc)`

literal 0
HcmV?d00001

diff --git a/tests/teststorage/contractstorage.fs.lock b/tests/teststorage/contractstorage.fs.lock
new file mode 100644
index 00000000..e002abb8
--- /dev/null
+++ b/tests/teststorage/contractstorage.fs.lock
@@ -0,0 +1 @@
+ 69453
diff --git a/tests/teststorage/contractstorage.fs.tmp b/tests/teststorage/contractstorage.fs.tmp
new file mode 100644
index 00000000..e69de29b
diff --git a/tests/util_test.py b/tests/util_test.py
new file mode 100644
index 00000000..c5c2eee9
--- /dev/null
+++ b/tests/util_test.py
@@ -0,0 +1,2 @@
+import unittest
+