From 3bf9c08834a0fab591130fc82d8c6c81b06a6ed8 Mon Sep 17 00:00:00 2001
From: Matheus Rocha <matheusrochacamarg@gmail.com>
Date: Fri, 10 Dec 2021 10:21:55 -0300
Subject: [PATCH] test: Final tests of the MVP's URLs and Views

---
 backend/authentication/tests/test_urls.py     |   8 +-
 backend/authentication/tests/test_views.py    |  84 ++++++++++++++++++
 backend/db.sqlite3                            | Bin 159744 -> 159744 bytes
 .../__pycache__/svr_model.cpython-38.pyc      | Bin 4400 -> 4394 bytes
 .../setups/ai_models/strategies/svr_model.py  |   3 +-
 backend/setups/tests/test_urls.py             |  11 ++-
 backend/setups/tests/test_views.py            |  41 +++++++++
 backend/setups/views.py                       |   6 +-
 backend/tools/tests/test_urls.py              |   3 +-
 backend/tools/tests/test_views.py             |  22 +++++
 10 files changed, 161 insertions(+), 17 deletions(-)
 create mode 100644 backend/authentication/tests/test_views.py
 create mode 100644 backend/setups/tests/test_views.py
 create mode 100644 backend/tools/tests/test_views.py

diff --git a/backend/authentication/tests/test_urls.py b/backend/authentication/tests/test_urls.py
index d9906b5..f4106d9 100644
--- a/backend/authentication/tests/test_urls.py
+++ b/backend/authentication/tests/test_urls.py
@@ -11,16 +11,16 @@ def setUp(self):
     self.register_url = reverse('auth register')
     self.user_info_url = reverse('auth user info')
 
-  def test_root_url_GET(self):
+  def test_root_url_resolve(self):
     self.assertEquals(resolve(self.root_url).func.view_class, UserAuthentication)
 
-  def test_login_url_GET(self):
+  def test_login_url_resolve(self):
     self.assertEquals(resolve(self.login_url).func, UserAuthentication.login)
 
-  def test_register_url_GET(self):
+  def test_register_url_resolve(self):
     self.assertEquals(resolve(self.register_url).func.view_class, UserAccountCRUD.Create)
     
-  def test_user_info_url_GET(self):
+  def test_user_info_url_resolve(self):
     self.assertEquals(resolve(self.user_info_url).func.view_class, UserInfoCRUD.Create)
 
   
diff --git a/backend/authentication/tests/test_views.py b/backend/authentication/tests/test_views.py
new file mode 100644
index 0000000..4ff633f
--- /dev/null
+++ b/backend/authentication/tests/test_views.py
@@ -0,0 +1,84 @@
+from django.test import TestCase, Client
+from django.urls import reverse
+from django.contrib import auth
+
+from authentication.views import UserAccount
+
+class TestAuthenticationViews(TestCase):
+  def setUp(self):
+    self.client = Client()
+    self.root_url = reverse('auth homepage')
+    self.login_url = reverse('auth login')
+    self.register_url = reverse('auth register')
+    self.user_info_url = reverse('auth user info')
+
+    self.user = UserAccount.objects.create(
+      cpf = "51937463060",
+      password = "123",
+      username = "thisisanusername",
+      email = "thisisanemail@gmail.com",
+      phone = "42977451163",
+      user_info = None
+    )
+
+    auth.models.User.objects.create_user(
+      username= "thisisanusername", email = "thisisanemail@gmail.com", password = "123")
+
+  def test_root_url_GET(self):
+    response = self.client.get(self.root_url)
+    self.assertEquals(response.status_code, 200)
+
+  def test_login_url_GET(self):
+    response = self.client.get(self.login_url)
+    self.assertEquals(response.status_code, 200)
+
+  def test_register_url_GET(self):
+    response = self.client.get(self.register_url)
+    self.assertEquals(response.status_code, 200)
+
+  def test_user_info_url_GET(self):
+    response = self.client.get(self.user_info_url)
+    self.assertEquals(response.status_code, 200)
+  
+  def test_login_url_POST(self):
+    response = self.client.post(
+      self.login_url,
+      {
+        "cpf": "51937463060",
+        "password": "123"
+      }
+      ,
+      'application/json'
+    )
+
+    self.assertEquals(response.status_code, 200)
+
+  def test_register_url_POST(self):
+    response = self.client.post(
+      self.register_url,
+      {
+          "cpf": "20906291003",
+          "password": "123",
+          "username": "supermegausername",
+          "email": "supermegaemail@gmail.com",
+          "phone": "42944578814",
+          "user_info": ""
+      }
+    )
+
+    self.assertEquals(response.status_code, 201)
+  
+  def test_user_info_url_POST(self):
+    response = self.client.post(
+      self.user_info_url,
+      {
+        "city": "São Paulo",
+        "state": "São Paulo",
+        "street": "Avenida Faria Lima",
+        "neighbourhood": "",
+        "age": 20
+      }
+    )
+
+    self.assertEquals(response.status_code, 201)
+
diff --git a/backend/db.sqlite3 b/backend/db.sqlite3
index c76e4ade965efc7b87549b39c888d4e515308004..5ca9fcedcd6009f1e69b61f06de641f6e9bc01fb 100644
GIT binary patch
delta 2852
zcmah~TWlj|753P%V>=%Io2Js1?uPElMy)Vw&;8;ObsgKW9Xr0?;+Hbc`0kmpJ!2>K
zi#OtB9~MCsM5OLR9}o{{wGvg3@(2qmMM4xoB~Tv_D<Oms52!_`MGJ`Elg%cZ#xS<d
z*wTFGeE&J$`R2}J+nvX@Z(TzkzI8o_Jbd?S|MXhyFQcZ{ZHEB?xq2}2g)A#q%_b5#
zc;r|)`q1aaT-Gle9E<l1AnW`t`YF2RczfxI{h{4q>n`52GE2FooaK`v{K^NGDSYs4
z!~UAVdJXkiZ2`1b?TB^aO|>ZyG~*@-3|{WWX208QWw0HZZiOazqM3*^N-;N`O9{j5
zBpjV69<5O?7b68e<*C-<8(Xbpi7He5^oY@#;oe4UP1voeid1`gyWNn4R@b|uOeF#%
z@LM=>3*&C!jGyNGB<bZz4x{n?YbdHydlmJ}rG_b7Eyqu^%z7v^+)%Mf1PkS|?wr_Z
zuMhjBW+XDM<daxW8;o{}B1x14V#0(&V@lu(!CrG3!2)Y-MHbZRK+&4DfvW#`#_s|>
ziuU8Amt`29Wv#z4ICOd*)U*GE7fz_IqWD~@bVBU~iBhm08<Ya&vPMf;Y>F}NezzcZ
zqkF0#m3%D5lh|1C^?Vt=)T=7VA~U3Yqwz*1&>!{#YeTg*kOirBdcT|~>!%n$iF+9e
zc9&e*nm^*{vl&Noq+G<4xnY%$)uMDFx7`?|6?eb0i)+20k4ldC3>nzj$@j>rM1&JL
zGD@g*OxVld-F9q`DmLcL__-dkevF1+FT+tdhV4&R?I(?cO`O}f9GxJNLWy6G^eWQ0
znp>}~4>Z}`%|~)lB`1eVb=Jo)V%FCmPEwvNGM9>mk{nya#yOIV=SfISYqr#uDx9{h
zHk03N7^k`lpJNC+2SvzwX$)s*a{r6A6M`<3o=cEwa<OzQm<)6(^<XSIju%3Mdf46D
zix*?ttdy+xD`c;mO*aKiiIjzgQj+9QZd=uc)6h27h=*l$tv?jBfwI$*C(2L~m6oav
zYU<_VW&8y1$2czsn@Dl{tJV{SH~$C2;)Y6O@L+U2ASf(9%!_3$I0?IhQxb~~)<rE}
zVw6TB-wbxPXwj1%2KPL^XfQ~Nz3`yC)fO_%&e|ERvtM%wa!`U^jG<_TH5d#Q!~Xu4
zR%SE<R)D(B8ach-G~0nftj@~CL3VxEnv{cbN#hC8J*mip8c_^u;jk~&o7D1|NjhHM
zX4JH=klNjnGt8zc6(o-7Q)}l9g`F7+&Uy)!Wk_BSK1X*&_Z_$|yq_ezB*{_)bM&Nh
z1zCo61fjT};=DXV^Bj9{U3vo=h;E3rM0R&0OEz{!k;-nn-pZC@vFymrXmM4Hk|Cj(
z5_6NrdUiY(6a83BWD~`4fy`AZ1)iP+Qn|EpfhLK=T3()jnqv-rB+Z;kFO-UEa-ZPh
z-BFqwj3R-}U^7wLb^9c#>kFp}ZG2b}cA2^~khg=QawE^j!+mOkm&(<Ewo}ZDG<LBN
z>&Iy?jpH;)9*yO@hzs})4)B@*JO&?tF!(EY4g3VW3qA#VKm>O|9DeQ|-MsOcb2@wQ
zsee0{O|K%)*(M6gh9#vZsA5ksn*buLI83X^>At-H4xeF{S4<w{!amx8$%UNT0Grcv
z6FF0v6-wykuiu{@-=89_8!y?xdj{|q@JH}_@H_Bp@C)!Bcmn<b{&uu{=lh^!HX+D@
zU5_?BTJ^Z7M~fcKdR&-At4WW>1ruUhw8J`f@V){33!eXD*o!}bKY)+Ghv1jseVF_K
zD9j!<>&eY}a<iV?tS2|?$<2Clv!2{+S}@rVqse50CIVBVj|{N6Q@{d?^XJZQIP*>y
z`U(07w)cDJ)MvLD0?XG>1VPSA7%mbG<>}>c24i@feL4Y{-)FlN03DX{)3lc*ND=~=
za{%i+0O_ZvgqNmyieYg248bnW5T1*iU<uq$5MGMoDUQX7r*jw74xGNALT<ce2K3Db
zVgv<6*lBkKwJxiI*qOGwB76-)>{rbY3)s0>7N<Cz6^BhIwzeuXW%#a#KVgy?+P5(8
zkSCNGjG$qLvomg(Ai%{2F4klG6y+!2dw`@dmgg~wCpebCDD2Y8PzT4;Um^Fgm(T|K
zTl5}~oZm#BI6nf%&i^=n=xjSagH!&Q!|gV0A!9*SJDl3Da$=PoPWG6FHXPS6Owllg
ztq$em>!`7DWtlkVKrF>G?D4l(jRWJ-#|F5m-vMueSDeSow~aq_#*-_q*Il+tdEvxY
z%!^>f<#O2soFFNJKfTVGAx}{R&OyT=$ir8B#y?vg7~oz#ICfyh1F#I0dtK14dVRL1
z!toE@Fj}tI&p^jNP8n}pz48Cfb@<lbkrxkNY#86!9O3C*e<52SYkS4b^)g?J^rTEV
m(%=)FXjj1VSVq~-XxUnZOe7*Zx3*3<g7|z>{@pN|tp5eoBLXP^

delta 946
zcmah{OH30{6rB%HN;~r)pon0sp*zy@=5;!s5`u-{Q?M=nPZNVfr=9-)MarfSH!e(+
z)YwE8EZiGOO}a!~xYQ666L;*4VQB~&)HpODLNxK@EpBpi?#X%QY#it}4)l*Y@#I$5
z5T4w<^WEA&T&2i9ec37F?WNUTsX^CXPulR(Yh!EWL$4KH!Z1u{aARPaenuTr{l=$;
z6N9^Q!^9hmn>{)T+w8|0d8)6p9PJ3W0y%kD5V*`3pZBeCIVTsH85uGMS2E**S7B#a
zDO-rjp7r?zH#N>EY0s?0FusCqP@Rq}42IQ+t6#~7<g60O%Atf3jc8+5)?N+4Q$SdH
z9=c2;2O~OcHY>*=PGBo1(Pq4bHzDX?MF-1Rc?Q{duGAGDD5a9R+k+_VP0dZY7S|VC
zi_TFLv(HG*ka;35rX+ceM~R+-dm`tT@-An{*Sn+^h8L&2;{l%+dU;J#O8NT{M2v+&
z7KmypoP%MkNZ>@?R@#ft*Hnh5*AOqJCMQP6q)GSuOk7IbRp-sYG=!@Yk%iTfu`xzr
z#nnvE#rX4cL0_Pd>&Za-f^Urtr)MYaOXpQi{jm;+tSrlNkgXIF_wia7pp1b7@D7y0
zA$S9J!EvQ!V-wVMkUV~GyI9>W;$c%G0gf<m0*<uZ7oa7I<BdceiR*PuTAl<r#=wsn
z@ep*>k$RjUNxhz`t%{g-wWy7q8aD~5#rOC&ykh)Dsnkd60f^HNsT2AWsM5damvlrc
z!ds)+OuF&Fnix~;seC?&P%;z~ln~_kj1XDz=WKy^cD>p~5o&V_b1o!=1e;iWe2vHu
z#(hk)^aLz}>-4^MbdUDjs@L(Y-F@Bqe^x>Zm__0lE)?XZ8kfT{t3MeRAuA$*N05c(
z>%GM1h8+y-gPp1o5+%^mN%r9vh}9PZL_>4aA5!&|moT-P{%5Vc^%ZX`x2eR=cs?{a
z=<tms<b=g3q!dpk$LCQY?D8aCUauHfpSHtMm2<+yv?Y;A=KN>1LVy3uPn94We*;Dc
B7FYlP

diff --git a/backend/setups/ai_models/strategies/__pycache__/svr_model.cpython-38.pyc b/backend/setups/ai_models/strategies/__pycache__/svr_model.cpython-38.pyc
index 7a4916a1107c9681913c753d4e66778317100f3a..d5b94a417505231ac2502fed28d58a382d369b80 100644
GIT binary patch
delta 69
zcmdm>v`UFLl$V!_0SGF2HYPsZ$lJ{<BLm_xvN3Wn@-Qhdaxro-i7*K;u`qEliZE{8
V#q7Y%sJ{6(Umqi*_~b5uWdN;s4C4R*

delta 75
zcmZ3bv_XkCl$V!_0SJs|u21~Ak++*!NeRSdWMkxD<Y7`^<YMGv5@8ZxVqpTZ7?~KD
a{<AS{-oos_&8V^YD_<WYqr~JEfn@*zpbj|z

diff --git a/backend/setups/ai_models/strategies/svr_model.py b/backend/setups/ai_models/strategies/svr_model.py
index 68fe478..23cd2e0 100644
--- a/backend/setups/ai_models/strategies/svr_model.py
+++ b/backend/setups/ai_models/strategies/svr_model.py
@@ -29,8 +29,7 @@ def model_run(self, name, X_train, X_test, y_train, y_test, X_deploy, y_deploy):
             return True
         else:
             y_pred = model.predict(X_test)
-            print(model.model_summary(y_pred, y_test,
-                  None, None))
+            print(model.model_summary(y_pred, y_test,None, None))
             model.save_model(f'{name}/SVR')
 
             return True
diff --git a/backend/setups/tests/test_urls.py b/backend/setups/tests/test_urls.py
index 8ffca49..eb4f3d3 100644
--- a/backend/setups/tests/test_urls.py
+++ b/backend/setups/tests/test_urls.py
@@ -1,5 +1,4 @@
-from django.http import response
-from django.test import TestCase, Client
+from django.test import TestCase
 from django.urls import reverse, resolve
 
 from setups.views import HomeView, SVRModelView
@@ -11,14 +10,14 @@ def setUp(self):
     self.svr_model_fit_url = reverse('SVR Model fit')
     self.svr_model_predict_url = reverse('SVR Model predict')
 
-  def test_root_URL(self):
+  def test_root_url_GET(self):
     self.assertEquals(resolve(self.root_url).func.view_class, HomeView)
 
-  def test_svr_model_URL(self):
+  def test_svr_model_url_GET(self):
     self.assertEquals(resolve(self.svr_model_url).func.view_class, SVRModelView)
 
-  def test_svr_model_fit_URL(self):
+  def test_svr_model_fit_url_GET(self):
     self.assertEquals(resolve(self.svr_model_fit_url).func, SVRModelView.fit)
 
-  def test_svr_model_predict_URL(self):
+  def test_svr_model_predict_url_GET(self):
     self.assertEquals(resolve(self.svr_model_predict_url).func, SVRModelView.predict)
diff --git a/backend/setups/tests/test_views.py b/backend/setups/tests/test_views.py
new file mode 100644
index 0000000..231ae52
--- /dev/null
+++ b/backend/setups/tests/test_views.py
@@ -0,0 +1,41 @@
+from django.http import response
+import requests
+from django.test import TestCase, Client
+from django.urls import reverse
+from django.contrib import auth
+import json
+
+class TestSetupsViews(TestCase):
+  def setUp(self):
+    self.root_url = reverse('Setups homepage')
+    self.svr_model_url = reverse('SVR Model GET/POST')
+    self.svr_model_fit_url = reverse('SVR Model fit')
+    self.svr_model_predict_url = reverse('SVR Model predict')
+
+  def test_root_url_GET(self):
+    response = self.client.get(self.root_url)
+    self.assertEquals(response.status_code, 200)
+
+  def test_svr_model_url_GET(self):
+    response = requests.get(
+      'https://btk-ai-app.herokuapp.com/setups/svr_model/',
+      json = {
+        'name': "PETR4"
+      }
+    )
+    self.assertEquals(response.status_code, 200)
+
+  def test_svr_model_fit_url_GET(self):
+    response = requests.get(
+      'http://localhost:8000'+self.svr_model_fit_url,
+      json = {
+        'name': "PETR4",
+        'start_date': '2020-01-01',
+        'end_date': '2020-12-31',
+        'train_size': 0.6,
+        'test_size': 0.3,
+        'deploy_size': 0.1
+      }
+    )
+    
+    self.assertEquals(response.status_code, 200)
\ No newline at end of file
diff --git a/backend/setups/views.py b/backend/setups/views.py
index 7e6902b..9133952 100644
--- a/backend/setups/views.py
+++ b/backend/setups/views.py
@@ -60,7 +60,6 @@ def fit(request):
 
     @api_view(['GET'])
     def predict(request):
-
         if request.method == 'GET':
             body_data = json.dumps(request.data)
             body = json.loads(body_data)
@@ -76,8 +75,9 @@ def predict(request):
             return JsonResponse({'prediction': y_pred[0]})
 
     def get(self, request):
-        body = json.loads(request.body)
-        asset_name = body['name']
+        body_data = json.dumps(request.data)
+        body_data = json.loads(body_data)
+        asset_name = body_data['name']
 
         model_summary = joblib.load(
             f'static/saved_models/{asset_name}/SVR_summary.pkl')
diff --git a/backend/tools/tests/test_urls.py b/backend/tools/tests/test_urls.py
index 5ff927a..3e80e6f 100644
--- a/backend/tools/tests/test_urls.py
+++ b/backend/tools/tests/test_urls.py
@@ -1,5 +1,4 @@
-from django.http import response
-from django.test import TestCase, Client
+from django.test import TestCase
 from django.urls import reverse, resolve
 
 from tools.views import HomeView, MarkowitzView
diff --git a/backend/tools/tests/test_views.py b/backend/tools/tests/test_views.py
new file mode 100644
index 0000000..16bd761
--- /dev/null
+++ b/backend/tools/tests/test_views.py
@@ -0,0 +1,22 @@
+from django.test import TestCase, Client
+from django.urls import reverse
+
+class TestToolsViews(TestCase):
+  def setUp(self) -> None:
+      self.client = Client()
+      self.root_url = reverse('Tools homepage')
+      self.markowivtz_url = reverse('Tools Markowitz')
+        
+  def test_root_url_GET(self):
+    response = self.client.get(
+      self.root_url
+    )
+
+    self.assertEquals(response.status_code, 200)
+  
+  def test_markowitz_url_GET(self):
+    response = self.client.get(
+      self.markowivtz_url
+    )
+
+    self.assertEquals(response.status_code, 200)
\ No newline at end of file