1*9c5db199SXin Li#!/usr/bin/python3 2*9c5db199SXin Li#pylint: disable-msg=C0111 3*9c5db199SXin Liimport unittest 4*9c5db199SXin Liimport common 5*9c5db199SXin Li#pylint: disable-msg=W0611 6*9c5db199SXin Lifrom autotest_lib.frontend import setup_django_lite_environment 7*9c5db199SXin Lifrom autotest_lib.frontend.afe import direct_afe 8*9c5db199SXin Li 9*9c5db199SXin Li 10*9c5db199SXin Liclass DirectAFETest(unittest.TestCase): 11*9c5db199SXin Li def testEntryCreation(self): 12*9c5db199SXin Li afe = direct_afe.directAFE() 13*9c5db199SXin Li 14*9c5db199SXin Li jobs = afe.get_jobs() 15*9c5db199SXin Li self.assertEquals(len(jobs), 0) 16*9c5db199SXin Li 17*9c5db199SXin Li hosts = afe.get_hosts() 18*9c5db199SXin Li self.assertEquals(len(hosts), 0) 19*9c5db199SXin Li 20*9c5db199SXin Li afe.create_host('a_host') 21*9c5db199SXin Li hosts = afe.get_hosts() 22*9c5db199SXin Li self.assertEquals(len(hosts), 1) 23*9c5db199SXin Li 24*9c5db199SXin Li afe.create_job('job_name', hosts=['a_host']) 25*9c5db199SXin Li jobs = afe.get_jobs() 26*9c5db199SXin Li self.assertEquals(len(jobs), 1) 27*9c5db199SXin Li 28*9c5db199SXin Liif __name__ == '__main__': 29*9c5db199SXin Li unittest.main() 30