aboutsummaryrefslogtreecommitdiffhomepage
path: root/infra/build
diff options
context:
space:
mode:
authorGravatar Oliver Chang <oliverchang@users.noreply.github.com>2021-01-22 02:43:42 +1100
committerGravatar GitHub <noreply@github.com>2021-01-21 07:43:42 -0800
commit5b51cfb00f89d127a561aa26d5eb122c3bce063f (patch)
treead04dde6ccfb221b9fc9884e1e6ca90eb73130e6 /infra/build
parent03de1c66f0dcd4cf4a82a4b8652c844dcc3c9bba (diff)
Fix build functions tests when invoked from unittest.TestSuite. (#5027)
Diffstat (limited to 'infra/build')
-rw-r--r--infra/build/functions/project_sync_test.py5
-rw-r--r--infra/build/functions/request_build_test.py4
-rw-r--r--infra/build/functions/request_coverage_build_test.py4
-rw-r--r--infra/build/functions/update_build_status_test.py5
4 files changed, 18 insertions, 0 deletions
diff --git a/infra/build/functions/project_sync_test.py b/infra/build/functions/project_sync_test.py
index 747a2761..7992ba9e 100644
--- a/infra/build/functions/project_sync_test.py
+++ b/infra/build/functions/project_sync_test.py
@@ -16,10 +16,15 @@
"""Unit tests for Cloud Function sync, which syncs the list of github projects
and uploads them to the Cloud Datastore."""
+import os
+import sys
import unittest
from google.cloud import ndb
+sys.path.append(os.path.dirname(__file__))
+# pylint: disable=wrong-import-position
+
from datastore_entities import Project
from project_sync import get_github_creds
from project_sync import get_projects
diff --git a/infra/build/functions/request_build_test.py b/infra/build/functions/request_build_test.py
index d7346721..171fa404 100644
--- a/infra/build/functions/request_build_test.py
+++ b/infra/build/functions/request_build_test.py
@@ -17,11 +17,15 @@
import json
import datetime
import os
+import sys
import unittest
from unittest import mock
from google.cloud import ndb
+sys.path.append(os.path.dirname(__file__))
+# pylint: disable=wrong-import-position
+
from datastore_entities import BuildsHistory
from datastore_entities import Project
from request_build import get_build_steps
diff --git a/infra/build/functions/request_coverage_build_test.py b/infra/build/functions/request_coverage_build_test.py
index 0264e80e..8a1cf443 100644
--- a/infra/build/functions/request_coverage_build_test.py
+++ b/infra/build/functions/request_coverage_build_test.py
@@ -17,11 +17,15 @@
import json
import datetime
import os
+import sys
import unittest
from unittest import mock
from google.cloud import ndb
+sys.path.append(os.path.dirname(__file__))
+# pylint: disable=wrong-import-position
+
from datastore_entities import Project
from build_and_run_coverage import get_build_steps
import test_utils
diff --git a/infra/build/functions/update_build_status_test.py b/infra/build/functions/update_build_status_test.py
index b877474f..db15b2a6 100644
--- a/infra/build/functions/update_build_status_test.py
+++ b/infra/build/functions/update_build_status_test.py
@@ -14,12 +14,17 @@
#
################################################################################
"""Unit tests for Cloud Function update builds status."""
+import os
+import sys
import unittest
from unittest import mock
from unittest.mock import MagicMock
from google.cloud import ndb
+sys.path.append(os.path.dirname(__file__))
+# pylint: disable=wrong-import-position
+
from datastore_entities import BuildsHistory
from datastore_entities import LastSuccessfulBuild
import test_utils