diff --git a/src/qgcunittest/QGCUASFileManagerTest.cc b/src/qgcunittest/QGCUASFileManagerTest.cc index 0d55b3c1a367c8d8d45e29bb6ea54cb64c591164..84134650a0e9c5febee4e0ef1d5966a2a135bafc 100644 --- a/src/qgcunittest/QGCUASFileManagerTest.cc +++ b/src/qgcunittest/QGCUASFileManagerTest.cc @@ -42,8 +42,16 @@ QGCUASFileManagerUnitTest::QGCUASFileManagerUnitTest(void) : // Called once before all test cases are run void QGCUASFileManagerUnitTest::initTestCase(void) { - _mockUAS.setMockSystemId(_systemIdServer); - _mockUAS.setMockMavlinkPlugin(&_mockFileServer); + _mockUAS = new MockUAS(); + Q_CHECK_PTR(_mockUAS); + + _mockUAS->setMockSystemId(_systemIdServer); + _mockUAS->setMockMavlinkPlugin(&_mockFileServer); +} + +void QGCUASFileManagerUnitTest::cleanupTestCase(void) +{ + delete _mockUAS; } // Called before every test case @@ -53,7 +61,7 @@ void QGCUASFileManagerUnitTest::init(void) Q_ASSERT(_multiSpy == NULL); - _fileManager = new QGCUASFileManager(NULL, &_mockUAS, _systemIdQGC); + _fileManager = new QGCUASFileManager(NULL, _mockUAS, _systemIdQGC); Q_CHECK_PTR(_fileManager); // Reset any internal state back to normal diff --git a/src/qgcunittest/QGCUASFileManagerTest.h b/src/qgcunittest/QGCUASFileManagerTest.h index cf29decd4226e1ef6888503557687f419a04f474..13b3a0d939e2b12d4b8a2592303b8ad3029a57c9 100644 --- a/src/qgcunittest/QGCUASFileManagerTest.h +++ b/src/qgcunittest/QGCUASFileManagerTest.h @@ -48,6 +48,7 @@ public: private slots: // Test case initialization void initTestCase(void); + void cleanupTestCase(void); void init(void); void cleanup(void); @@ -84,7 +85,7 @@ private: static const uint8_t _systemIdQGC = 255; static const uint8_t _systemIdServer = 128; - MockUAS _mockUAS; + MockUAS* _mockUAS; MockMavlinkFileServer _mockFileServer; QGCUASFileManager* _fileManager;