tile_unittest.py

# tile-generator
#
# Copyright (c) 2015-Present Pivotal Software, Inc. All Rights Reserved.
#
# 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
from click.testing import CliRunner
import os
import shutil
import tempfile
from . import tile

class TestTileInit(unittest.TestCase):
	# tile.init() changes the working directory. In normal usage,
	# this is fine, but for unit tests this changes the local
	# state, so we should restore it.
	def setUp(self):
		self.cwd = os.getcwd()

	def tearDown(self):
		os.chdir(self.cwd)

	def test_tile_init_works(self):
		tmpdir = tempfile.mkdtemp()
		try:
			tiledir = os.path.join(tmpdir, 'my-tile')
			runner = CliRunner()
			result = runner.invoke(tile.init_cmd, [tiledir])
			self.assertEqual(result.exit_code, 0)
			self.assertTrue(os.path.isfile(os.path.join(tiledir, 'tile.yml')))
		finally:
			shutil.rmtree(tmpdir)

	# Encountered unicode in `pcf apply-changes` output. Stashing
	# the next two tests to document the problem and solution, as
	# we may encounter this pattern elsewhere.
	## This passes (i.e., raises error as expected) in my
	## terminal, but not in CI pipeline.
	# def test_print_unicode_works(self):
	# 	with self.assertRaises(UnicodeEncodeError):
	# 		print(u'foo\xb1bar')

	def test_print_utf8_works(self):
		print('foo\xb1bar'.encode('utf-8'))
		self.assertTrue(True)

if __name__ == '__main__':
	unittest.main()