bench_new/bench/tests/test_utils.py

import os
import shutil
import subprocess
import unittest

from bench.app import App
from bench.bench import Bench
from bench.exceptions import InvalidRemoteException
from bench.utils import is_valid_xhiveframework_branch


class TestUtils(unittest.TestCase):
	def test_app_utils(self):
		git_url = "https://lab.membtech.com/xhiveframework/xhiveframework15"
		branch = "develop"
		app = App(name=git_url, branch=branch, bench=Bench("."))
		self.assertTrue(
			all(
				[
					app.name == git_url,
					app.branch == branch,
					app.tag == branch,
					app.is_url is True,
					app.on_disk is False,
					app.org == "xhiveframework",
					app.url == git_url,
				]
			)
		)

	def test_is_valid_xhiveframework_branch(self):
		with self.assertRaises(InvalidRemoteException):
			is_valid_xhiveframework_branch(
				"https://lab.membtech.com/xhiveframework/xhiveframework15.git", xhiveframework_branch="random-branch"
			)
			is_valid_xhiveframework_branch(
				"https://github.com/random/random.git", xhiveframework_branch="random-branch"
			)

		is_valid_xhiveframework_branch(
			"https://lab.membtech.com/xhiveframework/xhiveframework15.git", xhiveframework_branch="develop"
		)
		is_valid_xhiveframework_branch(
			"https://lab.membtech.com/xhiveframework/xhiveframework15.git", xhiveframework_branch="v13.29.0"
		)

	def test_app_states(self):
		bench_dir = "./sandbox"
		sites_dir = os.path.join(bench_dir, "sites")

		if not os.path.exists(sites_dir):
			os.makedirs(sites_dir)

		fake_bench = Bench(bench_dir)

		self.assertTrue(hasattr(fake_bench.apps, "states"))

		fake_bench.apps.states = {
			"xhiveframework": {
				"resolution": {"branch": "develop", "commit_hash": "234rwefd"},
				"version": "14.0.0-dev",
			}
		}
		fake_bench.apps.update_apps_states()

		self.assertEqual(fake_bench.apps.states, {})

		xhiveframework_path = os.path.join(bench_dir, "apps", "xhiveframework")

		os.makedirs(os.path.join(xhiveframework_path, "xhiveframework"))

		subprocess.run(["git", "init"], cwd=xhiveframework_path, capture_output=True, check=True)

		with open(os.path.join(xhiveframework_path, "xhiveframework", "__init__.py"), "w+") as f:
			f.write("__version__ = '11.0'")

		subprocess.run(["git", "add", "."], cwd=xhiveframework_path, capture_output=True, check=True)
		subprocess.run(
			["git", "config", "user.email", "bench-test_app_states@gha.com"],
			cwd=xhiveframework_path,
			capture_output=True,
			check=True,
		)
		subprocess.run(
			["git", "config", "user.name", "App States Test"],
			cwd=xhiveframework_path,
			capture_output=True,
			check=True,
		)
		subprocess.run(
			["git", "commit", "-m", "temp"], cwd=xhiveframework_path, capture_output=True, check=True
		)

		fake_bench.apps.update_apps_states(app_name="xhiveframework")

		self.assertIn("xhiveframework", fake_bench.apps.states)
		self.assertIn("version", fake_bench.apps.states["xhiveframework"])
		self.assertEqual("11.0", fake_bench.apps.states["xhiveframework"]["version"])

		shutil.rmtree(bench_dir)

	def test_ssh_ports(self):
		app = App("git@github.com:22:xhiveframework/xhiveframework15")
		self.assertEqual(
			(app.use_ssh, app.org, app.repo, app.app_name), (True, "xhiveframework", "xhiveframework", "xhiveframework")
		)