def test_append_range(self): '''Ensure set_criteria append range query constructed properly ''' criteria = {"arch": ["i86pc"], "ipv4": ["10.0.10.10", "10.0.10.300"]} criteria.setdefault("mem") criteria.setdefault("mac") set_criteria.set_criteria(criteria, "myxml", self.files.database, 'manifests', append=True) expect_query = "UPDATE manifests SET arch='i86pc',MINipv4=" + \ "'10.0.10.10',MAXipv4='10.0.10.300' WHERE name='myxml'" self.assertEquals(expect_query, self.mockquery.query)
def test_append_unbounded_min(self): '''Ensure set_criteria append min query constructed properly ''' criteria = {"arch": ["i86pc"], "mem": ["unbounded", 4096]} criteria.setdefault("ipv4") criteria.setdefault("mac") set_criteria.set_criteria(criteria, "myxml", self.files.database, AIdb.PROFILES_TABLE, append=True) expect_query = "UPDATE " + AIdb.PROFILES_TABLE + " SET arch='i86pc',"\ "MINmem=NULL,MAXmem='4096' WHERE name='myxml'" self.assertEquals(expect_query, self.mockquery.query)
def test_append_unbounded_max(self): '''Ensure set_criteria append max query constructed properly ''' criteria = {"arch": ["i86pc"], "mem": [2048, "unbounded"]} criteria.setdefault("ipv4") criteria.setdefault("mac") set_criteria.set_criteria(criteria, "myxml", self.files.database, 'manifests', append=True) expect_query = "UPDATE manifests SET arch='i86pc',MINmem='2048'," \ "MAXmem=NULL WHERE name='myxml'" self.assertEquals(expect_query, self.mockquery.query)
def test_unbounded_max(self): '''Ensure set_criteria max query constructed properly ''' criteria = {"arch": ["i86pc"], "mem": [1024, "unbounded"]} criteria.setdefault("ipv4") criteria.setdefault("mac") set_criteria.set_criteria(criteria, "myxml", self.files.database, 'manifests') expect_query = "UPDATE manifests SET arch='i86pc',MINmem='1024'," + \ "MAXmem=NULL,MINipv4=NULL,MAXipv4=NULL,MINmac=NULL," + \ "MAXmac=NULL WHERE name='myxml'" self.assertEquals(expect_query, self.mockquery.query)
def test_range(self): '''Ensure set_criteria max query constructed properly ''' criteria = {"arch": ["i86pc"], "ipv4": ["10.0.30.100", "10.0.50.400"]} criteria.setdefault("mac") criteria.setdefault("mem") set_criteria.set_criteria(criteria, "myxml", self.files.database, 'manifests') expect_query = "UPDATE manifests SET arch='i86pc',MINmem=NULL," + \ "MAXmem=NULL,MINipv4='10.0.30.100'," + \ "MAXipv4='10.0.50.400',MINmac=NULL,MAXmac=NULL " + \ "WHERE name='myxml'" self.assertEquals(expect_query, self.mockquery.query)