IO and sorting

Search…
IO and sorting
​You can find all the code for this chapter here​
​In the previous chapter we continued iterating on our application by adding a new endpoint /league. Along the way we learned about how to deal with JSON, embedding types and routing.
Our product owner is somewhat perturbed by the software losing the scores when the server was restarted. This is because our implementation of our store is in-memory. She is also not pleased that we didn't interpret the /league endpoint should return the players ordered by the number of wins!
The code so far
1
// server.go
2
package main
3
​
4
import (
5
	"encoding/json"
6
	"fmt"
7
	"net/http"
8
	"strings"
9
)
10
​
11
// PlayerStore stores score information about players
12
type PlayerStore interface {
13
	GetPlayerScore(name string) int
14
	RecordWin(name string)
15
	GetLeague() []Player
16
}
17
​
18
// Player stores a name with a number of wins
19
type Player struct {
20
	Name string
21
	Wins int
22
}
23
​
24
// PlayerServer is a HTTP interface for player information
25
type PlayerServer struct {
26
	store PlayerStore
27
	http.Handler
28
}
29
​
30
const jsonContentType = "application/json"
31
​
32
// NewPlayerServer creates a PlayerServer with routing configured
33
func NewPlayerServer(store PlayerStore) *PlayerServer {
34
	p := new(PlayerServer)
35
​
36
	p.store = store
37
​
38
	router := http.NewServeMux()
39
	router.Handle("/league", http.HandlerFunc(p.leagueHandler))
40
	router.Handle("/players/", http.HandlerFunc(p.playersHandler))
41
​
42
	p.Handler = router
43
​
44
	return p
45
}
46
​
47
func (p *PlayerServer) leagueHandler(w http.ResponseWriter, r *http.Request) {
48
	w.Header().Set("content-type", jsonContentType)
49
	json.NewEncoder(w).Encode(p.store.GetLeague())
50
}
51
​
52
func (p *PlayerServer) playersHandler(w http.ResponseWriter, r *http.Request) {
53
	player := strings.TrimPrefix(r.URL.Path, "/players/")
54
​
55
	switch r.Method {
56
	case http.MethodPost:
57
		p.processWin(w, player)
58
	case http.MethodGet:
59
		p.showScore(w, player)
60
	}
61
}
62
​
63
func (p *PlayerServer) showScore(w http.ResponseWriter, player string) {
64
	score := p.store.GetPlayerScore(player)
65
​
66
	if score == 0 {
67
		w.WriteHeader(http.StatusNotFound)
68
	}
69
​
70
	fmt.Fprint(w, score)
71
}
72
​
73
func (p *PlayerServer) processWin(w http.ResponseWriter, player string) {
74
	p.store.RecordWin(player)
75
	w.WriteHeader(http.StatusAccepted)
76
}
Copied!
1
// in_memory_player_store.go
2
package main
3
​
4
func NewInMemoryPlayerStore() *InMemoryPlayerStore {
5
	return &InMemoryPlayerStore{map[string]int{}}
6
}
7
​
8
type InMemoryPlayerStore struct {
9
	store map[string]int
10
}
11
​
12
func (i *InMemoryPlayerStore) GetLeague() []Player {
13
	var league []Player
14
	for name, wins := range i.store {
15
		league = append(league, Player{name, wins})
16
	}
17
	return league
18
}
19
​
20
func (i *InMemoryPlayerStore) RecordWin(name string) {
21
	i.store[name]++
22
}
23
​
24
func (i *InMemoryPlayerStore) GetPlayerScore(name string) int {
25
	return i.store[name]
26
}
Copied!
1
// main.go
2
package main
3
​
4
import (
5
	"log"
6
	"net/http"
7
)
8
​
9
func main() {
10
	server := NewPlayerServer(NewInMemoryPlayerStore())
11
	log.Fatal(http.ListenAndServe(":5000", server))
12
}
Copied!
You can find the corresponding tests in the link at the top of the chapter.
Store the data
There are dozens of databases we could use for this but we're going to go for a very simple approach. We're going to store the data for this application in a file as JSON.
This keeps the data very portable and is relatively simple to implement.
It won't scale especially well but given this is a prototype it'll be fine for now. If our circumstances change and it's no longer appropriate it'll be simple to swap it out for something different because of the PlayerStore abstraction we have used.
We will keep the InMemoryPlayerStore for now so that the integration tests keep passing as we develop our new store. Once we are confident our new implementation is sufficient to make the integration test pass we will swap it in and then delete InMemoryPlayerStore.
Write the test first
By now you should be familiar with the interfaces around the standard library for reading data (io.Reader), writing data (io.Writer) and how we can use the standard library to test these functions without having to use real files.
For this work to be complete we'll need to implement PlayerStore so we'll write tests for our store calling the methods we need to implement. We'll start with GetLeague.
1
//file_system_store_test.go
2
func TestFileSystemStore(t *testing.T) {
3
​
4
	t.Run("league from a reader", func(t *testing.T) {
5
		database := strings.NewReader(`[
6
            {"Name": "Cleo", "Wins": 10},
7
            {"Name": "Chris", "Wins": 33}]`)
8
​
9
		store := FileSystemPlayerStore{database}
10
​
11
		got := store.GetLeague()
12
​
13
		want := []Player{
14
			{"Cleo", 10},
15
			{"Chris", 33},
16
		}
17
​
18
		assertLeague(t, got, want)
19
	})
20
}
Copied!
We're using strings.NewReader which will return us a Reader, which is what our FileSystemPlayerStore will use to read data. In main we will open a file, which is also a Reader.
Try to run the test
1
# github.com/quii/learn-go-with-tests/io/v1
2
./file_system_store_test.go:15:12: undefined: FileSystemPlayerStore
Copied!
Write the minimal amount of code for the test to run and check the failing test output
Let's define FileSystemPlayerStore in a new file
1
//file_system_store.go
2
type FileSystemPlayerStore struct{}
Copied!
Try again
1
# github.com/quii/learn-go-with-tests/io/v1
2
./file_system_store_test.go:15:28: too many values in struct initializer
3
./file_system_store_test.go:17:15: store.GetLeague undefined (type FileSystemPlayerStore has no field or method GetLeague)
Copied!
It's complaining because we're passing in a Reader but not expecting one and it doesn't have GetLeague defined yet.
1
//file_system_store.go
2
type FileSystemPlayerStore struct {
3
	database io.Reader
4
}
5
​
6
func (f *FileSystemPlayerStore) GetLeague() []Player {
7
	return nil
8
}
Copied!
One more try...
1
=== RUN   TestFileSystemStore//league_from_a_reader
2
    --- FAIL: TestFileSystemStore//league_from_a_reader (0.00s)
3
        file_system_store_test.go:24: got [] want [{Cleo 10} {Chris 33}]
Copied!
Write enough code to make it pass
We've read JSON from a reader before
1
//file_system_store.go
2
func (f *FileSystemPlayerStore) GetLeague() []Player {
3
	var league []Player
4
	json.NewDecoder(f.database).Decode(&league)
5
	return league
6
}
Copied!
The test should pass.
Refactor
We have done this before! Our test code for the server had to decode the JSON from the response.
Let's try DRYing this up into a function.
Create a new file called league.go and put this inside.
1
//league.go
2
func NewLeague(rdr io.Reader) ([]Player, error) {
3
	var league []Player
4
	err := json.NewDecoder(rdr).Decode(&league)
5
	if err != nil {
6
		err = fmt.Errorf("problem parsing league, %v", err)
7
	}
8
​
9
	return league, err
10
}
Copied!
Call this in our implementation and in our test helper getLeagueFromResponse in server_test.go
1
//file_system_store.go
2
func (f *FileSystemPlayerStore) GetLeague() []Player {
3
	league, _ := NewLeague(f.database)
4
	return league
5
}
Copied!
We haven't got a strategy yet for dealing with parsing errors but let's press on.
Seeking problems
There is a flaw in our implementation. First of all, let's remind ourselves how io.Reader is defined.
1
type Reader interface {
2
	Read(p []byte) (n int, err error)
3
}
Copied!
With our file, you can imagine it reading through byte by byte until the end. What happens if you try to Read a second time?
Add the following to the end of our current test.
1
//file_system_store_test.go
2
​
3
// read again
4
got = store.GetLeague()
5
assertLeague(t, got, want)
Copied!
We want this to pass, but if you run the test it doesn't.
The problem is our Reader has reached the end so there is nothing more to read. We need a way to tell it to go back to the start.
​ReadSeeker is another interface in the standard library that can help.
1
type ReadSeeker interface {
2
	Reader
3
	Seeker
4
}
Copied!
Remember embedding? This is an interface comprised of Reader and Seeker​
1
type Seeker interface {
2
	Seek(offset int64, whence int) (int64, error)
3
}
Copied!
This sounds good, can we change FileSystemPlayerStore to take this interface instead?
1
//file_system_store.go
2
type FileSystemPlayerStore struct {
3
	database io.ReadSeeker
4
}
5
​
6
func (f *FileSystemPlayerStore) GetLeague() []Player {
7
	f.database.Seek(0, 0)
8
	league, _ := NewLeague(f.database)
9
	return league
10
}
Copied!
Try running the test, it now passes! Happily for us strings.NewReader that we used in our test also implements ReadSeeker so we didn't have to make any other changes.
Next we'll implement GetPlayerScore.
Write the test first
1
//file_system_store_test.go
2
t.Run("get player score", func(t *testing.T) {
3
	database := strings.NewReader(`[
4
        {"Name": "Cleo", "Wins": 10},
5
        {"Name": "Chris", "Wins": 33}]`)
6
​
7
	store := FileSystemPlayerStore{database}
8
​
9
	got := store.GetPlayerScore("Chris")
10
​
11
	want := 33
12
​
13
	if got != want {
14
		t.Errorf("got %d want %d", got, want)
15
	}
16
})
Copied!
Try to run the test
1
./file_system_store_test.go:38:15: store.GetPlayerScore undefined (type FileSystemPlayerStore has no field or method GetPlayerScore)
Copied!
Write the minimal amount of code for the test to run and check the failing test output
We need to add the method to our new type to get the test to compile.
1
//file_system_store.go
2
func (f *FileSystemPlayerStore) GetPlayerScore(name string) int {
3
	return 0
4
}
Copied!
Now it compiles and the test fails
1
=== RUN   TestFileSystemStore/get_player_score
2
    --- FAIL: TestFileSystemStore//get_player_score (0.00s)
3
        file_system_store_test.go:43: got 0 want 33
Copied!
Write enough code to make it pass
We can iterate over the league to find the player and return their score
1
//file_system_store.go
2
func (f *FileSystemPlayerStore) GetPlayerScore(name string) int {
3
​
4
	var wins int
5
​
6
	for _, player := range f.GetLeague() {
7
		if player.Name == name {
8
			wins = player.Wins
9
			break
10
		}
11
	}
12
​
13
	return wins
14
}
Copied!
Refactor
You will have seen dozens of test helper refactorings so I'll leave this to you to make it work
1
//file_system_store_test.go
2
t.Run("get player score", func(t *testing.T) {
3
	database := strings.NewReader(`[
4
        {"Name": "Cleo", "Wins": 10},
5
        {"Name": "Chris", "Wins": 33}]`)
6
​
7
	store := FileSystemPlayerStore{database}
8
​
9
	got := store.GetPlayerScore("Chris")
10
	want := 33
11
	assertScoreEquals(t, got, want)
12
})
Copied!
Finally, we need to start recording scores with RecordWin.
Write the test first
Our approach is fairly short-sighted for writes. We can't (easily) just update one "row" of JSON in a file. We'll need to store the whole new representation of our database on every write.
How do we write? We'd normally use a Writer but we already have our ReadSeeker. Potentially we could have two dependencies but the standard library already has an interface for us ReadWriteSeeker which lets us do all the things we'll need to do with a file.
Let's update our type
1
//file_system_store.go
2
type FileSystemPlayerStore struct {
3
	database io.ReadWriteSeeker
4
}
Copied!
See if it compiles
1
./file_system_store_test.go:15:34: cannot use database (type *strings.Reader) as type io.ReadWriteSeeker in field value:
2
    *strings.Reader does not implement io.ReadWriteSeeker (missing Write method)
3
./file_system_store_test.go:36:34: cannot use database (type *strings.Reader) as type io.ReadWriteSeeker in field value:
4
    *strings.Reader does not implement io.ReadWriteSeeker (missing Write method)
Copied!
It's not too surprising that strings.Reader does not implement ReadWriteSeeker so what do we do?
We have two choices
Create a temporary file for each test. *os.File implements ReadWriteSeeker. The pro of this is it becomes more of an integration test, we're really reading and writing from the file system so it will give us a very high level of confidence. The cons are we prefer unit tests because they are faster and generally simpler. We will also need to do more work around creating temporary files and then making sure they're removed after the test.
We could use a third party library. Mattetti has written a library filebuffer which implements the interface we need and doesn't touch the file system.
I don't think there's an especially wrong answer here, but by choosing to use a third party library I would have to explain dependency management! So we will use files instead.
Before adding our test we need to make our other tests compile by replacing the strings.Reader with an os.File.
Let's create some helper functions which will create a temporary file with some data inside it, and abstract our score tests
1
//file_system_store_test.go
2
func createTempFile(t testing.TB, initialData string) (io.ReadWriteSeeker, func()) {
3
	t.Helper()
4
​
5
	tmpfile, err := ioutil.TempFile("", "db")
6
​
7
	if err != nil {
8
		t.Fatalf("could not create temp file %v", err)
9
	}
10
​
11
	tmpfile.Write([]byte(initialData))
12
​
13
	removeFile := func() {
14
		tmpfile.Close()
15
		os.Remove(tmpfile.Name())
16
	}
17
​
18
	return tmpfile, removeFile
19
}
20
​
21
func assertScoreEquals(t testing.TB, got, want int) {
22
	t.Helper()
23
	if got != want {
24
		t.Errorf("got %d want %d", got, want)
25
	}
26
}
Copied!
​TempFile creates a temporary file for us to use. The "db" value we've passed in is a prefix put on a random file name it will create. This is to ensure it won't clash with other files by accident.
You'll notice we're not only returning our ReadWriteSeeker (the file) but also a function. We need to make sure that the file is removed once the test is finished. We don't want to leak details of the files into the test as it's prone to error and uninteresting for the reader. By returning a removeFile function, we can take care of the details in our helper and all the caller has to do is run defer cleanDatabase().
1
//file_system_store_test.go
2
func TestFileSystemStore(t *testing.T) {
3
​
4
	t.Run("league from a reader", func(t *testing.T) {
5
		database, cleanDatabase := createTempFile(t, `[
6
            {"Name": "Cleo", "Wins": 10},
7
            {"Name": "Chris", "Wins": 33}]`)
8
		defer cleanDatabase()
9
​
10
		store := FileSystemPlayerStore{database}
11
​
12
		got := store.GetLeague()
13
​
14
		want := []Player{
15
			{"Cleo", 10},
16
			{"Chris", 33},
17
		}
18
​
19
		assertLeague(t, got, want)
20
​
21
		// read again
22
		got = store.GetLeague()
23
		assertLeague(t, got, want)
24
	})
25
​
26
	t.Run("get player score", func(t *testing.T) {
27
		database, cleanDatabase := createTempFile(t, `[
28
            {"Name": "Cleo", "Wins": 10},
29
            {"Name": "Chris", "Wins": 33}]`)
30
		defer cleanDatabase()
31
​
32
		store := FileSystemPlayerStore{database}
33
​
34
		got := store.GetPlayerScore("Chris")
35
		want := 33
36
		assertScoreEquals(t, got, want)
37
	})
38
}
Copied!
Run the tests and they should be passing! There were a fair amount of changes but now it feels like we have our interface definition complete and it should be very easy to add new tests from now.
Let's get the first iteration of recording a win for an existing player
1
//file_system_store_test.go
2
t.Run("store wins for existing players", func(t *testing.T) {
3
	database, cleanDatabase := createTempFile(t, `[
4
        {"Name": "Cleo", "Wins": 10},
5
        {"Name": "Chris", "Wins": 33}]`)
6
	defer cleanDatabase()
7
​
8
	store := FileSystemPlayerStore{database}
9
​
10
	store.RecordWin("Chris")
11
​
12
	got := store.GetPlayerScore("Chris")
13
	want := 34
14
	assertScoreEquals(t, got, want)
15
})
Copied!
Try to run the test
./file_system_store_test.go:67:8: store.RecordWin undefined (type FileSystemPlayerStore has no field or method RecordWin)
Write the minimal amount of code for the test to run and check the failing test output
Add the new method
1
//file_system_store.go
2
func (f *FileSystemPlayerStore) RecordWin(name string) {
3
​
4
}
Copied!
1
=== RUN   TestFileSystemStore/store_wins_for_existing_players
2
    --- FAIL: TestFileSystemStore/store_wins_for_existing_players (0.00s)
3
        file_system_store_test.go:71: got 33 want 34
Copied!
Our implementation is empty so the old score is getting returned.
Write enough code to make it pass
1
//file_system_store.go
2
func (f *FileSystemPlayerStore) RecordWin(name string) {
3
	league := f.GetLeague()
4
​
5
	for i, player := range league {
6
		if player.Name == name {
7
			league[i].Wins++
8
		}
9
	}
10
​
11
	f.database.Seek(0, 0)
12
	json.NewEncoder(f.database).Encode(league)
13
}
Copied!
You may be asking yourself why I am doing league[i].Wins++ rather than player.Wins++.
When you range over a slice you are returned the current index of the loop (in our case i) and a copy of the element at that index. Changing the Wins value of a copy won't have any effect on the league slice that we iterate on. For that reason, we need to get the reference to the actual value by doing league[i] and then changing that value instead.
If you run the tests, they should now be passing.
Refactor
In GetPlayerScore and RecordWin, we are iterating over []Player to find a player by name.
We could refactor this common code in the internals of FileSystemStore but to me, it feels like this is maybe useful code we can lift into a new type. Working with a "League" so far has always been with []Player but we can create a new type called League. This will be easier for other developers to understand and then we can attach useful methods onto that type for us to use.
Inside league.go add the following
1
//league.go
2
type League []Player
3
​
4
func (l League) Find(name string) *Player {
5
	for i, p := range l {
6
		if p.Name == name {
7
			return &l[i]
8
		}
9
	}
10
	return nil
11
}
Copied!
Now if anyone has a League they can easily find a given player.
Change our PlayerStore interface to return League rather than []Player. Try to re-run the tests, you'll get a compilation problem because we've changed the interface but it's very easy to fix; just change the return type from []Player to League.
This lets us simplify our methods in file_system_store.
1
//file_system_store.go
2
func (f *FileSystemPlayerStore) GetPlayerScore(name string) int {
3
​
4
	player := f.GetLeague().Find(name)
5
​
6
	if player != nil {
7
		return player.Wins
8
	}
9
​
10
	return 0
11
}
12
​
13
func (f *FileSystemPlayerStore) RecordWin(name string) {
14
	league := f.GetLeague()
15
	player := league.Find(name)
16
​
17
	if player != nil {
18
		player.Wins++
19
	}
20
​
21
	f.database.Seek(0, 0)
22
	json.NewEncoder(f.database).Encode(league)
23
}
Copied!
This is looking much better and we can see how we might be able to find other useful functionality around League that can be refactored.
We now need to handle the scenario of recording wins of new players.
Write the test first
1
//file_system_store_test.go
2
t.Run("store wins for new players", func(t *testing.T) {
3
	database, cleanDatabase := createTempFile(t, `[
4
        {"Name": "Cleo", "Wins": 10},
5
        {"Name": "Chris", "Wins": 33}]`)
6
	defer cleanDatabase()
7
​
8
	store := FileSystemPlayerStore{database}
9
​
10
	store.RecordWin("Pepper")
11
​
12
	got := store.GetPlayerScore("Pepper")
13
	want := 1
14
	assertScoreEquals(t, got, want)
15
})
Copied!
Try to run the test
1
=== RUN   TestFileSystemStore/store_wins_for_new_players#01
2
    --- FAIL: TestFileSystemStore/store_wins_for_new_players#01 (0.00s)
3
        file_system_store_test.go:86: got 0 want 1
Copied!
Write enough code to make it pass
We just need to handle the scenario where Find returns nil because it couldn't find the player.
1
//file_system_store.go
2
func (f *FileSystemPlayerStore) RecordWin(name string) {
3
	league := f.GetLeague()
4
	player := league.Find(name)
5
​
6
	if player != nil {
7
		player.Wins++
8
	} else {
9
		league = append(league, Player{name, 1})
10
	}
11
​
12
	f.database.Seek(0, 0)
13
	json.NewEncoder(f.database).Encode(league)
14
}
Copied!
The happy path is looking ok so we can now try using our new Store in the integration test. This will give us more confidence that the software works and then we can delete the redundant InMemoryPlayerStore.
In TestRecordingWinsAndRetrievingThem replace the old store.
1
//server_integration_test.go
2
database, cleanDatabase := createTempFile(t, "")
3
defer cleanDatabase()
4
store := &FileSystemPlayerStore{database}
Copied!
If you run the test it should pass and now we can delete InMemoryPlayerStore. main.go will now have compilation problems which will motivate us to now use our new store in the "real" code.
1
//main.go
2
package main
3
​
4
import (
5
	"log"
6
	"net/http"
7
	"os"
8
)
9
​
10
const dbFileName = "game.db.json"
11
​
12
func main() {
13
	db, err := os.OpenFile(dbFileName, os.O_RDWR|os.O_CREATE, 0666)
14
​
15
	if err != nil {
16
		log.Fatalf("problem opening %s %v", dbFileName, err)
17
	}
18
​
19
	store := &FileSystemPlayerStore{db}
20
	server := NewPlayerServer(store)
21
​
22
	if err := http.ListenAndServe(":5000", server); err != nil {
23
		log.Fatalf("could not listen on port 5000 %v", err)
24
	}
25
}
Copied!
We create a file for our database.
The 2nd argument to os.OpenFile lets you define the permissions for opening the file, in our case O_RDWR means we want to read and write and os.O_CREATE means create the file if it doesn't exist.
The 3rd argument means sets permissions for the file, in our case, all users can read and write the file. (See superuser.com for a more detailed explanation).
Running the program now persists the data in a file in between restarts, hooray!
More refactoring and performance concerns
Every time someone calls GetLeague() or GetPlayerScore() we are reading the entire file and parsing it into JSON. We should not have to do that because FileSystemStore is entirely responsible for the state of the league; it should only need to read the file when the program starts up and only need to update the file when data changes.
We can create a constructor which can do some of this initialisation for us and store the league as a value in our FileSystemStore to be used on the reads instead.
1
//file_system_store.go
2
type FileSystemPlayerStore struct {
3
	database io.ReadWriteSeeker
4
	league   League
5
}
6
​
7
func NewFileSystemPlayerStore(database io.ReadWriteSeeker) *FileSystemPlayerStore {
8
	database.Seek(0, 0)
9
	league, _ := NewLeague(database)
10
	return &FileSystemPlayerStore{
11
		database: database,
12
		league:   league,
13
	}
14
}
Copied!
This way we only have to read from disk once. We can now replace all of our previous calls to getting the league from disk and just use f.league instead.
1
//file_system_store.go
2
func (f *FileSystemPlayerStore) GetLeague() League {
3
	return f.league
4
}
5
​
6
func (f *FileSystemPlayerStore) GetPlayerScore(name string) int {
7
​
8
	player := f.league.Find(name)
9
​
10
	if player != nil {
11
		return player.Wins
12
	}
13
​
14
	return 0
15
}
16
​
17
func (f *FileSystemPlayerStore) RecordWin(name string) {
18
	player := f.league.Find(name)
19
​
20
	if player != nil {
21
		player.Wins++
22
	} else {
23
		f.league = append(f.league, Player{name, 1})
24
	}
25
​
26
	f.database.Seek(0, 0)
27
	json.NewEncoder(f.database).Encode(f.league)
28
}
Copied!
If you try to run the tests it will now complain about initialising FileSystemPlayerStore so just fix them by calling our new constructor.
Another problem
There is some more naivety in the way we are dealing with files which could create a very nasty bug down the line.
When we RecordWin, we Seek back to the start of the file and then write the new data—but what if the new data was smaller than what was there before?
In our current case, this is impossible. We never edit or delete scores so the data can only get bigger. However, it would be irresponsible for us to leave the code like this; it's not unthinkable that a delete scenario could come up.
How will we test for this though? What we need to do is first refactor our code so we separate out the concern of the kind of data we write, from the writing. We can then test that separately to check it works how we hope.
We'll create a new type to encapsulate our "when we write we go from the beginning" functionality. I'm going to call it Tape. Create a new file with the following:
1
//tape.go
2
package main
3
​
4
import "io"
5
​
6
type tape struct {
7
	file io.ReadWriteSeeker
8
}
9
​
10
func (t *tape) Write(p []byte) (n int, err error) {
11
	t.file.Seek(0, 0)
12
	return t.file.Write(p)
13
}
Copied!
Notice that we're only implementing Write now, as it encapsulates the Seek part. This means our FileSystemStore can just have a reference to a Writer instead.
1
//file_system_store.go
2
type FileSystemPlayerStore struct {
3
	database io.Writer
4
	league   League
5
}
Copied!
Update the constructor to use Tape
1
//file_system_store.go
2
func NewFileSystemPlayerStore(database io.ReadWriteSeeker) *FileSystemPlayerStore {
3
	database.Seek(0, 0)
4
	league, _ := NewLeague(database)
5
​
6
	return &FileSystemPlayerStore{
7
		database: &tape{database},
8
		league:   league,
9
	}
10
}
Copied!
Finally, we can get the amazing payoff we wanted by removing the Seek call from RecordWin. Yes, it doesn't feel much, but at least it means if we do any other kind of writes we can rely on our Write to behave how we need it to. Plus it will now let us test the potentially problematic code separately and fix it.
Let's write the test where we want to update the entire contents of a file with something that is smaller than the original contents.
Write the test first
Our test will create a file with some content, try to write to it using the tape, and read it all again to see what's in the file. In tape_test.go:
1
//tape_test.go
2
func TestTape_Write(t *testing.T) {
3
	file, clean := createTempFile(t, "12345")
4
	defer clean()
5
​
6
	tape := &tape{file}
7
​
8
	tape.Write([]byte("abc"))
9
​
10
	file.Seek(0, 0)
11
	newFileContents, _ := ioutil.ReadAll(file)
12
​
13
	got := string(newFileContents)
14
	want := "abc"
15
​
16
	if got != want {
17
		t.Errorf("got %q want %q", got, want)
18
	}
19
}
Copied!
Try to run the test
1
=== RUN   TestTape_Write
2
--- FAIL: TestTape_Write (0.00s)
3
    tape_test.go:23: got 'abc45' want 'abc'
Copied!
As we thought! It writes the data we want, but leaves the rest of the original data remaining.
Write enough code to make it pass
os.File has a truncate function that will let us effectively empty the file. We should be able to just call this to get what we want.
Change tape to the following:
1
//tape.go
2
type tape struct {
3
	file *os.File
4
}
5
​
6
func (t *tape) Write(p []byte) (n int, err error) {
7
	t.file.Truncate(0)
8
	t.file.Seek(0, 0)
9
	return t.file.Write(p)
10
}
Copied!
The compiler will fail in a number of places where we are expecting an io.ReadWriteSeeker but we are sending in *os.File. You should be able to fix these problems yourself by now but if you get stuck just check the source code.
Once you get it refactoring our TestTape_Write test should be passing!
One other small refactor
In RecordWin we have the line json.NewEncoder(f.database).Encode(f.league).
We don't need to create a new encoder every time we write, we can initialise one in our constructor and use that instead.
Store a reference to an Encoder in our type and initialise it in the constructor:
1
//file_system_store.go
2
type FileSystemPlayerStore struct {
3
	database *json.Encoder
4
	league   League
5
}
6
​
7
func NewFileSystemPlayerStore(file *os.File) *FileSystemPlayerStore {
8
	file.Seek(0, 0)
9
	league, _ := NewLeague(file)
10
​
11
	return &FileSystemPlayerStore{
12
		database: json.NewEncoder(&tape{file}),
13
		league:   league,
14
	}
15
}
Copied!
Use it in RecordWin.
1
func (f *FileSystemPlayerStore) RecordWin(name string) {
2
	player := f.league.Find(name)
3
​
4
	if player != nil {
5
		player.Wins++
6
	} else {
7
		f.league = append(f.league, Player{name, 1})
8
	}
9
​
10
	f.database.Encode(f.league)
11
}
Copied!
Didn't we just break some rules there? Testing private things? No interfaces?
On testing private types
It's true that in general you should favour not testing private things as that can sometimes lead to your tests being too tightly coupled to the implementation, which can hinder refactoring in future.
However, we must not forget that tests should give us confidence.
We were not confident that our implementation would work if we added any kind of edit or delete functionality. We did not want to leave the code like that, especially if this was being worked on by more than one person who may not be aware of the shortcomings of our initial approach.
Finally, it's just one test! If we decide to change the way it works it won't be a disaster to just delete the test but we have at the very least captured the requirement for future maintainers.
Interfaces
We started off the code by using io.Reader as that was the easiest path for us to unit test our new PlayerStore. As we developed the code we moved on to io.ReadWriter and then io.ReadWriteSeeker. We then found out there was nothing in the standard library that actually implemented that apart from *os.File. We could've taken the decision to write our own or use an open source one but it felt pragmatic just to make temporary files for the tests.
Finally, we needed Truncate which is also on *os.File. It would've been an option to create our own interface capturing these requirements.
1
type ReadWriteSeekTruncate interface {
2
	io.ReadWriteSeeker
3
	Truncate(size int64) error
4
}
Copied!
But what is this really giving us? Bear in mind we are not mocking and it is unrealistic for a file system store to take any type other than an *os.File so we don't need the polymorphism that interfaces give us.
Don't be afraid to chop and change types and experiment like we have here. The great thing about using a statically typed language is the compiler will help you with every change.
Error handling
Before we start working on sorting we should make sure we're happy with our current code and remove any technical debt we may have. It's an important principle to get to working software as quickly as possible (stay out of the red state) but that doesn't mean we should ignore error cases!
If we go back to FileSystemStore.go we have league, _ := NewLeague(f.database) in our constructor.
NewLeague can return an error if it is unable to parse the league from the io.Reader that we provide.
It was pragmatic to ignore that at the time as we already had failing tests. If we had tried to tackle it at the same time, we would have been juggling two things at once.
Let's make it so our constructor is capable of returning an error.
1
//file_system_store.go
2
func NewFileSystemPlayerStore(file *os.File) (*FileSystemPlayerStore, error) {
3
	file.Seek(0, 0)
4
	league, err := NewLeague(file)
5
​
6
	if err != nil {
7
		return nil, fmt.Errorf("problem loading player store from file %s, %v", file.Name(), err)
8
	}
9
​
10
	return &FileSystemPlayerStore{
11
		database: json.NewEncoder(&tape{file}),
12
		league:   league,
13
	}, nil
14
}
Copied!
Remember it is very important to give helpful error messages (just like your tests). People on the internet jokingly say that most Go code is:
1
if err != nil {
2
	return err
3
}
Copied!
That is 100% not idiomatic. Adding contextual information (i.e what you were doing to cause the error) to your error messages makes operating your software far easier.
If you try to compile you'll get some errors.
1
./main.go:18:35: multiple-value NewFileSystemPlayerStore() in single-value context
2
./file_system_store_test.go:35:36: multiple-value NewFileSystemPlayerStore() in single-value context
3
./file_system_store_test.go:57:36: multiple-value NewFileSystemPlayerStore() in single-value context
4
./file_system_store_test.go:70:36: multiple-value NewFileSystemPlayerStore() in single-value context
5
./file_system_store_test.go:85:36: multiple-value NewFileSystemPlayerStore() in single-value context
6
./server_integration_test.go:12:35: multiple-value NewFileSystemPlayerStore() in single-value context
Copied!
In main we'll want to exit the program, printing the error.
1
//main.go
2
store, err := NewFileSystemPlayerStore(db)
3
​
4
if err != nil {
5
	log.Fatalf("problem creating file system player store, %v ", err)
6
}