class TestPNXReader(TestCase): def setUp(self): self.oai_reader = OAIReader() self.create_oai = self._load_oai('oai-01.xml') self.delete_oai = self._load_oai('oai-02.xml') def test_create_status(self): self.oai_reader.read(self.create_oai) self.assertEqual('new', self.oai_reader._get_status) def test_delete_status(self): self.oai_reader.read(self.delete_oai) self.assertEqual('deleted', self.oai_reader._get_status) def test_bad_status(self): self.oai_reader.read(self._load_oai('oai-03.xml')) with self.assertRaises(OAIError): self.oai_reader._get_status def test_id(self): self.oai_reader.read(self.create_oai) self.assertEqual('urm_publish-61441201100001021', self.oai_reader._get_id) def test_record(self): self.oai_reader.read(self.create_oai) self.assertEqual(self.oai_reader._marc_record.title(), 'Capital IQ') def _load_oai(self, filename): with open(filename, "r") as myfile: return myfile.read().replace('\n', '')
def setUp(self): self.oai_reader = OAIReader() self.create_oai = self._load_oai('oai-01.xml') self.delete_oai = self._load_oai('oai-02.xml')