updates uniprot to read IDs (#104)

README.md

@@ -76,6 +76,7 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
 ## [Unreleased]
+- Updates uniprot parser to read IDs [#104](https://github.com/Koeng101/dnadesign/pull/104)
 - Fixes RecursiveFragment to not add flanks to the initial input [#102](https://github.com/Koeng101/dnadesign/pull/102)
 - Fixes add flank bug, releases new version of python lib [#101](https://github.com/Koeng101/dnadesign/pull/101)
 - Adds feature for adding flanks to RecursiveFragment. [#100](https://github.com/Koeng101/dnadesign/pull/100)

lib/bio/uniprot/uniprot.go

@@ -65,35 +65,35 @@ type Parser struct {
 	decoder Decoder
 }
 
-// NewParser returns a Parser that uses r as the source
-// from which to parse fasta formatted sequences.
 func NewParser(r io.Reader) *Parser {
 	decoder := xml.NewDecoder(r)
 	return &Parser{decoder: decoder}
 }
 
 func (p *Parser) Next() (Entry, error) {
-	decoderToken, err := p.decoder.Token()
+	for {
+		decoderToken, err := p.decoder.Token()
 
-	// Check decoding
-	if err != nil {
-		// If we are the end of the file, return io.EOF
-		if err.Error() == "EOF" {
-			return Entry{}, io.EOF
-		}
-	}
-
-	// Actual parsing
-	startElement, ok := decoderToken.(xml.StartElement)
-	if ok && startElement.Name.Local == "entry" {
-		var e Entry
-		err = p.decoder.DecodeElement(&e, &startElement)
+		// Check decoding
 		if err != nil {
+			// If we are the end of the file, return io.EOF
+			if err.Error() == "EOF" {
+				return Entry{}, io.EOF
+			}
 			return Entry{}, err
 		}
-		return e, nil
+
+		// Actual parsing
+		startElement, ok := decoderToken.(xml.StartElement)
+		if ok && startElement.Name.Local == "entry" {
+			var e Entry
+			err = p.decoder.DecodeElement(&e, &startElement)
+			if err != nil {
+				return Entry{}, err
+			}
+			return e, nil
+		}
 	}
-	return p.Next()
 }
 
 // BaseURL encodes the base URL for the Uniprot REST API.

lib/bio/uniprot/uniprot_test.go

@@ -110,4 +110,11 @@ func TestGet(t *testing.T) {
 	if err == nil {
 		t.Errorf("Expected an error for invalid URL, but got none")
 	}
+	for _, reference := range entry.DbReference {
+		if reference.Type == "Pfam" {
+			if reference.Id != "PF01353" {
+				t.Errorf("Expected Pfam ID PF01353")
+			}
+		}
+	}
 }