import unittest

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, list))
      self.assertTrue(len(patients) > 0)