aboutsummaryrefslogtreecommitdiff
path: root/feed/feed_test.go
blob: f31a3ae9591207c514586aae7ae092a8e9115009 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
package feed

import (
  "compress/gzip"
  "encoding/json"
  "io"
  // "fmt"
  "os"
  "testing"
)

func openTest(path string) (io.Reader, error) {
  // open file for reading
  file, err := os.Open(path)
  if err != nil {
    return nil, err
  }

  // wrap in reader, return success
  return gzip.NewReader(file)
}

// Test feed parser
func TestFeedParser(t *testing.T) {
  t.Run("TestUnmarshalJSON", func(t *testing.T) {
    var f Feed

    // read test data, check for error
    src, err := openTest("testdata/nvdcve-1.1-2021.json.gz")
    if err != nil {
      t.Error(err)
    }

    // decode cve feed, check for error
    d := json.NewDecoder(src)
    if err := d.Decode(&f); err != nil {
      t.Error(err)
    }
  })
//   var f Feed
// 
//   // decode cve feed
//   d := json.NewDecoder(os.Stdin)
//   if err := d.Decode(&f); err != nil {
//     t.Error(err)
//   }
// 
//   var dst bytes.Buffer
// 
//   // create json encoder
//   e := json.NewEncoder(&dst)
//   if err := e.Encode(f); err != nil {
//     t.Error(err)
//   }
}