damus/damusTests/InvoiceTests.swift

//
//  InvoiceTests.swift
//  damusTests
//
//  Created by William Casarin on 2022-10-18.
//

import XCTest
@testable import damus


final class InvoiceTests: XCTestCase {

    override func setUpWithError() throws {
        // Put setup code here. This method is called before the invocation of each test method in the class.
    }

    override func tearDownWithError() throws {
        // Put teardown code here. This method is called after the invocation of each test method in the class.
    }
    
    
    func testParseAnyAmountInvoice() throws {
        let invstr = "LNBC1P3MR5UJSP5G7SA48YD4JWTTPCHWMY4QYN4UWZQCJQ8NMWKD6QE3HCRVYTDLH9SPP57YM9TSA9NN4M4XU59XMJCXKR7YDV29DDP6LVQUT46ZW6CU3KE9GQDQ9V9H8JXQ8P3MYLZJCQPJRZJQF60PZDVNGGQWQDNERZSQN35L8CVQ3QG2Z5NSZYD0D3Q0JW2TL6VUZA7FYQQWKGQQYQQQQLGQQQQXJQQ9Q9QXPQYSGQ39EM4QJMQFKZGJXZVGL7QJMYNSWA8PGDTAGXXRG5Z92M7VLCGKQK2L2THDF8LM0AUKAURH7FVAWDLRNMVF38W4EYJDNVN9V4Z9CRS5CQCV465C"
        
        guard let blockGroup: NdbBlockGroup = try? NdbBlockGroup.parse(content: invstr) else {
            XCTFail("Parsing threw an error")
            return
        }
        
        blockGroup.withList({ blockList in
            XCTAssertEqual(blockList.count, 1)
            let success: Bool? = blockList.useItem(at: 0, { block in
                switch block {
                case .invoice(let invoiceData):
                    let invoice = invoiceData.as_invoice()
                    XCTAssertEqual(invoice.amount, .any)
                    XCTAssertEqual(invoice.string, invstr)
                    return true
                default:
                    XCTFail("Block is not an invoice")
                    return false
                }
            })
            XCTAssertEqual(success, true)
        })
    }
    
    /*
    // gh-3144: It was decided on a standup meeting that we do not need invoices to render, few people use this feature.
    func testTextAfterInvoice() throws {
        let invstr = """
    LNBC1P3MR5UJSP5G7SA48YD4JWTTPCHWMY4QYN4UWZQCJQ8NMWKD6QE3HCRVYTDLH9SPP57YM9TSA9NN4M4XU59XMJCXKR7YDV29DDP6LVQUT46ZW6CU3KE9GQDQ9V9H8JXQ8P3MYLZJCQPJRZJQF60PZDVNGGQWQDNERZSQN35L8CVQ3QG2Z5NSZYD0D3Q0JW2TL6VUZA7FYQQWKGQQYQQQQLGQQQQXJQQ9Q9QXPQYSGQ39EM4QJMQFKZGJXZVGL7QJMYNSWA8PGDTAGXXRG5Z92M7VLCGKQK2L2THDF8LM0AUKAURH7FVAWDLRNMVF38W4EYJDNVN9V4Z9CRS5CQCV465C  hi there
    """
        
        guard let blockGroup: NdbBlockGroup = try? NdbBlockGroup.parse(content: invstr) else {
            XCTFail("Parsing threw an error")
            return
        }
        
        blockGroup.withList({ blockList in
            XCTAssertEqual(blockList.count, 2)
            
            // Check invoice block
            let invoiceSuccess: Bool? = blockList.useItem(at: 0, { block in
                switch block {
                case .invoice(let invoiceData):
                    guard let invoice = invoiceData.as_invoice() else {
                        XCTFail("Cannot get invoice from invoice block")
                        return false
                    }
                    XCTAssertEqual(invoice.amount, .any)
                    return true
                default:
                    XCTFail("First block is not an invoice")
                    return false
                }
            })
            XCTAssertEqual(invoiceSuccess, true)
            
            // Check text block
            let textSuccess: Bool? = blockList.useItem(at: 1, { block in
                switch block {
                case .text(let text):
                    XCTAssertEqual(text.as_str(), "  hi there")
                    return true
                default:
                    XCTFail("Second block is not text")
                    return false
                }
            })
            XCTAssertEqual(textSuccess, true)
        })
    }
     */
    
    func testParseInvoiceUpper() throws {
        let invstr = "LNBC100N1P357SL0SP5T9N56WDZTUN39LGDQLR30XQWKSG3K69Q4Q2RKR52APLUJW0ESN0QPP5MRQGLJK62Z20Q4NVGR6LZCYN6FHYLZCCWDVU4K77APG3ZMRKUJJQDPZW35XJUEQD9EJQCFQV3JHXCMJD9C8G6T0DCXQYJW5QCQPJRZJQT56H4GVP5YX36U2UZQA6QWCSK3E2DUUNFXPPZJ9VHYPC3WFE2WSWZ607UQQ3XQQQSQQQQQQQQQQQLQQYG9QYYSGQAGX5H20AEULJ3GDWX3KXS8U9F4MCAKDKWUAKASAMM9562FFYR9EN8YG20LG0YGNR9ZPWP68524KMDA0T5XP2WYTEX35PU8HAPYJAJXQPSQL29R"
        
        guard let blockGroup: NdbBlockGroup = try? NdbBlockGroup.parse(content: invstr) else {
            XCTFail("Parsing threw an error")
            return
        }
        
        blockGroup.withList({ blockList in
            XCTAssertEqual(blockList.count, 1)
            let success: Bool? = blockList.useItem(at: 0, { block in
                switch block {
                case .invoice(let invoiceData):
                    let invoice = invoiceData.as_invoice()
                    XCTAssertEqual(invoice.amount, .specific(10000))
                    XCTAssertEqual(invoice.expiry, 604800)
                    XCTAssertEqual(invoice.created_at, 1666139119)
                    XCTAssertEqual(invoice.string, invstr)
                    return true
                default:
                    XCTFail("Block is not an invoice")
                    return false
                }
            })
            XCTAssertEqual(success, true)
        })
    }

    func testParseInvoiceWithPrefix() throws {
        let invstr = "lightning:lnbc100n1p357sl0sp5t9n56wdztun39lgdqlr30xqwksg3k69q4q2rkr52aplujw0esn0qpp5mrqgljk62z20q4nvgr6lzcyn6fhylzccwdvu4k77apg3zmrkujjqdpzw35xjueqd9ejqcfqv3jhxcmjd9c8g6t0dcxqyjw5qcqpjrzjqt56h4gvp5yx36u2uzqa6qwcsk3e2duunfxppzj9vhypc3wfe2wswz607uqq3xqqqsqqqqqqqqqqqlqqyg9qyysgqagx5h20aeulj3gdwx3kxs8u9f4mcakdkwuakasamm9562ffyr9en8yg20lg0ygnr9zpwp68524kmda0t5xp2wytex35pu8hapyjajxqpsql29r"
        
        guard let blockGroup: NdbBlockGroup = try? NdbBlockGroup.parse(content: invstr) else {
            XCTFail("Parsing threw an error")
            return
        }
        
        blockGroup.withList({ blockList in
            XCTAssertEqual(blockList.count, 1)
            let success: Bool? = blockList.useItem(at: 0, { block in
                switch block {
                case .invoice(_):
                    return true
                default:
                    XCTFail("Block is not an invoice")
                    return false
                }
            })
            XCTAssertEqual(success, true)
        })
    }
    
    func testParseInvoiceWithPrefixCapitalized() throws {
        let invstr = "LIGHTNING:LNBC100N1P357SL0SP5T9N56WDZTUN39LGDQLR30XQWKSG3K69Q4Q2RKR52APLUJW0ESN0QPP5MRQGLJK62Z20Q4NVGR6LZCYN6FHYLZCCWDVU4K77APG3ZMRKUJJQDPZW35XJUEQD9EJQCFQV3JHXCMJD9C8G6T0DCXQYJW5QCQPJRZJQT56H4GVP5YX36U2UZQA6QWCSK3E2DUUNFXPPZJ9VHYPC3WFE2WSWZ607UQQ3XQQQSQQQQQQQQQQQLQQYG9QYYSGQAGX5H20AEULJ3GDWX3KXS8U9F4MCAKDKWUAKASAMM9562FFYR9EN8YG20LG0YGNR9ZPWP68524KMDA0T5XP2WYTEX35PU8HAPYJAJXQPSQL29R"
        
        guard let blockGroup: NdbBlockGroup = try? NdbBlockGroup.parse(content: invstr) else {
            XCTFail("Parsing threw an error")
            return
        }
        
        blockGroup.withList({ blockList in
            XCTAssertEqual(blockList.count, 1)
            let success: Bool? = blockList.useItem(at: 0, { block in
                switch block {
                case .invoice(_):
                    return true
                default:
                    XCTFail("Block is not an invoice")
                    return false
                }
            })
            XCTAssertEqual(success, true)
        })
    }

    /// Test parsing the specific invoice from GitHub issue that wasn't rendering
    func testParseSpecificFailingInvoice() throws {
        let invstr = "lnbc130n1p5h7alnpp5f83swv5wx9h25ansxsvkw7364c65vxktthy2m9ww5zf3cjrzp0vsdq9tfpygcqzysxqzjcsp5essuf0xnfeu4rpw7nllcggr6e9635xdpnaklr2fadtkwej0vvyfs9qxpqysgqddjjzxa2dwhntx8uvppx3u6pu864ul5dxkayp6jgf7n45ql5x7u9xzrvuav5rzsaz7h8d2gq455je2ezku40a5xrshu0w00ylprk03qq6kvvjd"

        guard let blockGroup: NdbBlockGroup = try? NdbBlockGroup.parse(content: invstr) else {
            XCTFail("Parsing threw an error")
            return
        }

        blockGroup.withList({ blockList in
            XCTAssertEqual(blockList.count, 1, "Expected 1 block, got \(blockList.count)")
            let success: Bool? = blockList.useItem(at: 0, { block in
                switch block {
                case .invoice(let invoiceData):
                    let invoice = invoiceData.as_invoice()
                    XCTAssertEqual(invoice.amount, .specific(13000))
                    return true
                case .text(let txt):
                    XCTFail("Expected invoice block, got text block")
                    return false
                default:
                    XCTFail("Block is not an invoice")
                    return false
                }
            })
            XCTAssertEqual(success, true)
        })
    }

    /*
    // gh-3144: It was decided on a standup meeting that we do not need invoices to render, few people use this feature.
    func testParseInvoice() throws {
        let invstr = " lnbc100n1p357sl0sp5t9n56wdztun39lgdqlr30xqwksg3k69q4q2rkr52aplujw0esn0qpp5mrqgljk62z20q4nvgr6lzcyn6fhylzccwdvu4k77apg3zmrkujjqdpzw35xjueqd9ejqcfqv3jhxcmjd9c8g6t0dcxqyjw5qcqpjrzjqt56h4gvp5yx36u2uzqa6qwcsk3e2duunfxppzj9vhypc3wfe2wswz607uqq3xqqqsqqqqqqqqqqqlqqyg9qyysgqagx5h20aeulj3gdwx3kxs8u9f4mcakdkwuakasamm9562ffyr9en8yg20lg0ygnr9zpwp68524kmda0t5xp2wytex35pu8hapyjajxqpsql29r"

        guard let blockGroup: NdbBlockGroup = try? NdbBlockGroup.parse(content: invstr) else {
            XCTFail("Parsing threw an error")
            return
        }
        blockGroup.withList({ blockList in
            XCTAssertEqual(blockList.count, 3)
            let success: Bool? = blockList.useItem(at: 1, { block in
                switch block {
                case .invoice(let invoiceData):
                    guard let invoice = invoiceData.as_invoice() else {
                        XCTFail("Cannot get invoice from invoice block")
                        return false
                    }
                    XCTAssertEqual(invoice.amount, .specific(10000))
                    XCTAssertEqual(invoice.expiry, 604800)
                    XCTAssertEqual(invoice.created_at, 1666139119)
                    XCTAssertEqual(invoice.string, invstr)
                    return true
                default:
                    XCTFail("Block is not an invoice")
                    return false
                }
            })
            XCTAssertEqual(success, true)
        })
    }
    */

}