class PositionsExtractorTest(unittest.TestCase): def setUp(self): self.__mapper = PositionsExtractor() def testExtractPosition(self): content = [{"position": "Ruck", "points": 100}] self.assertEquals(self.__mapper.extract(content), [{"position": "Ruck", "points": 100}]) def testExtractMultiplePositions(self): content = [{"position": "Ruck", "points": 1}, {"position": "Midfield", "points": 2}] self.assertEquals(self.__mapper.extract(content), [{"position": "Ruck", "points": 1}, {"position": "Midfield", "points": 2}]) def testExtractEmptyPositions(self): self.assertEqual(self.__mapper.extract([]), []) def testExtractNonePositions(self): self.assertEqual(self.__mapper.extract(None), [])
def getPositions(self, playerId): positionsExtractor = PositionsExtractor() positions = self.__dbHandler.execute('''select p.position, p.points from position p where id=?''', [playerId]) return positionsExtractor.extract(positions)