<reponame>sanderv32/cosmic
import merge
import unittest
from cs.CsAddress import CsAddress


class TestCsAddress(unittest.TestCase):
    def setUp(self):
        merge.DataBag.DPATH = "."
        self.csaddress = CsAddress("ips", { })

    def test_needs_vrrp(self):
        self.assertTrue(self.csaddress.needs_vrrp({ "nw_type": "guest" }))

    def test_get_guest_if(self):
        self.assertTrue(self.csaddress.get_guest_if() is None)

    def test_get_guest_ip(self):
        self.assertTrue(self.csaddress.get_guest_ip() is None)

    def test_get_guest_netmask(self):
        self.assertTrue(self.csaddress.get_guest_netmask() == "255.255.255.0")


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