1 files changed, 30 insertions, 2 deletions
diff --git a/Swiften/FileTransfer/UnitTest/OutgoingJingleFileTransferTest.cpp b/Swiften/FileTransfer/UnitTest/OutgoingJingleFileTransferTest.cpp
index 40e7233..4b2fb50 100644
--- a/Swiften/FileTransfer/UnitTest/OutgoingJingleFileTransferTest.cpp
+++ b/Swiften/FileTransfer/UnitTest/OutgoingJingleFileTransferTest.cpp
@@ -54,21 +54,33 @@ using namespace Swift;
 class OutgoingJingleFileTransferTest : public CppUnit::TestFixture {
                CPPUNIT_TEST_SUITE(OutgoingJingleFileTransferTest);
                CPPUNIT_TEST(test_SendSessionInitiateOnStart);
                CPPUNIT_TEST(test_FallbackToIBBAfterFailingS5B);
                CPPUNIT_TEST(test_ReceiveSessionTerminateAfterSessionInitiate);
+                CPPUNIT_TEST(test_DeclineEmitsFinishedStateCanceled);
                CPPUNIT_TEST_SUITE_END();
                class FTStatusHelper {
                public:
-                        bool finishedCalled;
+                        FTStatusHelper() : finishedCalled(false), error(FileTransferError::UnknownError) {
-                        FileTransferError::Type error;
+                        }
                        void handleFileTransferFinished(boost::optional<FileTransferError> error) {
                                finishedCalled = true;
                                if (error.is_initialized()) this->error = error.get().getType();
                        }
+                        void handleFileTransferStatusChanged(FileTransfer::State fileTransferSTate) {
+                                state = fileTransferSTate;
+                        }
+                public:
+                        bool finishedCalled;
+                        FileTransferError::Type error;
+                        boost::optional<FileTransfer::State> state;
                };
 public:
                boost::shared_ptr<OutgoingJingleFileTransfer> createTestling() {
                        JID to("test@foo.com/bla");
                        JingleFileTransferFileInfo fileInfo;
@@ -191,10 +203,26 @@ public:
                        fakeJingleSession->handleSessionTerminateReceived(JinglePayload::Reason(JinglePayload::Reason::Busy));
                        CPPUNIT_ASSERT_EQUAL(true, helper.finishedCalled);
                        CPPUNIT_ASSERT(FileTransferError::PeerError == helper.error);
                }
                
+                void test_DeclineEmitsFinishedStateCanceled() {
+                        boost::shared_ptr<OutgoingJingleFileTransfer> transfer = createTestling();
+                        transfer->start();
+                        getCall<FakeJingleSession::InitiateCall>(0);
+                        FTStatusHelper helper;
+                        helper.finishedCalled = false;
+                        transfer->onFinished.connect(bind(&FTStatusHelper::handleFileTransferFinished, &helper, _1));
+                        transfer->onStateChanged.connect(bind(&FTStatusHelper::handleFileTransferStatusChanged, &helper, _1));
+                        fakeJingleSession->handleSessionTerminateReceived(JinglePayload::Reason(JinglePayload::Reason::Decline));
+                        CPPUNIT_ASSERT_EQUAL(true, helper.finishedCalled);
+                        CPPUNIT_ASSERT(FileTransferError::UnknownError == helper.error);
+                        CPPUNIT_ASSERT_EQUAL(true, helper.state.is_initialized());
+                        CPPUNIT_ASSERT(FileTransfer::State::Canceled == helper.state.get().type);
+                }
 //TODO: some more testcases
 private:
        void addFileTransferDescription() {