class ImporterTests(TestCase): def setUp(self): self.mocker = Mocker() self.mock_repo = self.mocker.mock() expect(self.mock_repo.find_data_dates()).result([datetime.datetime.now()]).count(1) expect(self.mock_repo.file_is_modified(ANY, ANY, ANY)).result(True).count(1) expect(self.mock_repo.build_data_path(ANY, ANY)).result(None).count(1) self.db = [] self.db_mock = self.mocker.proxy(self.db) expect(self.db_mock.openFile(ARGS, KWARGS)).result(True) expect(self.db_mock.close()).result(None) self.mocker.replay() def tearDown(self): self.mocker.restore() def test_import_simple_data(self): self.db.append(FakeRow(DEN_CORTA="Tenerife", CODIGO_INT="1", DEN_LARGA="Isla de Tenerife")) self.db.append(FakeRow(DEN_CORTA="La Gomera", CODIGO_INT="2", DEN_LARGA="Isla de La Gomera")) im = Importer("fakefakefakefake", self.db_mock, self.mock_repo) im._import_simple_data(nombre="Islas") self.assertEqual(Isla.objects.get(idPincel=1).nombreCorto, "Tenerife") self.assertEqual(Isla.objects.get(idPincel=2).nombreCorto, "La Gomera") self.mocker.verify() def test_import_simple_data_iterative(self): """ More consistent and iterative way of testing the importer. Testing the simple ones. """ pass
def setUp(self): self.mocker = Mocker() self.mock_repo = self.mocker.mock() expect(self.mock_repo.find_data_dates()).result([datetime.datetime.now()]).count(1) expect(self.mock_repo.file_is_modified(ANY, ANY, ANY)).result(True).count(1) expect(self.mock_repo.build_data_path(ANY, ANY)).result(None).count(1) self.db = [] self.db_mock = self.mocker.proxy(self.db) expect(self.db_mock.openFile(ARGS, KWARGS)).result(True) expect(self.db_mock.close()).result(None) self.mocker.replay()