package cpedict

import (
  "compress/gzip"
  "encoding/xml"
  "os"
  "testing"
  "time"
  "reflect"
)

func TestDictionaryXMLUnmarshal(t *testing.T) {
  // open test data
  f, err := os.Open("testdata/test-0.xml.gz")
  if err != nil {
    t.Error(err)
    return
  }
  defer f.Close()

  // create gzip reader
  gz, err := gzip.NewReader(f)
  if err != nil {
    t.Error(err)
    return
  }
  defer gz.Close()

  // create xml decoder, decode xml
  d := xml.NewDecoder(gz)

  // decode xml, check for error
  var dict Dictionary
  if err := d.Decode(&dict); err != nil {
    t.Error(err)
    return
  }

  // build expected time
  var expTime time.Time
  if err := expTime.UnmarshalText([]byte("2022-02-02T04:51:00.437Z")); err != nil {
    t.Error(err)
    return
  }

  // expected generator
  expGenerator := Generator {
    ProductName: "National Vulnerability Database (NVD)",
    ProductVersion: "4.9",
    SchemaVersion: "2.3",
    Timestamp: expTime,
  }

  // compare generator
  if !reflect.DeepEqual(dict.Generator, expGenerator) {
    t.Errorf("got \"%v\", exp \"%v\"", dict.Generator, expGenerator)
    return
  }

  // check item count
  gotNumItems := len(dict.Items)
  expNumItems := 19
  if gotNumItems != expNumItems {
    t.Errorf("item count: got %d, exp %d", gotNumItems, expNumItems)
    return
  }

  // build expected item
  expItems := []Item {
    // first item in the list
    Item {
      CpeUri: "cpe:/a:%240.99_kindle_books_project:%240.99_kindle_books:6::~~~android~~",

      Cpe23Item: Cpe23Item {
        Name: "cpe:2.3:a:\\$0.99_kindle_books_project:\\$0.99_kindle_books:6:*:*:*:*:android:*:*",
      },

      Titles: []Title {
        Title {
          Lang: "en-US",
          Text: "$0.99 Kindle Books project $0.99 Kindle Books (aka com.kindle.books.for99) for android 6.0",
        },
      },

      References: []Reference {
        Reference {
          Href: "https://play.google.com/store/apps/details?id=com.kindle.books.for99",
          Text: "Product information",
        },

        Reference {
          Href: "https://docs.google.com/spreadsheets/d/1t5GXwjw82SyunALVJb2w0zi3FoLRIkfGPc7AMjRF0r4/edit?pli=1#gid=1053404143",
          Text: "Government Advisory",
        },
      },
    },

    // last item in the list
    Item {
      CpeUri: "cpe:/a:3com:3c16116-us:2.0",

      Cpe23Item: Cpe23Item {
        Name: "cpe:2.3:a:3com:3c16116-us:2.0:*:*:*:*:*:*:*",
      },

      Titles: []Title {
        Title {
          Lang: "ja-JP",
          Text: "スリーコム WebCache 3000 2.0",
        },

        Title {
          Lang: "en-US",
          Text: "3Com WebCache 3000 2.0",
        },
      },
    },
  }

  // build item comparisons
  compares := []struct {
    name string
    exp  Item
    got  Item
  } {
    { "head", expItems[0], dict.Items[0] },
    { "tail", expItems[1], dict.Items[len(dict.Items) - 1] },
  }

  for _, row := range(compares) {
    t.Run(row.name, func(t *testing.T) {
      if !reflect.DeepEqual(row.got, row.exp) {
        t.Errorf("got \"%v\", exp \"%v\"", row.got, row.exp)
      }
    })
  }
}