import unittest
import pandas as pd

from data_service import DataService

class TestDataService(unittest.TestCase):
  def test_get_patients(self):
    # TODO: I'm just stubbing out what I think the API could look like.
    #data_service = DataService().to("").withCreds("").build()
    with DataService().build() as data_service:
      patients = data_service.get_patients()
      self.assertTrue(isinstance(patients, pd.DataFrame))
      self.assertTrue("PATIENT_ID" in patients.columns.values)
      self.assertTrue("FIRST" in patients.columns.values)
      self.assertTrue(patients.size > 0)

  def test_get_patient_documents(self):
    with DataService().build() as data_service:
      documents = data_service.get_documents()
      self.assertTrue(isinstance(documents, list))
      self.assertTrue(len(documents) > 0)

  def test_get_patient_metadatas(self):
    with DataService().build() as data_service:
      metadatas = data_service.get_document_metadatas()
      self.assertTrue(isinstance(metadatas, list))
      self.assertTrue(len(metadatas) > 0)