test_node.py
View Options

	#!/usr/bin/env python3

	from importlib.machinery import SourceFileLoader
	from unittest_data_provider import data_provider
	import unittest


	salt_test_case = SourceFileLoader("salt_test_case", "salt_test_case.py").load_module()
	node = SourceFileLoader("node", "../_modules/node.py").load_module()


	class Testinstance(unittest.TestCase, salt_test_case.SaltTestCase):
	def setUp(self):
	self.initialize_mocks()
	self.instance = node

	self.mock_pillar("data/forests.yaml")

	self.mock_grains()
	self.grains["id"] = "egladil"

	def test_get_wwwroot(self):
	self.assertEqual("wwwroot/lothlorien.forest/egladil", node.get_wwwroot())
	self.assertEqual("wwwroot/entwash.node/www", node.get_wwwroot("entwash"))

	def test_has_web_content(self):
	self.assertTrue(node.has_web_content(".ll/carasgaladhon"))
	self.assertFalse(node.has_web_content(".arda/onodlo"))

	self.assertTrue(node.has_web_content(".arda/onodlo", "entwash"))

	self.assertFalse(node.has_web_content("notexisting"))

	def test_filter_by_role(self):
	node_key = self.grains["id"]

	self.assertEqual(["Caras Galadhon"], node.filter_by_role("items_by_role"))

	self.assertEqual(["Onodlo"], node.filter_by_role("items_by_role", "entwash"))

	# No role
	self.pillar["nodes"][node_key]["roles"] = []
	self.assertEqual([], node.filter_by_role("items_by_role"))

	# More than one role
	self.pillar["nodes"][node_key]["roles"] = ["border", "treecity"]
	self.assertEqual(
	["Caras Galadhon", "Onodlo"], sorted(node.filter_by_role("items_by_role"))
	)

	def test_filter_by_role_with_star(self):
	node_key = self.grains["id"]

	self.assertEqual(
	["Air", "Caras Galadhon"], node.filter_by_role("items_by_role_with_star")
	)

	self.assertEqual(
	["Air", "Onodlo"], node.filter_by_role("items_by_role_with_star", "entwash")
	)

	# No role
	self.pillar["nodes"][node_key]["roles"] = []
	self.assertEqual(["Air"], node.filter_by_role("items_by_role_with_star"))

	# More than one role
	self.pillar["nodes"][node_key]["roles"] = ["border", "treecity"]
	self.assertEqual(
	["Air", "Caras Galadhon", "Onodlo"],
	sorted(node.filter_by_role("items_by_role_with_star")),
	)

	def test_filter_by_name(self):
	self.assertEqual(["Caras Galadhon"], node.filter_by_name("items_by_name"))

	self.assertEqual(
	["Caras Galadhon"], node.filter_by_name("items_by_name", "egladil")
	)

	self.grains["id"] = "entwash"
	self.assertEqual([], node.filter_by_name("items_by_name"))

	def test_filter_by_name_with_star(self):
	self.assertEqual(
	["Air", "Caras Galadhon"], node.filter_by_name("items_by_name_with_star")
	)

	self.assertEqual(
	["Air", "Caras Galadhon"],
	node.filter_by_name("items_by_name_with_star", "egladil"),
	)

	self.grains["id"] = "entwash"
	self.assertEqual(["Air"], node.filter_by_name("items_by_name_with_star"))

	def test_get_ipv6_list(self):
	self.grains["ipv6"] = [
	"::1",
	"2001:470:1f13:ce7:ca5:cade:fab:1e",
	"2001:470:1f12:ce7::2",
	]

	self.assertEqual(
	"[::1] [2001:470:1f13:ce7:ca5:cade:fab:1e] [2001:470:1f12:ce7::2]",
	node.get_ipv6_list(),
	)

	resolved_networks = lambda: (
	(
	"egladil",
	{
	"ipv4_address": "1.2.3.4",
	"ipv4_gateway": "1.2.3.254",
	},
	),
	(
	"entwash",
	{
	"ipv4_address": "10.100.0.5",
	"ipv4_gateway": "10.100.0.1",
	},
	),
	)

	@data_provider(resolved_networks)
	def test_resolve_network(self, id, expected):
	self.grains["id"] = id
	self.assertEqual(expected, node.resolve_network())

	def test_resolve_network_without_gateway(self):
	expected = {
	"ipv4_address": "10.100.0.5",
	"ipv4_gateway": "",
	}

	self.grains["id"] = "entwash"
	del self.pillar["nodes"]["entwash"]["network"]["interfaces"]["net02"]["ipv4"][
	"gateway"
	]

	self.assertEqual(expected, node.resolve_network())

	def test_resolve_network_without_any_network(self):
	expected = {
	"ipv4_address": "",
	"ipv4_gateway": "",
	}

	self.grains["id"] = "entwash"
	del self.pillar["nodes"]["entwash"]["network"]

	self.assertEqual(expected, node.resolve_network())


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

File Metadata

Mime Type: text/x-python
Expires: Tue, May 13, 16:48 (21 h, 31 m)
Storage Engine: blob
Storage Format: Raw Data
Storage Handle: 2568742
Default Alt Text: test_node.py (4 KB)

test_node.py
No OneTemporary
Actions

test_node.py
View Options

File Metadata

Event Timeline

test_node.pyNo OneTemporaryActions

test_node.pyView Options

File Metadata

Event Timeline

test_node.py
No OneTemporary
Actions

test_node.py
View Options