Initial word table SQLite importer

main_test.go

@@ -0,0 +1,134 @@
+package main
+
+import (
+	"archive/zip"
+	"context"
+	"database/sql"
+	"os"
+	"path/filepath"
+	"testing"
+
+	_ "modernc.org/sqlite"
+)
+
+func TestParseDocumentXMLExtractsTableCells(t *testing.T) {
+	xml := `<?xml version="1.0" encoding="UTF-8"?>
+<w:document xmlns:w="http://schemas.openxmlformats.org/wordprocessingml/2006/main">
+  <w:body>
+    <w:tbl>
+      <w:tr>
+        <w:tc><w:tcPr><w:gridSpan w:val="2"/></w:tcPr><w:p><w:r><w:t>报备表</w:t></w:r></w:p></w:tc>
+      </w:tr>
+      <w:tr>
+        <w:tc><w:p><w:r><w:t>序号</w:t></w:r></w:p></w:tc>
+        <w:tc><w:p><w:r><w:t>事件过程</w:t></w:r></w:p></w:tc>
+      </w:tr>
+      <w:tr>
+        <w:tc><w:p><w:r><w:t>1</w:t></w:r></w:p></w:tc>
+        <w:tc><w:p><w:r><w:t>起因</w:t><w:br/><w:t>处理经过</w:t></w:r></w:p></w:tc>
+      </w:tr>
+    </w:tbl>
+  </w:body>
+</w:document>`
+
+	tables, err := parseDocumentXML([]byte(xml))
+	if err != nil {
+		t.Fatalf("parseDocumentXML returned error: %v", err)
+	}
+	if len(tables) != 1 {
+		t.Fatalf("got %d tables, want 1", len(tables))
+	}
+	if got := tables[0].Rows[0].Cells[0].Text; got != "报备表" {
+		t.Fatalf("title cell text = %q, want 报备表", got)
+	}
+	if got := tables[0].Rows[0].Cells[0].ColSpan; got != 2 {
+		t.Fatalf("title cell colspan = %d, want 2", got)
+	}
+	if got := tables[0].Rows[2].Cells[1].Text; got != "起因\n处理经过" {
+		t.Fatalf("multiline text = %q, want 起因\\n处理经过", got)
+	}
+}
+
+func TestInferHeaderFieldsSkipsTitleRow(t *testing.T) {
+	table := Table{
+		Rows: []Row{
+			{Cells: []Cell{{Text: "报备表", ColSpan: 2}}},
+			{Cells: []Cell{{Text: "序号", ColSpan: 1}, {Text: "事件过程", ColSpan: 1}}},
+			{Cells: []Cell{{Text: "1", ColSpan: 1}, {Text: "客户反映设备位置不合理", ColSpan: 1}}},
+		},
+	}
+
+	fields := inferFields(table)
+	if len(fields) != 2 {
+		t.Fatalf("got %d fields, want 2", len(fields))
+	}
+	if fields[0].Key != "序号" || fields[0].Value != "1" || fields[0].SourceKind != "header_row" {
+		t.Fatalf("unexpected first field: %#v", fields[0])
+	}
+	if fields[1].Key != "事件过程" || fields[1].Value != "客户反映设备位置不合理" {
+		t.Fatalf("unexpected second field: %#v", fields[1])
+	}
+}
+
+func TestImportDocumentWritesSQLiteTables(t *testing.T) {
+	dbPath := filepath.Join(t.TempDir(), "import.sqlite")
+	db, err := sql.Open("sqlite", dbPath)
+	if err != nil {
+		t.Fatalf("open sqlite: %v", err)
+	}
+	defer db.Close()
+
+	xml := `<?xml version="1.0" encoding="UTF-8"?>
+<w:document xmlns:w="http://schemas.openxmlformats.org/wordprocessingml/2006/main">
+  <w:body>
+    <w:tbl>
+      <w:tr><w:tc><w:tcPr><w:gridSpan w:val="2"/></w:tcPr><w:p><w:r><w:t>报备表</w:t></w:r></w:p></w:tc></w:tr>
+      <w:tr><w:tc><w:p><w:r><w:t>序号</w:t></w:r></w:p></w:tc><w:tc><w:p><w:r><w:t>事件过程</w:t></w:r></w:p></w:tc></w:tr>
+      <w:tr><w:tc><w:p><w:r><w:t>1</w:t></w:r></w:p></w:tc><w:tc><w:p><w:r><w:t>客户反映设备位置不合理</w:t></w:r></w:p></w:tc></w:tr>
+    </w:tbl>
+  </w:body>
+</w:document>`
+
+	docxPath := filepath.Join(t.TempDir(), "sample.docx")
+	if err := writeTestDocx(docxPath, []byte(xml)); err != nil {
+		t.Fatalf("write test docx: %v", err)
+	}
+
+	if err := importDocument(context.Background(), db, docxPath); err != nil {
+		t.Fatalf("importDocument: %v", err)
+	}
+
+	var cells int
+	if err := db.QueryRow(`SELECT COUNT(*) FROM cells`).Scan(&cells); err != nil {
+		t.Fatalf("count cells: %v", err)
+	}
+	if cells != 5 {
+		t.Fatalf("got %d cells, want 5", cells)
+	}
+
+	var value string
+	if err := db.QueryRow(`SELECT value FROM fields WHERE key = '事件过程'`).Scan(&value); err != nil {
+		t.Fatalf("query field: %v", err)
+	}
+	if value != "客户反映设备位置不合理" {
+		t.Fatalf("field value = %q", value)
+	}
+}
+
+func writeTestDocx(path string, documentXML []byte) error {
+	f, err := os.Create(path)
+	if err != nil {
+		return err
+	}
+	defer f.Close()
+
+	zw := zip.NewWriter(f)
+	defer zw.Close()
+
+	w, err := zw.Create("word/document.xml")
+	if err != nil {
+		return err
+	}
+	_, err = w.Write(documentXML)
+	return err
+}