xref: /aosp_15_r20/external/autotest/frontend/afe/direct_afe_unittest.py (revision 9c5db1993ded3edbeafc8092d69fe5de2ee02df7)
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