diff options
Diffstat (limited to 'importer')
| -rw-r--r-- | importer/importer.go | 89 | ||||
| -rw-r--r-- | importer/importer_test.go | 149 |
2 files changed, 0 insertions, 238 deletions
diff --git a/importer/importer.go b/importer/importer.go deleted file mode 100644 index 73a2cd8..0000000 --- a/importer/importer.go +++ /dev/null @@ -1,89 +0,0 @@ -package importer - -import ( - // "bufio" - "encoding/json" - //"fmt" - "io" - "log" - "os" - - "adammathes.com/neko/models/feed" - "adammathes.com/neko/models/item" -) - -type IItem struct { - Title string `json:"title"` - Url string `json:"url"` - Description string `json:"description"` - ReadState bool `json:"read"` - Starred bool `json:"starred"` - Date *IDate `json:"date"` - Feed *IFeed `json:"feed"` -} - -type IFeed struct { - Url string `json:"url"` - Title string `json:"title"` - WebUrl string `json:"web_url"` -} - -type IDate struct { - Date string `json:"$date"` -} - -func ImportJSON(filename string) error { - - f, err := os.Open(filename) - if err != nil { - return err - } - defer f.Close() - - dec := json.NewDecoder(f) - for { - var ii IItem - if err := dec.Decode(&ii); err == io.EOF { - break - } else if err != nil { - return err - } else { - err := InsertIItem(&ii) - if err != nil { - log.Println(err) - } - } - } - return nil -} - -func InsertIItem(ii *IItem) error { - var f feed.Feed - - if ii.Feed == nil { - return nil - } - err := f.ByUrl(ii.Feed.Url) - if err != nil { - f.Url = ii.Feed.Url - f.Title = ii.Feed.Title - err = f.Create() - if err != nil { - return err - } - } - - var i item.Item - i.FeedId = f.Id - i.Title = ii.Title - i.Url = ii.Url - i.Description = ii.Description - - if ii.Date != nil { - i.PublishDate = ii.Date.Date - } - - err = i.Create() - log.Printf("inserted %s\n", i.Url) - return err -} diff --git a/importer/importer_test.go b/importer/importer_test.go deleted file mode 100644 index 59f06f1..0000000 --- a/importer/importer_test.go +++ /dev/null @@ -1,149 +0,0 @@ -package importer - -import ( - "os" - "path/filepath" - "testing" - - "adammathes.com/neko/config" - "adammathes.com/neko/models" -) - -func setupTestDB(t *testing.T) { - t.Helper() - config.Config.DBFile = filepath.Join(t.TempDir(), "test.db") - models.InitDB() - t.Cleanup(func() { - if models.DB != nil { - models.DB.Close() - } - }) -} - -func TestInsertIItem(t *testing.T) { - setupTestDB(t) - - ii := &IItem{ - Title: "Test Article", - Url: "https://example.com/article", - Description: "A test article description", - ReadState: false, - Starred: true, - Date: &IDate{Date: "2024-01-15 10:00:00"}, - Feed: &IFeed{ - Url: "https://example.com/feed", - Title: "Example Feed", - }, - } - - InsertIItem(ii) - - // Verify the feed was created - var feedCount int - models.DB.QueryRow("SELECT COUNT(*) FROM feed").Scan(&feedCount) - if feedCount != 1 { - t.Errorf("Expected 1 feed, got %d", feedCount) - } - - // Verify the item was created - var itemCount int - models.DB.QueryRow("SELECT COUNT(*) FROM item").Scan(&itemCount) - if itemCount != 1 { - t.Errorf("Expected 1 item, got %d", itemCount) - } -} - -func TestInsertIItemNilFeed(t *testing.T) { - setupTestDB(t) - - ii := &IItem{ - Title: "No Feed Item", - Url: "https://example.com/nofeed", - Feed: nil, - } - - // Should not panic - InsertIItem(ii) - - var itemCount int - models.DB.QueryRow("SELECT COUNT(*) FROM item").Scan(&itemCount) - if itemCount != 0 { - t.Errorf("Expected 0 items (nil feed should be skipped), got %d", itemCount) - } -} - -func TestInsertIItemExistingFeed(t *testing.T) { - setupTestDB(t) - - // Insert feed first - models.DB.Exec("INSERT INTO feed(url, title) VALUES(?, ?)", "https://example.com/feed", "Existing Feed") - - ii := &IItem{ - Title: "New Article", - Url: "https://example.com/new-article", - Description: "New article desc", - Date: &IDate{Date: "2024-01-15"}, - Feed: &IFeed{ - Url: "https://example.com/feed", - Title: "Existing Feed", - }, - } - - InsertIItem(ii) - - // Should still be just 1 feed - var feedCount int - models.DB.QueryRow("SELECT COUNT(*) FROM feed").Scan(&feedCount) - if feedCount != 1 { - t.Errorf("Expected 1 feed (reuse existing), got %d", feedCount) - } -} - -func TestImportJSON(t *testing.T) { - setupTestDB(t) - - dir := t.TempDir() - jsonFile := filepath.Join(dir, "import.json") - - content := `{"title":"Article 1","url":"https://example.com/1","description":"desc1","read":false,"starred":false,"date":{"$date":"2024-01-01"},"feed":{"url":"https://example.com/feed","title":"Feed 1"}} -{"title":"Article 2","url":"https://example.com/2","description":"desc2","read":true,"starred":true,"date":{"$date":"2024-01-02"},"feed":{"url":"https://example.com/feed","title":"Feed 1"}}` - - err := os.WriteFile(jsonFile, []byte(content), 0644) - if err != nil { - t.Fatal(err) - } - - ImportJSON(jsonFile) - - var itemCount int - models.DB.QueryRow("SELECT COUNT(*) FROM item").Scan(&itemCount) - if itemCount != 2 { - t.Errorf("Expected 2 items after import, got %d", itemCount) - } - - var feedCount int - models.DB.QueryRow("SELECT COUNT(*) FROM feed").Scan(&feedCount) - if feedCount != 1 { - t.Errorf("Expected 1 feed after import, got %d", feedCount) - } -} - -func TestImportJSONInvalid(t *testing.T) { - setupTestDB(t) - dir := t.TempDir() - jsonFile := filepath.Join(dir, "invalid.json") - os.WriteFile(jsonFile, []byte("not json"), 0644) - - err := ImportJSON(jsonFile) - if err == nil { - t.Error("ImportJSON should error on invalid JSON") - } -} - -func TestImportJSONNonexistent(t *testing.T) { - setupTestDB(t) - err := ImportJSON("/nonexistent/file.json") - if err == nil { - t.Error("ImportJSON should error on nonexistent file") - } -} |