1 files changed, 126 insertions, 0 deletions
diff --git a/gae/webapp/src/endpoint/schedule_info_test.py b/gae/webapp/src/endpoint/schedule_info_test.py
new file mode 100644
index 0000000..61e69ae
--- /dev/null
+++ b/gae/webapp/src/endpoint/schedule_info_test.py
@@ -0,0 +1,126 @@
+#!/usr/bin/env python
+#
+# Copyright (C) 2018 The Android Open Source Project
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+import unittest
+
+try:
+    from unittest import mock
+except ImportError:
+    import mock
+
+from webapp.src import vtslab_status as Status
+from webapp.src.endpoint import schedule_info
+from webapp.src.proto import model
+from webapp.src.testing import unittest_base
+
+
+class ScheduleInfoTest(unittest_base.UnitTestBase):
+    """A class to test schedule_info endpoint API."""
+
+    def setUp(self):
+        """Initializes test"""
+        super(ScheduleInfoTest, self).setUp()
+
+    def testSetWithSimpleMessage(self):
+        """Asserts schedule_info/set API receives a simple message."""
+        # As of June 8, 2018, these are uploaded from host controller.
+        container = (
+            schedule_info.SCHEDULE_INFO_RESOURCE.combined_message_class(
+                manifest_branch=self.GetRandomString(),
+                build_storage_type=Status.STORAGE_TYPE_DICT["PAB"],
+                build_target=self.GetRandomString(),
+                require_signed_device_build=False,
+                has_bootloader_img=True,
+                has_radio_img=True,
+                test_name=self.GetRandomString(),
+                period=360,
+                priority="high",
+                device=[self.GetRandomString()],
+                required_host_equipment=[self.GetRandomString()],
+                required_device_equipment=[self.GetRandomString()],
+                device_pab_account_id=self.GetRandomString(),
+                shards=1,
+                param=[self.GetRandomString()],
+                retry_count=1,
+                gsi_storage_type=Status.STORAGE_TYPE_DICT["PAB"],
+                gsi_branch=self.GetRandomString(),
+                gsi_build_target=self.GetRandomString(),
+                gsi_pab_account_id=self.GetRandomString(),
+                gsi_vendor_version=self.GetRandomString(),
+                test_storage_type=Status.STORAGE_TYPE_DICT["PAB"],
+                test_branch=self.GetRandomString(),
+                test_build_target=self.GetRandomString(),
+                test_pab_account_id=self.GetRandomString(),
+                image_package_repo_base=self.GetRandomString(),
+                report_bucket=[self.GetRandomString()],
+                report_spreadsheet_id=[self.GetRandomString()],
+                report_persistent_url=[self.GetRandomString()],
+                report_reference_url=[self.GetRandomString()],
+            ))
+        api = schedule_info.ScheduleInfoApi()
+        response = api.set(container)
+
+        self.assertEqual(response.return_code, model.ReturnCodeMessage.SUCCESS)
+
+    def testSetWithEmptyRepeatedField(self):
+        """Asserts schedule_info/set API receives a message.
+
+        This test sets required_host_equipment to empty and sends to endpoint
+        method.
+        """
+        # As of June 8, 2018, these are uploaded from host controller.
+        container = (
+            schedule_info.SCHEDULE_INFO_RESOURCE.combined_message_class(
+                manifest_branch=self.GetRandomString(),
+                build_storage_type=Status.STORAGE_TYPE_DICT["PAB"],
+                build_target=self.GetRandomString(),
+                require_signed_device_build=False,
+                has_bootloader_img=True,
+                has_radio_img=True,
+                test_name=self.GetRandomString(),
+                period=360,
+                priority="high",
+                device=[self.GetRandomString()],
+                required_host_equipment=[self.GetRandomString()],
+                required_device_equipment=[self.GetRandomString()],
+                device_pab_account_id=self.GetRandomString(),
+                shards=1,
+                param=[self.GetRandomString()],
+                retry_count=1,
+                gsi_storage_type=Status.STORAGE_TYPE_DICT["PAB"],
+                gsi_branch=self.GetRandomString(),
+                gsi_build_target=self.GetRandomString(),
+                gsi_pab_account_id=self.GetRandomString(),
+                gsi_vendor_version=self.GetRandomString(),
+                test_storage_type=Status.STORAGE_TYPE_DICT["PAB"],
+                test_branch=self.GetRandomString(),
+                test_build_target=self.GetRandomString(),
+                test_pab_account_id=self.GetRandomString(),
+                image_package_repo_base=self.GetRandomString(),
+                report_bucket=[],
+                report_spreadsheet_id=[],
+                report_persistent_url=[],
+                report_reference_url=[],
+            ))
+        api = schedule_info.ScheduleInfoApi()
+        response = api.set(container)
+
+        self.assertEqual(response.return_code, model.ReturnCodeMessage.SUCCESS)
+
+
+if __name__ == "__main__":
+    unittest.main()