tests: deduplicate package creation framework code

[apt.git] / test / libapt / sourcelist_test.cc

diff --git a/test/libapt/sourcelist_test.cc b/test/libapt/sourcelist_test.cc
index b9dd47207d6add9734c38bec64a6223de8bac6e8..83c441092f6622e43db9b31199c6a58ae7ce1f13 100644 (file)
--- a/test/libapt/sourcelist_test.cc
+++ b/test/libapt/sourcelist_test.cc
@@ -1,52 +1,39 @@
+#include <config.h>
+

 #include <apt-pkg/sourcelist.h>
 #include <apt-pkg/sourcelist.h>

-#include <apt-pkg/tagfile.h>
+#include <apt-pkg/fileutl.h>

 
 

-#include "assert.h"
+#include <string>

 #include <stdlib.h>
 #include <string.h>
 #include <unistd.h>
 
 #include <stdlib.h>
 #include <string.h>
 #include <unistd.h>
 

-char *tempfile = NULL;
-int tempfile_fd = -1;
+#include <gtest/gtest.h>

 
 

-void remove_tmpfile(void)
-{
-   if (tempfile_fd > 0)
-      close(tempfile_fd);
-   if (tempfile != NULL) {
-      unlink(tempfile);
-      free(tempfile);
-   }
-}
+#include "file-helpers.h"

 
 

-int main(int argc, char *argv[])
+TEST(SourceListTest,ParseFileDeb822)

 {
 {

-   const char contents[] = ""
-      "Type: deb\n"
-      "URI: http://ftp.debian.org/debian\n"
+   FileFd fd;
+   std::string tempfile;
+   createTemporaryFile("parsefiledeb822.XXXXXX.sources", fd, &tempfile,
+      "Types: deb\n"
+      "URIs: http://ftp.debian.org/debian\n"

       "Suites: stable\n"
       "Suites: stable\n"

-      "Sections: main\n"
+      "Components: main\n"

       "Description: short\n"
       " long description that can be very long\n"
       "\n"
       "Description: short\n"
       " long description that can be very long\n"
       "\n"

-      "Type: deb\n"
-      "URI: http://ftp.debian.org/debian\n"
-      "Suite: unstable\n"
-      "Section: main non-free\n"
-      ;
-
-   FileFd fd;
-   tempfile = strdup("apt-test.XXXXXXXX");
-   tempfile_fd = mkstemp(tempfile);
-
-   /* (Re-)Open (as FileFd), write and seek to start of the temp file */
-   equals(fd.OpenDescriptor(tempfile_fd, FileFd::ReadWrite), true);
-   equals(fd.Write(contents, strlen(contents)), true);
-   equals(fd.Seek(0), true);
-
-   pkgSourceList sources(tempfile);
-   equals(sources.size(), 2);
-
-   /* clean up handled by atexit handler, so just return here */
-   return 0;
+      "Types: deb\n"
+      "URIs: http://ftp.debian.org/debian\n"
+      "Suites: unstable\n"
+      "Components: main non-free\n");
+   fd.Close();
+
+   pkgSourceList sources;
+   EXPECT_EQ(true, sources.Read(tempfile));
+   EXPECT_EQ(2, sources.size());
+
+   if (tempfile.empty() == false)
+      unlink(tempfile.c_str());

 }
 }