TDQuoteStateTest.m

//  Copyright 2010 Todd Ditchendorf
//
//  Licensed under the Apache License, Version 2.0 (the "License");
//  you may not use this file except in compliance with the License.
//  You may obtain a copy of the License at
//
//  http://www.apache.org/licenses/LICENSE-2.0
//
//  Unless required by applicable law or agreed to in writing, software
//  distributed under the License is distributed on an "AS IS" BASIS,
//  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//  See the License for the specific language governing permissions and
//  limitations under the License.

#import "TDQuoteStateTest.h"


@implementation TDQuoteStateTest

- (void)setUp {
    t = [PKTokenizer tokenizer];
    quoteState = t.quoteState;
}


- (void)tearDown {
}


- (void)testQuotedString {
    s = @"'stuff'";
    t.string = s;
    PKToken *tok = [t nextToken];
    TDEqualObjects(s, tok.stringValue);
}


- (void)testQuotedStringEscaped {
    s = @"'it\\'s'";
    t.string = s;
    PKToken *tok = [t nextToken];
    TDEqualObjects(s, tok.stringValue);
}


- (void)testQuotedStringEscaped2 {
    s = @"'it\\'s'";
    t.string = s;
    quoteState.usesCSVStyleEscaping = YES;
    
    PKToken *tok = [t nextToken];
    TDEqualObjects(@"'it\\'", tok.stringValue);
    
    tok = [t nextToken];
    TDEqualObjects(@"s'", tok.stringValue);
    TDTrue(tok.isWord);
    
    tok = [t nextToken];
    TDEquals([PKToken EOFToken], tok);
}


- (void)testQuotedStringEscapedDouble {
    s = @"'it''s'";
    t.string = s;
    PKToken *tok = [t nextToken];
    TDEqualObjects(@"'it'", tok.stringValue);

    tok = [t nextToken];
    TDEqualObjects(@"'s'", tok.stringValue);

    tok = [t nextToken];
    TDEquals([PKToken EOFToken], tok);
}


- (void)testQuotedStringEscapedDouble2 {
    s = @"'it''s'";
    t.string = s;
    quoteState.usesCSVStyleEscaping = YES;
    
    PKToken *tok = [t nextToken];
    TDEqualObjects(s, tok.stringValue);
    
    tok = [t nextToken];
    TDEquals([PKToken EOFToken], tok);
}


- (void)testQuotedStringEscapedDouble3 {
    s = @"'it''s' cool";
    t.string = s;
    quoteState.usesCSVStyleEscaping = YES;
    
    PKToken *tok = [t nextToken];
    TDEqualObjects(@"'it''s'", tok.stringValue);
    
    tok = [t nextToken];
    TDEqualObjects(@"cool", tok.stringValue);
    
    tok = [t nextToken];
    TDEquals([PKToken EOFToken], tok);
}


- (void)testQuotedStringEscapedDouble4 {
    s = @"'it''s'cool";
    t.string = s;
    quoteState.usesCSVStyleEscaping = YES;
    
    PKToken *tok = [t nextToken];
    TDEqualObjects(@"'it''s'", tok.stringValue);
    
    tok = [t nextToken];
    TDEqualObjects(@"cool", tok.stringValue);
    
    tok = [t nextToken];
    TDEquals([PKToken EOFToken], tok);
}


- (void)testQuotedStringEOFTerminated {
    s = @"'stuff";
    t.string = s;
    PKToken *tok = [t nextToken];
    TDEqualObjects(s, tok.stringValue);
}


- (void)testQuotedStringRepairEOFTerminated {
    s = @"'stuff";
    t.string = s;
    quoteState.balancesEOFTerminatedQuotes = YES;
    PKToken *tok = [t nextToken];
    TDEqualObjects(@"'stuff'", tok.stringValue);
}


- (void)testQuotedStringPlus {
    s = @"'a quote here' more";
    t.string = s;
    PKToken *tok = [t nextToken];
    TDEqualObjects(@"'a quote here'", tok.stringValue);
}


- (void)test14CharQuotedString {
    s = @"'123456789abcef'";
    t.string = s;
    PKToken *tok = [t nextToken];
    TDEqualObjects(s, tok.stringValue);
    TDTrue(tok.isQuotedString);
}


- (void)test15CharQuotedString {
    s = @"'123456789abcefg'";
    t.string = s;
    PKToken *tok = [t nextToken];
    TDEqualObjects(s, tok.stringValue);
    TDTrue(tok.isQuotedString);
}


- (void)test16CharQuotedString {
    s = @"'123456789abcefgh'";
    t.string = s;
    PKToken *tok = [t nextToken];
    TDEqualObjects(s, tok.stringValue);
    TDTrue(tok.isQuotedString);
}


- (void)test31CharQuotedString {
    s = @"'123456789abcefgh123456789abcefg'";
    t.string = s;
    PKToken *tok = [t nextToken];
    TDEqualObjects(s, tok.stringValue);
    TDTrue(tok.isQuotedString);
}


- (void)test32CharQuotedString {
    s = @"'123456789abcefgh123456789abcefgh'";
    t.string = s;
    PKToken *tok = [t nextToken];
    TDEqualObjects(s, tok.stringValue);
    TDTrue(tok.isQuotedString);
}

@end