test/com/isode/stroke/parser/payloadparsers/PayloadsParserTester.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64

/*
 * Copyright (c) 2012 Isode Limited, London, England.
 * All rights reserved.
 */
/*
 * Copyright (c) 2010 Remko Tronçon
 * All rights reserved.
 */

package com.isode.stroke.parser.payloadparsers;

import com.isode.stroke.elements.Payload;
import com.isode.stroke.eventloop.EventLoop;
import com.isode.stroke.parser.AttributeMap;
import com.isode.stroke.parser.PayloadParser;
import com.isode.stroke.parser.PayloadParserFactory;
import com.isode.stroke.parser.PlatformXMLParserFactory;
import com.isode.stroke.parser.XMLParser;
import com.isode.stroke.parser.XMLParserClient;

public class PayloadsParserTester implements XMLParserClient {
    private XMLParser xmlParser;
    private FullPayloadParserFactoryCollection factories = new FullPayloadParserFactoryCollection();
    private PayloadParser payloadParser;
    private int level;

    public PayloadsParserTester(EventLoop eventLoop) {
        level = 0;
        xmlParser = PlatformXMLParserFactory.createXMLParser(this);
    }

    public boolean parse(String data) {
        return xmlParser.parse(data);
    }

    public void handleStartElement(String element, String ns,
            AttributeMap attributes) {
        if (level == 0) {
            assert (payloadParser == null);
            PayloadParserFactory payloadParserFactory = factories
                    .getPayloadParserFactory(element, ns, attributes);
            assert (payloadParserFactory != null);
            payloadParser = payloadParserFactory.createPayloadParser();
        }
        payloadParser.handleStartElement(element, ns, attributes);
        level++;
    }

    public void handleEndElement(String element, String ns) {
        level--;
        payloadParser.handleEndElement(element, ns);
    }

    public void handleCharacterData(String data) {
        payloadParser.handleCharacterData(data);
    }

    public Payload getPayload() {
        if (payloadParser == null) {
            return null;
        }
        return payloadParser.getPayload();
    }
}