from django.test import SimpleTestCase
import json
import requests

url = "http://127.0.0.1:8000/Tierra_2/alm02/"
headers = {'content-type': 'application/json'}


# Create your tests here.
class Alm02TestCase(SimpleTestCase):
    def test_alm02_importSHPtoPostgis(self):
        print("############# importSHPtoPostgis #############")
        # resultado = views.importSHPtoPostgis(None, 'c:/tierra2/SHP/')
        resultado = requests.post(url + "importshptopostgis/",
                                  data=json.dumps({"source":"c:/tierra2/SHP/"}),
                                  headers=headers)
        print("test_alm02_importSHPtoPostgis:" + resultado.text)
        self.assertTrue(str(resultado.content).find('Tablas creadas correctamente!') >= 0)

    def test_alm02_importGDBtoPostgis(self):
        print("############# importGDBtoPostgis #############")
        # resultado = views.importGDBtoPostgis(None, 'c:/tierra2/GDB/')
        resultado = requests.post(url + "importgdbtopostgis/",
                                  data=json.dumps({"source":"c:/tierra2/gdb/Transportation.gdb"}),
                                  headers=headers)
        print("test_alm02_importGDBtoPostgis:" + resultado.text)
        self.assertTrue(str(resultado.content).find('Tablas creadas correctamente!') >= 0)

    def test_alm02_importPostgis(self):
        print("############# importPostgis #############")
        resultado = requests.post(url + "importpostgis/",
                                  data=json.dumps({"layers":"parcela1,parcela2","conn":"PG: host=10.10.87.155 dbname=pruebas user=postgres password=postgres port='5432'"}),
                                  headers=headers)
        print("test_alm02_importPostgis:" + resultado.text)
        self.assertTrue(str(resultado.content).find('Tablas creadas correctamente!') >= 0)

    def test_alm02_importPostgistoSHP(self):
        print("############# importPostgistoSHP #############")
        resultado = requests.post(url + "importpostgistoshp/",
                                  data=json.dumps({"layers":"parcela1,parcela2","conn":"PG: host=10.10.87.155 dbname=pruebas user=postgres password=postgres port='5432'","shp":"c:/tierra2/SHP/"}),
                                  headers=headers)
        print("test_alm02_importPostgistoSHP:" + resultado.text)
        self.assertTrue(str(resultado.content).find('Ficheros creados correctamente!') >= 0)