forked from arduino/go-apt-client
-
Notifications
You must be signed in to change notification settings - Fork 0
/
repos_test.go
122 lines (103 loc) · 4.38 KB
/
repos_test.go
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
//
// This file is part of go-apt-client library
//
// Copyright (C) 2017 Arduino AG (http://www.arduino.cc/)
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
package apt
import (
"encoding/json"
"io/ioutil"
"os"
"testing"
"github.com/stretchr/testify/require"
)
func TestParseAPTConfigFolder(t *testing.T) {
repos, err := ParseAPTConfigFolder("testdata/apt")
require.NoError(t, err, "running List command")
expectedData, err := ioutil.ReadFile("testdata/TestParseAPTConfigFolder.json")
require.NoError(t, err, "Reading test data")
expected := []*Repository{}
err = json.Unmarshal(expectedData, &expected)
require.NoError(t, err, "Decoding expected data")
for i, repo := range repos {
require.EqualValues(t, expected[i], repo, "Comparing element %d", i)
}
}
func TestAddAndRemoveRepository(t *testing.T) {
// test cleanup
defer os.Remove("testdata/apt2/sources.list.d/managed.list")
defer os.Remove("testdata/apt2/sources.list.d/managed.list.save")
defer os.Remove("testdata/apt2/sources.list.d/managed.list.new")
repo1 := &Repository{
Enabled: true,
SourceRepo: false,
URI: "http://ppa.launchpad.net/webupd8team/java/ubuntu",
Distribution: "zesty",
Components: "main",
Comment: "",
}
repo2 := &Repository{
Enabled: false,
SourceRepo: true,
URI: "http://ppa.launchpad.net/webupd8team/java/ubuntu",
Distribution: "zesty",
Components: "main",
Comment: "",
}
err := AddRepository(repo1, "testdata/apt2")
require.NoError(t, err, "Adding repository")
err = AddRepository(repo2, "testdata/apt2")
require.NoError(t, err, "Adding repository")
// check that we have repo1 and repo2 added
repos, err := ParseAPTConfigFolder("testdata/apt2")
require.NoError(t, err, "running List command")
require.True(t, repos.Contains(repo1), "Configuration contains: %#v", repo1)
require.True(t, repos.Contains(repo2), "Configuration contains: %#v", repo2)
err = AddRepository(repo2, "testdata/apt2")
require.Error(t, err, "Adding repository again")
// no changes should have happened
repos, err = ParseAPTConfigFolder("testdata/apt2")
require.NoError(t, err, "running List command")
require.True(t, repos.Contains(repo1), "Configuration contains: %#v", repo1)
require.True(t, repos.Contains(repo2), "Configuration contains: %#v", repo2)
err = RemoveRepository(repo2, "testdata/apt2")
require.NoError(t, err, "Removing repository")
// repo2 should be removed
repos, err = ParseAPTConfigFolder("testdata/apt2")
require.NoError(t, err, "running List command")
require.True(t, repos.Contains(repo1), "Configuration contains: %#v", repo1)
require.False(t, repos.Contains(repo2), "Configuration contains: %#v", repo2)
err = RemoveRepository(repo2, "testdata/apt2")
require.Error(t, err, "Removing repository again")
// no changes should have happened
repos, err = ParseAPTConfigFolder("testdata/apt2")
require.NoError(t, err, "running List command")
require.True(t, repos.Contains(repo1), "Configuration contains: %#v", repo1)
require.False(t, repos.Contains(repo2), "Configuration contains: %#v", repo2)
err = EditRepository(repo1, repo2, "testdata/apt2")
require.NoError(t, err, "editing repository %#V -> %#V", repo1, repo2)
// repo2 should be changed to repo1
repos, err = ParseAPTConfigFolder("testdata/apt2")
require.NoError(t, err, "running List command")
require.False(t, repos.Contains(repo1), "Configuration contains: %#v", repo1)
require.True(t, repos.Contains(repo2), "Configuration contains: %#v", repo2)
err = EditRepository(repo1, repo2, "testdata/apt2")
require.Error(t, err, "editing again repository %#v -> %#v", repo1, repo2)
// no changes should have happened
repos, err = ParseAPTConfigFolder("testdata/apt2")
require.NoError(t, err, "running List command")
require.False(t, repos.Contains(repo1), "Configuration contains: %#v", repo1)
require.True(t, repos.Contains(repo2), "Configuration contains: %#v", repo2)
}