From cae274b379751752b58e17d730393396e6c6f31f Mon Sep 17 00:00:00 2001
From: Thomas Nixon <tom@tomn.co.uk>
Date: Mon, 13 Nov 2023 13:26:57 +0000
Subject: [PATCH] add test for too many big chunks

---
 tests/file_tests.cpp | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)

diff --git a/tests/file_tests.cpp b/tests/file_tests.cpp
index f96fe90..a4b4d43 100644
--- a/tests/file_tests.cpp
+++ b/tests/file_tests.cpp
@@ -343,3 +343,38 @@ TEST_CASE("write_read_big_axml", "[.big]") {
 
   remove(filename.c_str());
 }
+
+/// dummy chunk that just writes spaces
+class MegaChunk : public Chunk {
+ public:
+  MegaChunk(uint32_t id, uint64_t size) : id_(id), size_(size) {}
+
+  uint32_t id() const override { return id_; }
+  uint64_t size() const override { return size_; }
+
+  void write(std::ostream& stream) const override {
+    for (size_t i = 0; i < size_; i++) stream << ' ';
+  }
+
+ private:
+  uint32_t id_;
+  uint64_t size_;
+};
+
+TEST_CASE("write_too_many_big_chunks", "[.big]") {
+  std::string filename = "too_many_big_chunks.wav";
+
+  auto bw64File = writeFile(filename, 1, 48000, 16, nullptr);
+
+  // TODO: setAxmlChunk should really be renamed...
+  bw64File->setAxmlChunk(
+      std::make_shared<MegaChunk>(utils::fourCC("mega"), 0x100000000ull));
+  bw64File->setAxmlChunk(
+      std::make_shared<MegaChunk>(utils::fourCC("megb"), 0x100000000ull));
+
+  REQUIRE_THROWS_WITH(
+      bw64File->close(),
+      "ds64 chunk is too large (52 bytes) to overwrite JUNK chunk (40 bytes)");
+
+  remove(filename.c_str());
+}