test e alterações feitas no wscserver

Macieski
1 parent 558c4cac
Showing 3 changed files with 110 additions and 2 deletions Show diff stats
wscserver/tests/__init__.py
wscserver/tests/test_conect.py
wscserver/view/restfulview.py
@@ -0,0 +1 @@
+#
@@ -0,0 +1,89 @@
+#!/usr/env python
+# -*- coding: utf-8 -*
+import unittest
+import requests
+import json
+#import hashlib
+#from pyramid_restler.view import RESTfulView
+#from collections import defaultdict
+
+class TestConect(unittest.TestCase):
+    """
+        Testando WSCserver 
+    """
+    def setUp(self):
+        self.rest_url = 'http://10.1.0.121/wscserver'
+        pass
+
+
+    def test_carrega_classe(self):
+        """
+        Testa carregar instanciar classe do REST
+        """
+        url = self.rest_url + '/rest/coleta'
+        r = requests.get(url)
+        coleta = r.json()
+        coleta_str = r.text
+        fd = open('/tmp/coleta.json', 'w+')
+        fd.write(coleta_str)
+        fd.close()
+
+        assert(coleta)
+
+    def test_carrega_resultados(self):
+        """
+        Carrega resultados em uma variavel
+        """
+        url = self.rest_url + '/rest/coleta'
+        r = requests.get(url)
+        coleta = r.json()
+        results = coleta.get('results')
+
+        assert(results)
+
+
+    def test_carrega_computador(self):
+        """
+        Carrega resultados para um computador
+        """
+        url = self.rest_url + '/rest/coleta'
+        r = requests.get(url)
+        coleta = r.json()
+        results = coleta.get('results')
+        computador = results[0]
+
+        assert(computador)
+
+
+    def test_carrega_elemento(self):
+        """
+        Carrega um elemento de um computador
+        """
+        url = self.rest_url + '/rest/coleta'
+        r = requests.get(url)
+        coleta = r.json()
+        results = coleta.get('results')
+        computador = results[0]
+        software = computador.get('SoftwareList')
+
+        assert(software)
+
+    def test_url_rest(self):
+        """
+        Testa URL do rest
+        """
+        url = self.rest_url +'/rest/coleta'
+        get_url = requests.get(url)
+        assert(get_url)
+
+    def test_url_zip(self):
+        """
+        Testa URL do zip
+        """
+        url = self.rest_url +'/zip/coleta'
+        get_url = requests.get(url)
+        assert(get_url)
+
+    def tearDown(self):
+       pass
+
@@ -17,6 +17,7 @@ class CustomRESTfulView(RESTfulView):
         saida = {}
         results = []
+
         for computador in dados_banco:
             # Cria as chaves dos computadores
             saida[computador['id_computador']] = {}
@@ -27,8 +28,9 @@ class CustomRESTfulView(RESTfulView):
         for x in dados_banco:
             # cria as chaves de propriedades nas classes
+            class_name = x['nm_class_name']
             saida[x['id_computador']][x['nm_class_name']][
-                    x['nm_property_name']] = x['te_class_property_value']
+                 class_name +'_'+  x['nm_property_name']] = x['te_class_property_value']
         # Ordena os objetos json
         for computador in saida:
@@ -49,12 +51,28 @@ class CustomRESTfulView(RESTfulView):
             results.append(classes)
-
         valores = {
             'results': results,
             'result_count': len(results)
         }
+        results = valores['results']
+        for computador in results:
+
+            SoftwareList = computador['SoftwareList']
+            new_SoftwareList = [ ]
+
+            for soft_key, soft_value in SoftwareList.items():
+
+                new_SoftwareList_element = {
+                    'name': soft_key,
+                    'value': soft_value
+                }
+                new_SoftwareList.append(new_SoftwareList_element)
+
+            computador['SoftwareList'] = new_SoftwareList
+
+
         response_data = dict(
             body=json.dumps(valores),
             content_type='application/json'
@@ -0,0 +1,89 @@		@@ -0,0 +1,89 @@
	1	+#!/usr/env python
	2	+# -- coding: utf-8 -
	3	+import unittest
	4	+import requests
	5	+import json
	6	+#import hashlib
	7	+#from pyramid_restler.view import RESTfulView
	8	+#from collections import defaultdict
	9	+
	10	+class TestConect(unittest.TestCase):
	11	+ """
	12	+ Testando WSCserver
	13	+ """
	14	+ def setUp(self):
	15	+ self.rest_url = 'http://10.1.0.121/wscserver'
	16	+ pass
	17	+
	18	+
	19	+ def test_carrega_classe(self):
	20	+ """
	21	+ Testa carregar instanciar classe do REST
	22	+ """
	23	+ url = self.rest_url + '/rest/coleta'
	24	+ r = requests.get(url)
	25	+ coleta = r.json()
	26	+ coleta_str = r.text
	27	+ fd = open('/tmp/coleta.json', 'w+')
	28	+ fd.write(coleta_str)
	29	+ fd.close()
	30	+
	31	+ assert(coleta)
	32	+
	33	+ def test_carrega_resultados(self):
	34	+ """
	35	+ Carrega resultados em uma variavel
	36	+ """
	37	+ url = self.rest_url + '/rest/coleta'
	38	+ r = requests.get(url)
	39	+ coleta = r.json()
	40	+ results = coleta.get('results')
	41	+
	42	+ assert(results)
	43	+
	44	+
	45	+ def test_carrega_computador(self):
	46	+ """
	47	+ Carrega resultados para um computador
	48	+ """
	49	+ url = self.rest_url + '/rest/coleta'
	50	+ r = requests.get(url)
	51	+ coleta = r.json()
	52	+ results = coleta.get('results')
	53	+ computador = results[0]
	54	+
	55	+ assert(computador)
	56	+
	57	+
	58	+ def test_carrega_elemento(self):
	59	+ """
	60	+ Carrega um elemento de um computador
	61	+ """
	62	+ url = self.rest_url + '/rest/coleta'
	63	+ r = requests.get(url)
	64	+ coleta = r.json()
	65	+ results = coleta.get('results')
	66	+ computador = results[0]
	67	+ software = computador.get('SoftwareList')
	68	+
	69	+ assert(software)
	70	+
	71	+ def test_url_rest(self):
	72	+ """
	73	+ Testa URL do rest
	74	+ """
	75	+ url = self.rest_url +'/rest/coleta'
	76	+ get_url = requests.get(url)
	77	+ assert(get_url)
	78	+
	79	+ def test_url_zip(self):
	80	+ """
	81	+ Testa URL do zip
	82	+ """
	83	+ url = self.rest_url +'/zip/coleta'
	84	+ get_url = requests.get(url)
	85	+ assert(get_url)
	86	+
	87	+ def tearDown(self):
	88	+ pass
	89	+