1*333d2b36SAndroid Build Coastguard Worker# Copyright 2018 Google Inc. All rights reserved. 2*333d2b36SAndroid Build Coastguard Worker# 3*333d2b36SAndroid Build Coastguard Worker# Licensed under the Apache License, Version 2.0 (the "License"); 4*333d2b36SAndroid Build Coastguard Worker# you may not use this file except in compliance with the License. 5*333d2b36SAndroid Build Coastguard Worker# You may obtain a copy of the License at 6*333d2b36SAndroid Build Coastguard Worker# 7*333d2b36SAndroid Build Coastguard Worker# http://www.apache.org/licenses/LICENSE-2.0 8*333d2b36SAndroid Build Coastguard Worker# 9*333d2b36SAndroid Build Coastguard Worker# Unless required by applicable law or agreed to in writing, software 10*333d2b36SAndroid Build Coastguard Worker# distributed under the License is distributed on an "AS IS" BASIS, 11*333d2b36SAndroid Build Coastguard Worker# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12*333d2b36SAndroid Build Coastguard Worker# See the License for the specific language governing permissions and 13*333d2b36SAndroid Build Coastguard Worker# limitations under the License. 14*333d2b36SAndroid Build Coastguard Worker 15*333d2b36SAndroid Build Coastguard Workerimport os 16*333d2b36SAndroid Build Coastguard Workerimport sys 17*333d2b36SAndroid Build Coastguard Worker 18*333d2b36SAndroid Build Coastguard Worker# This file checks the visible python state against expected values when run 19*333d2b36SAndroid Build Coastguard Worker# inside a hermetic par file. 20*333d2b36SAndroid Build Coastguard Worker 21*333d2b36SAndroid Build Coastguard Workerfailed = False 22*333d2b36SAndroid Build Coastguard Workerdef assert_equal(what, a, b): 23*333d2b36SAndroid Build Coastguard Worker global failed 24*333d2b36SAndroid Build Coastguard Worker if a != b: 25*333d2b36SAndroid Build Coastguard Worker print("Expected %s('%s') == '%s'" % (what, a, b)) 26*333d2b36SAndroid Build Coastguard Worker failed = True 27*333d2b36SAndroid Build Coastguard Worker 28*333d2b36SAndroid Build Coastguard Workerarchive = sys.modules["__main__"].__loader__.archive 29*333d2b36SAndroid Build Coastguard Worker 30*333d2b36SAndroid Build Coastguard Workerassert_equal("__name__", __name__, "testpkg.par_test") 31*333d2b36SAndroid Build Coastguard WorkerfileName = __file__ 32*333d2b36SAndroid Build Coastguard Workerif fileName.endswith('.pyc'): 33*333d2b36SAndroid Build Coastguard Worker fileName = fileName[:-1] 34*333d2b36SAndroid Build Coastguard Workerassert_equal("__file__", fileName, os.path.join(archive, "testpkg/par_test.py")) 35*333d2b36SAndroid Build Coastguard Worker 36*333d2b36SAndroid Build Coastguard Workerassert_equal("__package__", __package__, "testpkg") 37*333d2b36SAndroid Build Coastguard Worker 38*333d2b36SAndroid Build Coastguard Workerassert_equal("__loader__.archive", __loader__.archive, archive) 39*333d2b36SAndroid Build Coastguard Workerassert_equal("__loader__.prefix", __loader__.prefix, "testpkg/") 40*333d2b36SAndroid Build Coastguard Worker 41*333d2b36SAndroid Build Coastguard Workerif failed: 42*333d2b36SAndroid Build Coastguard Worker sys.exit(1) 43