Initial commit

2025-11-11 16:40:10 -07:00
commit b79e599640
57 changed files with 11811 additions and 0 deletions
@@ -0,0 +1,264 @@
+package internal
+
+
+import (
+	"os"
+	"strings"
+	"testing"
+	"time"
+)
+
+const sampleXML = `<?xml version='1.0' encoding='UTF-8' standalone='yes' ?>
+<?xml-stylesheet type="text/xsl" href="sms.xsl"?>
+<smses count="2">
+  <sms protocol="0" address="332" date="1285799668193" type="2" subject="null" body="Sample Message Sent from the phone" toa="null" sc_toa="null" service_center="null" read="1" status="-1" locked="0" readable_date="Sep 30, 2010 8:34:28 AM" contact_name="(Unknown)" />
+  <sms protocol="0" address="4433221123" date="1289643415810" type="1" subject="null" body="Sample Message received by the phone" toa="null" sc_toa="null" service_center="null" read="0" status="-1" locked="0" readable_date="Nov 13, 2010 9:16:55 PM" contact_name="(Unknown)" />
+</smses>`
+
+func TestSampleXMLParsing(t *testing.T) {
+	// Parse the XML
+	reader := strings.NewReader(sampleXML)
+	result, err := ParseSMSBackup(reader)
+	if err != nil {
+		t.Fatalf("Failed to parse XML: %v", err)
+	}
+
+	// Verify we got 2 messages
+	if len(result.Messages) != 2 {
+		t.Errorf("Expected 2 messages, got %d", len(result.Messages))
+	}
+
+	// Verify first message (sent)
+	msg1 := result.Messages[0]
+	if msg1.Address != "332" {
+		t.Errorf("Expected address '332', got '%s'", msg1.Address)
+	}
+	if msg1.Type != 2 {
+		t.Errorf("Expected type 2 (sent), got %d", msg1.Type)
+	}
+	if msg1.Body != "Sample Message Sent from the phone" {
+		t.Errorf("Expected body 'Sample Message Sent from the phone', got '%s'", msg1.Body)
+	}
+	if msg1.Protocol != 0 {
+		t.Errorf("Expected protocol 0, got %d", msg1.Protocol)
+	}
+	if !msg1.Read {
+		t.Errorf("Expected message to be read (read=1)")
+	}
+	if msg1.Status != -1 {
+		t.Errorf("Expected status -1, got %d", msg1.Status)
+	}
+	// Check date: 1285799668193 milliseconds = Sep 30, 2010 8:34:28 AM
+	expectedDate1 := time.Unix(1285799668, 0)
+	if !msg1.Date.Equal(expectedDate1) {
+		t.Errorf("Expected date %v, got %v", expectedDate1, msg1.Date)
+	}
+
+	// Verify second message (received)
+	msg2 := result.Messages[1]
+	// Phone number normalization adds +1 to 10-digit US numbers
+	if msg2.Address != "+14433221123" {
+		t.Errorf("Expected address '+14433221123', got '%s'", msg2.Address)
+	}
+	if msg2.Type != 1 {
+		t.Errorf("Expected type 1 (received), got %d", msg2.Type)
+	}
+	if msg2.Body != "Sample Message received by the phone" {
+		t.Errorf("Expected body 'Sample Message received by the phone', got '%s'", msg2.Body)
+	}
+	if msg2.Read {
+		t.Errorf("Expected message to be unread (read=0)")
+	}
+	// Check date: 1289643415810 milliseconds = Nov 13, 2010 9:16:55 PM
+	expectedDate2 := time.Unix(1289643415, 0)
+	if !msg2.Date.Equal(expectedDate2) {
+		t.Errorf("Expected date %v, got %v", expectedDate2, msg2.Date)
+	}
+
+	// Verify no call logs in this sample
+	if len(result.Calls) != 0 {
+		t.Errorf("Expected 0 call logs, got %d", len(result.Calls))
+	}
+}
+
+func TestSampleXMLDatabaseIngestion(t *testing.T) {
+	// Create a temporary database file
+	tmpDB := "test_messages.db"
+	defer os.Remove(tmpDB) // Clean up after test
+
+	// Initialize database
+	err := InitDB(tmpDB)
+	if err != nil {
+		t.Fatalf("Failed to initialize database: %v", err)
+	}
+	defer db.Close()
+
+	// Parse the XML
+	reader := strings.NewReader(sampleXML)
+	result, err := ParseSMSBackup(reader)
+	if err != nil {
+		t.Fatalf("Failed to parse XML: %v", err)
+	}
+
+	// Insert messages into database
+	messageCount := 0
+	for i := range result.Messages {
+		err := InsertMessage(db, &result.Messages[i])
+		if err != nil {
+			t.Errorf("Failed to insert message %d: %v", i, err)
+			continue
+		}
+		messageCount++
+
+		// Verify the ID was set
+		if result.Messages[i].ID == 0 {
+			t.Errorf("Message %d: ID was not set after insert", i)
+		}
+	}
+
+	// Verify we inserted 2 messages
+	if messageCount != 2 {
+		t.Errorf("Expected to insert 2 messages, inserted %d", messageCount)
+	}
+
+	// Retrieve messages from database and verify
+	messages, err := GetMessages(db, "332", nil, nil)
+	if err != nil {
+		t.Fatalf("Failed to retrieve messages for address '332': %v", err)
+	}
+	if len(messages) != 1 {
+		t.Errorf("Expected 1 message for address '332', got %d", len(messages))
+	} else {
+		msg := messages[0]
+		if msg.Body != "Sample Message Sent from the phone" {
+			t.Errorf("Retrieved message has wrong body: '%s'", msg.Body)
+		}
+		if msg.Type != 2 {
+			t.Errorf("Retrieved message has wrong type: %d", msg.Type)
+		}
+		if msg.Protocol != 0 {
+			t.Errorf("Retrieved message has wrong protocol: %d", msg.Protocol)
+		}
+		if msg.Status != -1 {
+			t.Errorf("Retrieved message has wrong status: %d", msg.Status)
+		}
+		if !msg.Read {
+			t.Errorf("Retrieved message should be marked as read")
+		}
+	}
+
+	// Retrieve second message
+	messages2, err := GetMessages(db, "+14433221123", nil, nil)
+	if err != nil {
+		t.Fatalf("Failed to retrieve messages for address '+14433221123': %v", err)
+	}
+	if len(messages2) != 1 {
+		t.Errorf("Expected 1 message for address '+14433221123', got %d", len(messages2))
+	} else {
+		msg := messages2[0]
+		if msg.Body != "Sample Message received by the phone" {
+			t.Errorf("Retrieved message has wrong body: '%s'", msg.Body)
+		}
+		if msg.Type != 1 {
+			t.Errorf("Retrieved message has wrong type: %d", msg.Type)
+		}
+		if msg.Read {
+			t.Errorf("Retrieved message should be marked as unread")
+		}
+	}
+
+	// Test GetConversations
+	conversations, err := GetConversations(db, nil, nil)
+	if err != nil {
+		t.Fatalf("Failed to get conversations: %v", err)
+	}
+	if len(conversations) != 2 {
+		t.Errorf("Expected 2 conversations, got %d", len(conversations))
+	}
+
+	// Verify conversations are sorted by date (most recent first)
+	// Second message (1289643415) should be first as it's more recent
+	if len(conversations) == 2 {
+		if conversations[0].Address != "+14433221123" {
+			t.Errorf("Expected first conversation to be '+14433221123', got '%s'", conversations[0].Address)
+		}
+		if conversations[1].Address != "332" {
+			t.Errorf("Expected second conversation to be '332', got '%s'", conversations[1].Address)
+		}
+		if conversations[0].MessageCount != 1 {
+			t.Errorf("Expected first conversation to have 1 message, got %d", conversations[0].MessageCount)
+		}
+		if conversations[0].Type != "conversation" {
+			t.Errorf("Expected conversation type to be 'conversation', got '%s'", conversations[0].Type)
+		}
+	}
+
+	// Test date range functionality
+	startDate := time.Unix(1289000000, 0) // After first message, before second
+	messages3, err := GetMessages(db, "332", &startDate, nil)
+	if err != nil {
+		t.Fatalf("Failed to retrieve messages with date filter: %v", err)
+	}
+	if len(messages3) != 0 {
+		t.Errorf("Expected 0 messages after start date, got %d", len(messages3))
+	}
+
+	// Get date range
+	minDate, maxDate, err := GetDateRange(db)
+	if err != nil {
+		t.Fatalf("Failed to get date range: %v", err)
+	}
+	expectedMin := time.Unix(1285799668, 0)
+	expectedMax := time.Unix(1289643415, 0)
+	if !minDate.Equal(expectedMin) {
+		t.Errorf("Expected min date %v, got %v", expectedMin, minDate)
+	}
+	if !maxDate.Equal(expectedMax) {
+		t.Errorf("Expected max date %v, got %v", expectedMax, maxDate)
+	}
+}
+
+func TestEmptyXML(t *testing.T) {
+	emptyXML := `<?xml version='1.0' encoding='UTF-8' standalone='yes' ?>
+<smses count="0">
+</smses>`
+
+	reader := strings.NewReader(emptyXML)
+	result, err := ParseSMSBackup(reader)
+	if err != nil {
+		t.Fatalf("Failed to parse empty XML: %v", err)
+	}
+
+	if len(result.Messages) != 0 {
+		t.Errorf("Expected 0 messages, got %d", len(result.Messages))
+	}
+	if len(result.Calls) != 0 {
+		t.Errorf("Expected 0 calls, got %d", len(result.Calls))
+	}
+}
+
+func TestInvalidXML(t *testing.T) {
+	invalidXML := `<?xml version='1.0' encoding='UTF-8' standalone='yes' ?>
+<smses count="1">
+  <sms protocol="invalid" address="123" date="notanumber" type="2" body="Test" />
+</smses>`
+
+	reader := strings.NewReader(invalidXML)
+	result, err := ParseSMSBackup(reader)
+
+	// Should parse but skip invalid entries or use defaults
+	if err != nil {
+		t.Fatalf("Parser should handle invalid data gracefully: %v", err)
+	}
+
+	// The message might be parsed with default values for invalid fields
+	if len(result.Messages) > 0 {
+		msg := result.Messages[0]
+		// Protocol "invalid" should parse as 0
+		if msg.Protocol != 0 {
+			t.Logf("Invalid protocol parsed as: %d", msg.Protocol)
+		}
+		// Date "notanumber" should result in Unix epoch
+		t.Logf("Invalid date parsed as: %v", msg.Date)
+	}
+}