Python SnollaWorker.check_allowed_originsの例

コード例 #1

ファイル: test_snolla.py プロジェクト: nblock/snolla

    def test_handle_extracted_action(self, mock_origin):
        mock_origin.return_value = False
        obj = SnollaWorker(self.cfg, None, None)

        # Check False
        self.assertFalse(obj.check_allowed_origins(self.commit))
        mock_origin.assert_called_once_with(self.commit['origin'], self.cfg['general']['allowed_origins'])

        # Check True
        mock_origin.reset_mock()
        mock_origin.return_value = True
        self.assertTrue(obj.check_allowed_origins(self.commit))
        mock_origin.assert_called_once_with(self.commit['origin'], self.cfg['general']['allowed_origins'])