summaryrefslogtreecommitdiffstats
path: root/test/libapt/getlistoffilesindir_test.cc
blob: f00235504005605c2238c750f9c5056d1f51c81b (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
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
#include <config.h>

#include <apt-pkg/error.h>
#include <apt-pkg/fileutl.h>

#include <iostream>
#include <string>
#include <vector>

#include <gtest/gtest.h>

#include "file-helpers.h"

#define P(x)	tempdir + "/" + x

TEST(FileUtlTest,GetListOfFilesInDir)
{
   std::string tempdir;
   createTemporaryDirectory("getlistoffiles", tempdir);

   createFile(tempdir, "anormalfile");
   createFile(tempdir, "01yet-anothernormalfile");
   createFile(tempdir, "anormalapt.conf");
   createFile(tempdir, "01yet-anotherapt.conf");
   createFile(tempdir, "anormalapt.list");
   createFile(tempdir, "01yet-anotherapt.list");
   createFile(tempdir, "wrongextension.wron");
   createFile(tempdir, "wrong-extension.wron");
   createFile(tempdir, "strangefile.");
   createFile(tempdir, "s.t.r.a.n.g.e.f.i.l.e");
   createFile(tempdir, ".hiddenfile");
   createFile(tempdir, ".hiddenfile.conf");
   createFile(tempdir, ".hiddenfile.list");
   createFile(tempdir, "multi..dot");
   createFile(tempdir, "multi.dot.conf");
   createFile(tempdir, "multi.dot.list");
   createFile(tempdir, "disabledfile.disabled");
   createFile(tempdir, "disabledfile.conf.disabled");
   createFile(tempdir, "disabledfile.list.disabled");
   createFile(tempdir, "invälid.conf");
   createFile(tempdir, "invalíd");
   createFile(tempdir, "01invalíd");
   createDirectory(tempdir, "invaliddir");
   createDirectory(tempdir, "directory.conf");
   createDirectory(tempdir, "directory.list");
   createDirectory(tempdir, "directory.wron");
   createDirectory(tempdir, "directory.list.disabled");
   createLink(tempdir, "anormalfile", "linkedfile.list");
   createLink(tempdir, "invaliddir", "linkeddir.list");
   createLink(tempdir, "non-existing-file", "brokenlink.list");

   // Files with no extension
   _error->PushToStack();
   std::vector<std::string> files = GetListOfFilesInDir(tempdir, "", true);
   ASSERT_EQ(2u, files.size());
   EXPECT_EQ(P("01yet-anothernormalfile"), files[0]);
   EXPECT_EQ(P("anormalfile"), files[1]);

   // Files with no extension - should be the same as above
   files = GetListOfFilesInDir(tempdir, "", true, true);
   ASSERT_EQ(2u, files.size());
   EXPECT_EQ(P("01yet-anothernormalfile"), files[0]);
   EXPECT_EQ(P("anormalfile"), files[1]);

   // Files with impossible extension
   files = GetListOfFilesInDir(tempdir, "impossible", true);
   EXPECT_TRUE(files.empty());

   // Files with impossible or no extension
   files = GetListOfFilesInDir(tempdir, "impossible", true, true);
   ASSERT_EQ(2u, files.size());
   EXPECT_EQ(P("01yet-anothernormalfile"), files[0]);
   EXPECT_EQ(P("anormalfile"), files[1]);

   // Files with list extension - nothing more
   files = GetListOfFilesInDir(tempdir, "list", true);
   ASSERT_EQ(4u, files.size());
   EXPECT_EQ(P("01yet-anotherapt.list"), files[0]);
   EXPECT_EQ(P("anormalapt.list"), files[1]);
   EXPECT_EQ(P("linkedfile.list"), files[2]);
   EXPECT_EQ(P("multi.dot.list"), files[3]);

   // Files with conf or no extension
   files = GetListOfFilesInDir(tempdir, "conf", true, true);
   ASSERT_EQ(5u, files.size());
   EXPECT_EQ(P("01yet-anotherapt.conf"), files[0]);
   EXPECT_EQ(P("01yet-anothernormalfile"), files[1]);
   EXPECT_EQ(P("anormalapt.conf"), files[2]);
   EXPECT_EQ(P("anormalfile"), files[3]);
   EXPECT_EQ(P("multi.dot.conf"), files[4]);

   // Files with disabled extension - nothing more
   files = GetListOfFilesInDir(tempdir, "disabled", true);
   ASSERT_EQ(3u, files.size());
   EXPECT_EQ(P("disabledfile.conf.disabled"), files[0]);
   EXPECT_EQ(P("disabledfile.disabled"), files[1]);
   EXPECT_EQ(P("disabledfile.list.disabled"), files[2]);

   // Files with disabled or no extension
   files = GetListOfFilesInDir(tempdir, "disabled", true, true);
   ASSERT_EQ(5u, files.size());
   EXPECT_EQ(P("01yet-anothernormalfile"), files[0]);
   EXPECT_EQ(P("anormalfile"), files[1]);
   EXPECT_EQ(P("disabledfile.conf.disabled"), files[2]);
   EXPECT_EQ(P("disabledfile.disabled"), files[3]);
   EXPECT_EQ(P("disabledfile.list.disabled"), files[4]);

   // discard the unknown file extension messages
   if (_error->PendingError())
      _error->MergeWithStack();
   else
      _error->RevertToStack();
   removeDirectory(tempdir);
}